46 lines
940 B
Go
46 lines
940 B
Go
package tool
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
// CheckContext 检查上下文是否取消
|
|
func CheckContext(ctx context.Context) error {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err() // 返回取消原因
|
|
default:
|
|
return nil // 上下文仍然有效
|
|
}
|
|
}
|
|
|
|
// Success 成功响应
|
|
// @param httpMsg http.ResponseWriter
|
|
// @param data 返回的数据
|
|
func Success(httpMsg http.ResponseWriter, data any) {
|
|
ret := map[string]interface{}{
|
|
"code": "200",
|
|
"msg": "成功",
|
|
"data": data,
|
|
}
|
|
json.NewEncoder(httpMsg).Encode(ret)
|
|
}
|
|
|
|
// Error 错误响应
|
|
// @param httpMsg http.ResponseWriter
|
|
// @param msg 错误信息
|
|
// @param code 错误码
|
|
func Error(httpMsg http.ResponseWriter, msg string, code int) {
|
|
fmt.Println("错误：" + msg)
|
|
codeStr := strconv.FormatInt(int64(code), 10)
|
|
ret := map[string]interface{}{
|
|
"code": codeStr,
|
|
"msg": msg,
|
|
}
|
|
json.NewEncoder(httpMsg).Encode(ret)
|
|
}
|