feat: 重构响应处理，添加通用响应函数以简化代码

atsf_server/controller/agent.go

@@ -3,8 +3,6 @@ package controller
 import (
 	"atsflare/model"
 	"atsflare/service"
-	"encoding/json"
-	"net/http"
 
 	"github.com/gin-gonic/gin"
 )
@@ -21,13 +19,11 @@ import (
 // @Router /api/agent/nodes/register [post]
 func AgentRegister(c *gin.Context) {
 	var payload service.AgentNodePayload
-	if err := json.NewDecoder(c.Request.Body).Decode(&payload); err != nil {
-		c.JSON(http.StatusBadRequest, gin.H{
-			"success": false,
-			"message": "无效的参数",
-		})
+	if err := decodeJSONBody(c.Request.Body, &payload); err != nil {
+		respondBadRequest(c, "")
 		return
 	}
+
 	var (
 		result *service.AgentRegistrationResponse
 		err    error
@@ -38,17 +34,10 @@ func AgentRegister(c *gin.Context) {
 		result, err = service.RegisterNodeWithDiscovery(payload)
 	}
 	if err != nil {
-		c.JSON(http.StatusOK, gin.H{
-			"success": false,
-			"message": err.Error(),
-		})
+		respondFailure(c, err.Error())
 		return
 	}
-	c.JSON(http.StatusOK, gin.H{
-		"success": true,
-		"message": "",
-		"data":    result,
-	})
+	respondSuccess(c, result)
 }
 
 // AgentHeartbeat godoc
@@ -63,33 +52,23 @@ func AgentRegister(c *gin.Context) {
 // @Router /api/agent/nodes/heartbeat [post]
 func AgentHeartbeat(c *gin.Context) {
 	var payload service.AgentNodePayload
-	if err := json.NewDecoder(c.Request.Body).Decode(&payload); err != nil {
-		c.JSON(http.StatusBadRequest, gin.H{
-			"success": false,
-			"message": "无效的参数",
-		})
+	if err := decodeJSONBody(c.Request.Body, &payload); err != nil {
+		respondBadRequest(c, "")
 		return
 	}
+
 	authNode, ok := c.Get("agent_node")
 	if !ok {
-		c.JSON(http.StatusUnauthorized, gin.H{
-			"success": false,
-			"message": "无权进行此操作，Agent Token 无效",
-		})
+		respondUnauthorized(c, "鏃犳潈杩涜姝ゆ搷浣滐紝Agent Token 鏃犳晥")
 		return
 	}
+
 	node, err := service.HeartbeatNode(authNode.(*model.Node), payload)
 	if err != nil {
-		c.JSON(http.StatusOK, gin.H{
-			"success": false,
-			"message": err.Error(),
-		})
+		respondFailure(c, err.Error())
 		return
 	}
-	c.JSON(http.StatusOK, gin.H{
-		"success":        true,
-		"message":        "",
-		"data":           node.Node,
+	respondSuccessWithExtras(c, node.Node, gin.H{
 		"agent_settings": node.AgentSettings,
 		"active_config":  node.ActiveConfig,
 	})
@@ -105,17 +84,10 @@ func AgentHeartbeat(c *gin.Context) {
 func AgentGetActiveConfig(c *gin.Context) {
 	config, err := service.GetActiveConfigForAgent()
 	if err != nil {
-		c.JSON(http.StatusOK, gin.H{
-			"success": false,
-			"message": "当前没有激活版本",
-		})
+		respondFailure(c, "褰撳墠娌℃湁婵€娲荤増鏈?")
 		return
 	}
-	c.JSON(http.StatusOK, gin.H{
-		"success": true,
-		"message": "",
-		"data":    config,
-	})
+	respondSuccess(c, config)
 }
 
 // AgentReportApplyLog godoc
@@ -130,30 +102,21 @@ func AgentGetActiveConfig(c *gin.Context) {
 // @Router /api/agent/apply-logs [post]
 func AgentReportApplyLog(c *gin.Context) {
 	var payload service.ApplyLogPayload
-	if err := json.NewDecoder(c.Request.Body).Decode(&payload); err != nil {
-		c.JSON(http.StatusBadRequest, gin.H{
-			"success": false,
-			"message": "无效的参数",
-		})
+	if err := decodeJSONBody(c.Request.Body, &payload); err != nil {
+		respondBadRequest(c, "")
 		return
 	}
-	authNode, ok := c.Get("agent_node")
-	if ok {
+
+	if authNode, ok := c.Get("agent_node"); ok {
 		payload.NodeID = authNode.(*model.Node).NodeID
 	}
+
 	log, err := service.ReportApplyLog(payload)
 	if err != nil {
-		c.JSON(http.StatusOK, gin.H{
-			"success": false,
-			"message": err.Error(),
-		})
+		respondFailure(c, err.Error())
 		return
 	}
-	c.JSON(http.StatusOK, gin.H{
-		"success": true,
-		"message": "",
-		"data":    log,
-	})
+	respondSuccess(c, log)
 }
 
 // GetNodes godoc
@@ -166,17 +129,10 @@ func AgentReportApplyLog(c *gin.Context) {
 func GetNodes(c *gin.Context) {
 	nodes, err := service.ListNodeViews()
 	if err != nil {
-		c.JSON(http.StatusOK, gin.H{
-			"success": false,
-			"message": err.Error(),
-		})
+		respondFailure(c, err.Error())
 		return
 	}
-	c.JSON(http.StatusOK, gin.H{
-		"success": true,
-		"message": "",
-		"data":    nodes,
-	})
+	respondSuccess(c, nodes)
 }
 
 // GetApplyLogs godoc
@@ -190,15 +146,8 @@ func GetNodes(c *gin.Context) {
 func GetApplyLogs(c *gin.Context) {
 	logs, err := service.ListApplyLogs(c.Query("node_id"))
 	if err != nil {
-		c.JSON(http.StatusOK, gin.H{
-			"success": false,
-			"message": err.Error(),
-		})
+		respondFailure(c, err.Error())
 		return
 	}
-	c.JSON(http.StatusOK, gin.H{
-		"success": true,
-		"message": "",
-		"data":    logs,
-	})
+	respondSuccess(c, logs)
 }