152 lines
3.0 KiB
Go
152 lines
3.0 KiB
Go
package controller
|
|
|
|
import (
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"fmt"
|
|
"mh-server/lib/status"
|
|
"strconv"
|
|
|
|
"github.com/codinl/go-logger"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type RespRet struct {
|
|
Code int `json:"code"`
|
|
Msg string `json:"msg"`
|
|
Data interface{} `json:"data"`
|
|
}
|
|
|
|
func RespJson(c *gin.Context, code int, data interface{}) {
|
|
result := RespRet{
|
|
Code: code,
|
|
Msg: status.StatusMsg(code),
|
|
Data: data,
|
|
}
|
|
body, err := json.Marshal(result)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
c.Data(status.OK, "application/json; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
func RespOK(c *gin.Context, data interface{}) {
|
|
result := RespRet{
|
|
Code: 200,
|
|
Msg: "成功",
|
|
Data: data,
|
|
}
|
|
body, err := json.Marshal(result)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
c.Data(status.OK, "application/json; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
func RespError(c *gin.Context, data ...string) {
|
|
var respMsg string
|
|
if len(data) > 0 {
|
|
respMsg = data[0]
|
|
} else {
|
|
respMsg = "服务器开小差了"
|
|
}
|
|
result := RespRet{
|
|
Code: 500,
|
|
Msg: respMsg,
|
|
Data: "",
|
|
}
|
|
body, err := json.Marshal(result)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
c.Data(status.OK, "application/json; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
func Resp(c *gin.Context, code int, msg string, data interface{}) {
|
|
result := RespRet{
|
|
Code: code,
|
|
Msg: msg,
|
|
Data: data,
|
|
}
|
|
body, err := json.Marshal(result)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
c.Data(status.OK, "application/json; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
func RespNotice(c *gin.Context, code string, msg string) {
|
|
result := struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"message"`
|
|
}{
|
|
Code: code,
|
|
Message: msg,
|
|
}
|
|
body, err := json.Marshal(result)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
c.Data(status.OK, "application/json; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
func RespBodyXML(c *gin.Context, data interface{}) {
|
|
body, err := xml.Marshal(data)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
fmt.Println("回复消息:", string(body))
|
|
c.Data(status.OK, "application/xml; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
func Toast(c *gin.Context, data interface{}) {
|
|
result := struct {
|
|
Code int `json:"code"`
|
|
Msg interface{} `json:"msg"`
|
|
Data interface{} `json:"data"`
|
|
}{
|
|
Code: status.ToastErr,
|
|
Msg: data,
|
|
Data: data,
|
|
}
|
|
body, err := json.Marshal(result)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
body = []byte{}
|
|
}
|
|
c.Data(status.OK, "application/json; charset=utf-8", body)
|
|
c.Abort()
|
|
}
|
|
|
|
// 比较 Code 是否为 200
|
|
func CompareCode(code interface{}) bool {
|
|
switch v := code.(type) {
|
|
case int:
|
|
return v == status.OK
|
|
case string:
|
|
// 尝试将字符串转换为整数
|
|
if intValue, err := strconv.Atoi(v); err == nil {
|
|
return intValue == status.OK
|
|
}
|
|
// 如果转换失败，比较字符串值
|
|
return v == strconv.Itoa(status.OK)
|
|
case float64:
|
|
return int(v) == status.OK
|
|
default:
|
|
// 其他类型视为不相等
|
|
return false
|
|
}
|
|
}
|