use uuid to update user plan

Fitcel · Apr 15, 2023 · 3bd07b2 · 3bd07b2
1 parent 5b6fb6c
commit 3bd07b2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 10 deletions.
diff --git a/controllers/user.go b/controllers/user.go
@@ -9,14 +9,10 @@ func (c *Controller) AddUser(user models.User) (*models.User, error) {
 	return c.Model.AddUser(user)
 }
 
-func (c *Controller) UpdateUser(userID string, dietID string) error {
-	userid, err := strconv.Atoi(userID)
-	if err != nil {
-		return err
-	}
+func (c *Controller) UpdateUser(UUID string, dietID string) error {
 	dietid, err := strconv.Atoi(dietID)
 	if err != nil {
 		return err
 	}
-	return c.Model.UpdateUser(uint(userid), uint(dietid))
+	return c.Model.UpdateUser(UUID, uint(dietid))
 }
diff --git a/handlers/user.go b/handlers/user.go
@@ -21,7 +21,7 @@ func (h Handler) AddUser(c echo.Context) error {
 }
 
 func (h Handler) UpdateUser(c echo.Context) error {
-	userID := c.QueryParam("userID")
+	userID := c.QueryParam("UUID")
 	dietID := c.QueryParam("dietID")
 	if err := h.Controller.UpdateUser(userID, dietID); err != nil {
 		return c.JSON(http.StatusBadRequest, err.Error())

diff --git a/models/user.go b/models/user.go
@@ -13,7 +13,6 @@ type User struct {
 	Email     string     `json:"email"`
 	CreatedAt *time.Time `json:"createdAt"`
 	UpdatedAt *time.Time `json:"updatedAt"`
-	DeletedAt *time.Time `json:"deletedAt"`
 	DietID    *uint
 	Diet      *Diet
 }
@@ -34,6 +33,6 @@ func (m Model) AddUser(user User) (*User, error) {
 	err = m.DB.Create(&user).Error
 	return &user, err
 }
-func (m Model) UpdateUser(userID uint, dietID uint) error {
-	return m.DB.Model(User{ID: userID}).Update("diet_id", dietID).Error
+func (m Model) UpdateUser(UUID string, dietID uint) error {
+	return m.DB.Debug().Model(&User{}).Where("uuid", UUID).Update("diet_id", dietID).Error
 }