您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
 
 
 
 

98 行
2.6 KiB

  1. package controller
  2. import (
  3. "context"
  4. "encoding/json"
  5. "log/slog"
  6. "net/http"
  7. "strconv"
  8. "github.com/AFASystems/presence/internal/pkg/api/response"
  9. "github.com/AFASystems/presence/internal/pkg/model"
  10. "github.com/AFASystems/presence/internal/pkg/validation"
  11. "github.com/gorilla/mux"
  12. "gorm.io/gorm"
  13. )
  14. func FloorAddController(db *gorm.DB, context context.Context) http.HandlerFunc {
  15. return func(w http.ResponseWriter, r *http.Request) {
  16. var floor model.Floor
  17. if err := json.NewDecoder(r.Body).Decode(&floor); err != nil {
  18. response.BadRequest(w, "invalid request body")
  19. return
  20. }
  21. if err := validation.Struct(&floor); err != nil {
  22. response.BadRequest(w, err.Error())
  23. return
  24. }
  25. if err := db.WithContext(context).Create(&floor).Error; err != nil {
  26. response.InternalError(w, "failed to create floor", err)
  27. return
  28. }
  29. response.JSON(w, http.StatusCreated, map[string]string{"status": "created"})
  30. }
  31. }
  32. func FloorListController(db *gorm.DB, context context.Context) http.HandlerFunc {
  33. return func(w http.ResponseWriter, r *http.Request) {
  34. query := r.URL.Query()
  35. lStr := query.Get("limit")
  36. if lStr == "" {
  37. lStr = "100"
  38. }
  39. limit, err := strconv.Atoi(lStr)
  40. if err != nil {
  41. slog.Error("invalid limit parameter", "value", lStr)
  42. response.BadRequest(w, "invalid limit parameter")
  43. return
  44. }
  45. oStr := query.Get("offset")
  46. if oStr == "" {
  47. oStr = "0"
  48. }
  49. offset, err := strconv.Atoi(oStr)
  50. if err != nil {
  51. slog.Error("invalid offset parameter", "value", oStr)
  52. response.BadRequest(w, "invalid offset parameter")
  53. return
  54. }
  55. var floors []model.Floor
  56. if err := db.WithContext(context).Limit(limit).Offset(offset).Find(&floors).Error; err != nil {
  57. response.InternalError(w, "failed to list floors", err)
  58. return
  59. }
  60. response.JSON(w, http.StatusOK, floors)
  61. }
  62. }
  63. func FloorUpdateController(db *gorm.DB, context context.Context) http.HandlerFunc {
  64. return func(w http.ResponseWriter, r *http.Request) {
  65. var floor model.Floor
  66. if err := json.NewDecoder(r.Body).Decode(&floor); err != nil {
  67. response.BadRequest(w, "invalid request body")
  68. return
  69. }
  70. }
  71. }
  72. func FloorDeleteController(db *gorm.DB, context context.Context) http.HandlerFunc {
  73. return func(w http.ResponseWriter, r *http.Request) {
  74. id := mux.Vars(r)["id"]
  75. res := db.WithContext(context).Delete(&model.Floor{}, "id = ?", id)
  76. if res.RowsAffected == 0 {
  77. response.NotFound(w, "floor not found")
  78. return
  79. }
  80. if res.Error != nil {
  81. response.InternalError(w, "failed to delete floor", res.Error)
  82. return
  83. }
  84. response.JSON(w, http.StatusOK, map[string]string{"status": "deleted"})
  85. }
  86. }