package middleware
import (
	"net/http"
	"github.com/gin-gonic/gin"
)
// ResponseData 表示統(tǒng)一響應(yīng)的JSON格式
type ResponseData struct {
	Code    int         `json:"code"`    // 狀態(tài)碼
	Message string      `json:"message"` // 響應(yīng)消息
	Data    interface{} `json:"data"`    // 響應(yīng)數(shù)據(jù)
}

// ErrorResponse 是一個輔助函數(shù)，用于創(chuàng)建錯誤響應(yīng)
// 參數(shù)：
//   c *gin.Context：Gin上下文對象，用于處理HTTP請求和響應(yīng)
//   code int：HTTP狀態(tài)碼，表示請求處理的結(jié)果
//   message string：響應(yīng)消息，用于描述響應(yīng)的錯誤信息或提示信息
func ErrorResponse(c *gin.Context, code int, message string) {
	c.JSON(code, ResponseData{
		Code:    code,
		Message: message,
		Data:    nil,
	})
}

// SuccessResponse 是一個輔助函數(shù)，用于創(chuàng)建成功響應(yīng)
// 參數(shù)：
//   c *gin.Context：Gin上下文對象，用于處理HTTP請求和響應(yīng)
//   code int：HTTP狀態(tài)碼，表示請求處理的結(jié)果
//   data interface{}：響應(yīng)數(shù)據(jù)，用于描述請求處理成功后返回的具體數(shù)據(jù)
func SuccessResponse(c *gin.Context, code int, data interface{}) {
	c.JSON(code, ResponseData{
		Code:    code,
		Message: "成功",
		Data:    data,
	})
}

// UnifiedResponseMiddleware 是處理統(tǒng)一HTTP響應(yīng)格式的中間件
// 該中間件將在將響應(yīng)發(fā)送給客戶端之前攔截響應(yīng)，并根據(jù)你指定的格式進行格式化。
// 返回值：
//   gin.HandlerFunc：Gin中間件處理函數(shù)
func UnifiedResponseMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Next()
		// 檢查是否在處理請求時發(fā)生了錯誤
		// 如果發(fā)生了錯誤，通過ErrorResponse函數(shù)創(chuàng)建一個錯誤響應(yīng)，并返回給客戶端
		if len(c.Errors) > 0 {
			err := c.Errors.Last()
			ErrorResponse(c, http.StatusInternalServerError, err.Error())
			return
		}
		// 檢查是否設(shè)置了響應(yīng)狀態(tài)碼
		// 如果未設(shè)置響應(yīng)狀態(tài)碼，默認將狀態(tài)碼設(shè)置為200（OK）
		if c.Writer.Status() == 0 {
			c.Writer.WriteHeader(http.StatusOK)
		}
		// 如果沒有錯誤，則格式化響應(yīng)
		// 檢查是否設(shè)置了"response_data"鍵的值，如果有，則調(diào)用SuccessResponse函數(shù)創(chuàng)建一個成功響應(yīng)，并返回給客戶端
		if c.Writer.Status() >= http.StatusOK && c.Writer.Status() < http.StatusMultipleChoices {
			data, exists := c.Get("response_data")
			if exists {
				SuccessResponse(c, c.Writer.Status(), data)
				return
			}
		}
	}
}

package middleware
import (
	"net/http"
	"github.com/gin-gonic/gin"
)
type ResponseData struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data"`
}
// ErrorResponse是一個輔助函數(shù)，用于創(chuàng)建錯誤響應(yīng)
func ErrorResponse(c *gin.Context, code int, message string) {
	c.JSON(code, ResponseData{
		Code:    code,
		Message: message,
		Data:    nil,
	})
}
// SuccessResponse是一個輔助函數(shù)，用于創(chuàng)建成功響應(yīng)
func SuccessResponse(c *gin.Context, code int, data interface{}) {
	c.JSON(code, ResponseData{
		Code:    code,
		Message: "成功",
		Data:    data,
	})
}
// UnifiedResponseMiddleware是處理統(tǒng)一HTTP響應(yīng)格式的中間件
func UnifiedResponseMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Next()
		// 檢查是否在處理請求時發(fā)生了錯誤
		if len(c.Errors) > 0 {
			err := c.Errors.Last()
			ErrorResponse(c, http.StatusInternalServerError, err.Error())
			return
		}
		// 檢查是否設(shè)置了響應(yīng)狀態(tài)碼
		if c.Writer.Status() == 0 {
			c.Writer.WriteHeader(http.StatusOK)
		}
		// 如果沒有錯誤，則格式化響應(yīng)
		if c.Writer.Status() >= http.StatusOK && c.Writer.Status() < http.StatusMultipleChoices {
			data, exists := c.Get("response_data")
			if exists {
				SuccessResponse(c, c.Writer.Status(), data)
				return
			}
		}
	}
}

package main
import (
	"github.com/gin-gonic/gin"
	"your-app/middleware"
)
func main() {
	r := gin.Default()
	// 使用統(tǒng)一響應(yīng)中間件
	r.Use(middleware.UnifiedResponseMiddleware())
	// 在這里定義你的路由和處理程序
	r.Run(":8080")
}