Initial commit

controllers/helper.go

@@ -0,0 +1,81 @@
+package controllers
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+
+	"git.beifan.cn/trace-system/backend-go/models"
+)
+
+// GetCurrentUser 从上下文中获取当前用户
+// 如果用户未认证，返回 false 并返回 401 错误
+func GetCurrentUser(ctx *gin.Context) (models.User, bool) {
+	user, exists := ctx.Get("user")
+	if !exists {
+		ctx.JSON(http.StatusUnauthorized, gin.H{
+			"message": "未认证",
+		})
+		return models.User{}, false
+	}
+
+	userModel, ok := user.(models.User)
+	if !ok {
+		ctx.JSON(http.StatusUnauthorized, gin.H{
+			"message": "用户信息无效",
+		})
+		return models.User{}, false
+	}
+
+	return userModel, true
+}
+
+// BindJSON 绑定 JSON 请求体
+// 如果解析失败，返回错误并返回 400 响应
+func BindJSON(ctx *gin.Context, obj interface{}) bool {
+	if err := ctx.ShouldBindJSON(obj); err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{
+			"message": "无效的请求数据",
+			"error":   err.Error(),
+		})
+		return false
+	}
+	return true
+}
+
+// BindQuery 绑定查询参数
+// 如果解析失败，返回错误并返回 400 响应
+func BindQuery(ctx *gin.Context, obj interface{}) bool {
+	if err := ctx.ShouldBindQuery(obj); err != nil {
+		ctx.JSON(http.StatusBadRequest, gin.H{
+			"message": "无效的查询参数",
+			"error":   err.Error(),
+		})
+		return false
+	}
+	return true
+}
+
+// ErrorResponse 返回错误响应
+func ErrorResponse(ctx *gin.Context, status int, message string, err ...error) {
+	response := gin.H{
+		"message": message,
+	}
+	if len(err) > 0 && err[0] != nil {
+		response["error"] = err[0].Error()
+	}
+	ctx.JSON(status, response)
+}
+
+// SuccessResponse 返回成功响应
+func SuccessResponse(ctx *gin.Context, message string, data ...gin.H) {
+	response := gin.H{
+		"message": message,
+	}
+	if len(data) > 0 {
+		for key, value := range data[0] {
+			response[key] = value
+		}
+	}
+	ctx.JSON(http.StatusOK, response)
+}