Golang快速开发框架——新增统一消息处理模块（七）

package mess

import "strings"

// RetMessage 主消息结构体
type RetMessage struct {
    Code string      `json:"code"`
    Msg  string      `json:"msg"`
    Data interface{} `json:"data"`
}

// ReturnReqMess 常规消息生成函数
func ReturnReqMess(code, msg string, data interface{}) *RetMessage {
    mess := RetMessage{
        Code: code,
        Msg:  msg,
        Data: data,
    }
    return &mess
}

// ResErrMess 异常消息生成函数
// msg 可以增加多个提示信息
func ResErrMess(code string, msg ...string) *RetMessage {
    return &RetMessage{code, strings.Join(msg, ""), nil}
}

// ResMessOK 成功消息返回通用函数
// data 返回的结构对象
func ResMessOK(data interface{}) *RetMessage {
    return &RetMessage{"0", "成功", data}
}

// 测试网关
func testRouter(r *gin.Engine) {
    // 测试500中间件地址
    r.GET("/demo", func(c *gin.Context) {
        a := c.DefaultQuery("a", "1")
        i, _ := strconv.Atoi(a)
        // 当a参数是默认值1时会报错
        x := 0 / (1 - i)
        log.Println(x)
        c.JSON(http.StatusInternalServerError, fmt.Errorf("测试异常"))
    })

    // 测试-成功消息，常规赋值方式
    r.GET("/mess", func(c *gin.Context) {
        data := []string{
            "测试用户", "12岁", "男",
        }
        // 返回成功消息
        c.JSON(http.StatusOK, mess.ReturnReqMess("0", "成功", data))
    })
    // 测试-成功消息
    r.GET("/messOK", func(c *gin.Context) {
        data := []string{
            "测试用户002", "20岁", "女",
        }
        // 返回成功消息
        c.JSON(http.StatusOK, mess.ResMessOK(data))
    })
    // 测试-成功消息，常规赋值方式
    r.GET("/messErr", func(c *gin.Context) {
        // 返回成功消息
        c.JSON(http.StatusOK, mess.ResErrMess("1001011", "常规业务错误：", "用户名或密码错误"))
    })
}