u
This commit is contained in:
1 parent
6849579652
commit
d60be4a75b
24 files changed
+463
-179
No files matched your search
@@ -0,0 +1,39 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
response "base-go-v2/internal/reponse"
|
||||
"base-go-v2/internal/router"
|
||||
"base-go-v2/internal/utils/jwtx"
|
||||
)
|
||||
|
||||
func AuthMiddleware() router.HandlerFunc {
|
||||
return func(c *router.Context) error {
|
||||
token := c.Header("token")
|
||||
if token == "" {
|
||||
c.Abort()
|
||||
return response.Fail(c).
|
||||
Code(response.HttpCode.Unauthorized).
|
||||
Message("未提供 Token").
|
||||
Send()
|
||||
}
|
||||
|
||||
tokenVerify := jwtx.VerifyToken(token)
|
||||
|
||||
// Access Token 有效
|
||||
if tokenVerify.IsValid {
|
||||
c.Set("user_id", tokenVerify.Claims.UserID)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Access Token 失效 → 返回刷新提示
|
||||
if !tokenVerify.IsValid {
|
||||
c.Abort()
|
||||
return response.Fail(c).
|
||||
Code(response.HttpCode.RefreshToken).
|
||||
Message("Access Token 已过期,请刷新").
|
||||
Send()
|
||||
}
|
||||
|
||||
return c.Next()
|
||||
}
|
||||
}
|
||||
@@ -28,7 +28,7 @@ func ErrorMiddleware() router.HandlerFunc {
|
||||
// be 是 nil 指针类型
|
||||
if errors.As(err, &be) {
|
||||
// 业务错误,不记录日志
|
||||
return response.Error(c).Message(err.Error()).Send()
|
||||
return response.Fail(c).Message(err.Error()).Send()
|
||||
}
|
||||
}
|
||||
|
||||
@@ -40,6 +40,6 @@ func ErrorMiddleware() router.HandlerFunc {
|
||||
logx.String("error", err.Error()),
|
||||
logx.String("stack", stack),
|
||||
)
|
||||
return response.Error(c).Message("服务器内部错误").Send()
|
||||
return response.Fail(c).Message("服务器内部错误").Send()
|
||||
}
|
||||
}
|
||||
@@ -21,7 +21,7 @@ func RecoveryMiddleware() router.HandlerFunc {
|
||||
)
|
||||
|
||||
// 返回统一 JSON 响应
|
||||
_ = response.Error(c).Message("系统 Panic").Send()
|
||||
_ = response.Fail(c).Message("系统 Panic").Send()
|
||||
}
|
||||
}()
|
||||
|
||||
|
||||
Reference in new issue
Block a user