u
This commit is contained in:
1 parent
1fd0697aee
commit
9af3234672
13 files changed
+358
-26
No files matched your search
@@ -2,14 +2,18 @@ package middleware
|
||||
|
||||
import (
|
||||
"base-framework/pkg/router"
|
||||
"log"
|
||||
"time"
|
||||
)
|
||||
|
||||
func Logger() router.HandlerFunc {
|
||||
return func(c *router.Context) {
|
||||
start := time.Now()
|
||||
// 执行后续中间件和处理器
|
||||
c.Next()
|
||||
log.Printf("[%s] %s in %v", c.Request.Method, c.Request.URL.Path, time.Since(start))
|
||||
elapsed := time.Since(start)
|
||||
// 即使业务中断,这里也能执行,打印耗时
|
||||
path := c.Request.URL.Path
|
||||
method := c.Request.Method
|
||||
println(method, path, "耗时:", elapsed.String())
|
||||
}
|
||||
}
|
||||
@@ -24,7 +24,6 @@ func CreateToken(orgID, userID string) (string, error) {
|
||||
StandardClaims: jwt.StandardClaims{
|
||||
ExpiresAt: expireTime,
|
||||
IssuedAt: time.Now().Unix(),
|
||||
Issuer: "your-app-name", // 可以自定义
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,100 @@
|
||||
package response
|
||||
|
||||
import (
|
||||
"base-framework/pkg/router"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// 统一响应结构体
|
||||
type Result struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"message"`
|
||||
Data interface{} `json:"data,omitempty"`
|
||||
}
|
||||
|
||||
// 常量状态码
|
||||
const (
|
||||
CodeSuccess = "0000"
|
||||
CodeFail = "9999"
|
||||
)
|
||||
|
||||
// code 对应默认提示
|
||||
var codeMessages = map[string]string{
|
||||
CodeSuccess: "请求成功",
|
||||
CodeFail: "请求失败",
|
||||
|
||||
"1001": "缺少用户ID",
|
||||
"1002": "未授权",
|
||||
"1003": "无权限访问",
|
||||
"1004": "资源不存在",
|
||||
|
||||
"9000": "系统内部错误",
|
||||
}
|
||||
|
||||
// 兜底提示
|
||||
const defaultMessage = "未知错误"
|
||||
|
||||
// Builder 链式构造器
|
||||
type Builder struct {
|
||||
c *router.Context
|
||||
result Result
|
||||
}
|
||||
|
||||
// Success 构造成功响应
|
||||
func Success(c *router.Context) *Builder {
|
||||
return &Builder{
|
||||
c: c,
|
||||
result: Result{
|
||||
Code: CodeSuccess,
|
||||
Message: getMessage(CodeSuccess),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// Error 构造失败响应
|
||||
func Error(c *router.Context) *Builder {
|
||||
return &Builder{
|
||||
c: c,
|
||||
result: Result{
|
||||
Code: CodeFail,
|
||||
Message: getMessage(CodeFail),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// Code 设置状态码(自动填充默认提示,除非后面手动改)
|
||||
func (b *Builder) Code(code string) *Builder {
|
||||
b.result.Code = code
|
||||
if b.result.Message == "" || b.result.Message == getMessage(b.result.Code) {
|
||||
b.result.Message = getMessage(code)
|
||||
}
|
||||
return b
|
||||
}
|
||||
|
||||
// Message 设置提示消息(覆盖默认的)
|
||||
func (b *Builder) Message(msg string) *Builder {
|
||||
b.result.Message = msg
|
||||
return b
|
||||
}
|
||||
|
||||
// Data 设置返回数据
|
||||
func (b *Builder) Data(data interface{}) *Builder {
|
||||
b.result.Data = data
|
||||
return b
|
||||
}
|
||||
|
||||
// Send 发送响应
|
||||
func (b *Builder) Send() {
|
||||
if b.result.Data == nil {
|
||||
b.result.Data = ""
|
||||
}
|
||||
b.c.JSON(http.StatusOK, b.result)
|
||||
}
|
||||
|
||||
// getMessage 获取状态码对应提示,兜底 defaultMessage
|
||||
func getMessage(code string) string {
|
||||
if msg, ok := codeMessages[code]; ok {
|
||||
return msg
|
||||
}
|
||||
return defaultMessage
|
||||
}
|
||||
Reference in new issue
Block a user