fix: group api param id

labring · Nov 21, 2024 · 8119ff8 · 8119ff8
1 parent 9738150
commit 8119ff8
Showing 1 changed file with 22 additions and 8 deletions.
diff --git a/service/aiproxy/controller/group.go b/service/aiproxy/controller/group.go
@@ -125,21 +125,28 @@ func GetGroupDashboard(c *gin.Context) {
 }
 
 type UpdateGroupQPMRequest struct {
-	ID  string `json:"id"`
-	QPM int64  `json:"qpm"`
+	QPM int64 `json:"qpm"`
 }
 
 func UpdateGroupQPM(c *gin.Context) {
+	id := c.Param("id")
+	if id == "" {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": "invalid parameter",
+		})
+		return
+	}
 	req := UpdateGroupQPMRequest{}
 	err := json.NewDecoder(c.Request.Body).Decode(&req)
-	if err != nil || req.ID == "" {
+	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"success": false,
 			"message": "invalid parameter",
 		})
 		return
 	}
-	err = model.UpdateGroupQPM(req.ID, req.QPM)
+	err = model.UpdateGroupQPM(id, req.QPM)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"success": false,
@@ -154,21 +161,28 @@ func UpdateGroupQPM(c *gin.Context) {
 }
 
 type UpdateGroupStatusRequest struct {
-	ID     string `json:"id"`
-	Status int    `json:"status"`
+	Status int `json:"status"`
 }
 
 func UpdateGroupStatus(c *gin.Context) {
+	id := c.Param("id")
+	if id == "" {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": "invalid parameter",
+		})
+		return
+	}
 	req := UpdateGroupStatusRequest{}
 	err := json.NewDecoder(c.Request.Body).Decode(&req)
-	if err != nil || req.ID == "" {
+	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"success": false,
 			"message": "invalid parameter",
 		})
 		return
 	}
-	err = model.UpdateGroupStatus(req.ID, req.Status)
+	err = model.UpdateGroupStatus(id, req.Status)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"success": false,