feat(controllers/reward.SetComment)!: match api changes

ShanghaitechGeekPie · May 12, 2024 · c8df9e7 · c8df9e7
1 parent 74e8f66
commit c8df9e7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/internal/controllers/reward.go b/internal/controllers/reward.go
@@ -9,5 +9,5 @@ import (
 func RewardRoutes(r fiber.Router) {
 	route := r.Group("/reward")
 	route.Get("/ranklist", reward.Ranklist)
-	route.Post("/set/:id", reward.SetComment)
+	route.Post("/set", reward.SetComment)
 }
diff --git a/internal/controllers/reward/setcomment.go b/internal/controllers/reward/setcomment.go
@@ -6,13 +6,13 @@ import (
 	"coursebench-backend/pkg/errors"
 	"coursebench-backend/pkg/models"
 	"coursebench-backend/pkg/queries"
-	"strconv"
 
 	"github.com/gofiber/fiber/v2"
 	"gorm.io/gorm"
 )
 
 type SetCommentRequest struct {
+	ID     int `json:"id"`
 	Reward int `json:"reward"`
 }
 
@@ -23,12 +23,6 @@ func SetComment(c *fiber.Ctx) error {
 		return errors.Wrap(err, errors.InvalidArgument)
 	}
 
-	idRaw := c.Params("id", "GG")
-	id, err := strconv.Atoi(idRaw)
-	if err != nil {
-		return errors.New(errors.InvalidArgument)
-	}
-
 	uid, err := session.GetUserID(c)
 	if err != nil {
 		uid = 0
@@ -44,7 +38,7 @@ func SetComment(c *fiber.Ctx) error {
 	}
 
 	comment := &models.Comment{}
-	result := db.First(&comment, id)
+	result := db.First(&comment, request.ID)
 	if result.Error != nil {
 		return errors.Wrap(result.Error, errors.DatabaseError)
 	}