104 lines
3.0 KiB
Go
104 lines
3.0 KiB
Go
// Package response ...
|
|
//
|
|
// Description : response ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2022-06-25 20:52
|
|
package response
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.zhangdeman.cn/zhangdeman/exception"
|
|
"git.zhangdeman.cn/zhangdeman/gin/define"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
const (
|
|
// hasSendResponseFlag 已经发送响应数据的标识, 一旦识别到, 重复调用不会执行
|
|
hasSendResponseFlag = "GIN_PKG_HAS_SEND_RESPONSE"
|
|
)
|
|
|
|
// Success 成功的响应
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 22:37 2022/6/25
|
|
func Success(ctx *gin.Context, data interface{}) {
|
|
successException := exception.NewSuccess(data)
|
|
Send(ctx, successException.GetCode(), successException.GetHttpCode(), successException.GetData())
|
|
}
|
|
|
|
// Send 基础的发送数据
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 22:40 2022/6/25
|
|
func Send(ctx *gin.Context, code interface{}, httpCode int, data interface{}) {
|
|
if ctx.GetBool(hasSendResponseFlag) {
|
|
// 已经发送过数据, 后面在发送数据, 不执行
|
|
return
|
|
}
|
|
// 设置数据已发送的标识
|
|
defer ctx.Set(hasSendResponseFlag, true)
|
|
finishRequestTime := time.Now().UnixNano()
|
|
responseData := map[string]interface{}{
|
|
define.ResponseCodeField: code,
|
|
define.ResponseMessageField: exception.GetMessage(code),
|
|
define.ResponseTraceIDField: ctx.GetString(define.TraceIDField),
|
|
define.ResponseRequestIDField: ctx.GetString(define.RequestIDField),
|
|
define.ResponseDataField: data,
|
|
define.HandleRequestCostField: (finishRequestTime - ctx.GetInt64(define.StartRequestTimeField)) / 1e6,
|
|
}
|
|
// 记录响应数据
|
|
recordData, _ := json.Marshal(responseData)
|
|
// 记录完成时间
|
|
ctx.Set(define.FinishRequestTimeField, finishRequestTime)
|
|
ctx.Set(define.RecordResponseDataField, string(recordData))
|
|
responseException := exception.New(code, httpCode, responseData)
|
|
ctx.JSON(responseException.GetHttpCode(), responseException.GetData())
|
|
}
|
|
|
|
// SendWithStatusOK ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 22:41 2022/6/25
|
|
func SendWithStatusOK(ctx *gin.Context, code interface{}, data map[string]interface{}) {
|
|
Send(ctx, code, http.StatusOK, data)
|
|
}
|
|
|
|
// SendWithException 使用exception发送数据
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 13:08 2022/6/26
|
|
func SendWithException(ctx *gin.Context, e exception.IException, data interface{}) {
|
|
if nil == e {
|
|
e = exception.NewSuccess(data)
|
|
}
|
|
if nil != e.GetData() {
|
|
Send(ctx, e.GetCode(), e.GetHttpCode(), e.GetData())
|
|
} else {
|
|
Send(ctx, e.GetCode(), e.GetHttpCode(), data)
|
|
}
|
|
}
|
|
|
|
// JSON ctx.JSON 的平替, 增加了数据是否已相应的标识
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 14:51 2023/2/15
|
|
func JSON(ctx *gin.Context, httpCode int, data interface{}) {
|
|
if ctx.GetBool(hasSendResponseFlag) {
|
|
// 已经发送过数据, 后面在发送数据, 不执行
|
|
return
|
|
}
|
|
// 设置数据已发送的标识
|
|
defer ctx.Set(hasSendResponseFlag, true)
|
|
ctx.JSON(httpCode, data)
|
|
}
|