package/go_process_manager
1
0
Fork 0
mirror of https://github.com/lzh-1625/go_process_manager.git synced 2025-10-05 16:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

edit token check

Browse Source
This commit is contained in:
akrike
2025-09-03 22:06:30 +08:00
parent 2e37eae3ca
commit e14df1fd99
1 changed files with 5 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
internal/app/middle/token.go
View File

@@ -1,7 +1,6 @@
package middle package middle
import ( import (
"errors"
"slices" "slices"
"strings" "strings"
@@ -46,36 +45,19 @@ func CheckToken() gin.HandlerFunc {
return strings.HasPrefix(c.Request.URL.Path, s) return strings.HasPrefix(c.Request.URL.Path, s)
}) { }) {
var token string var token string
if c.Request.Header.Get("token") != "" { if c.Request.Header.Get("Authorization") != "" {
token = c.Request.Header.Get("token") token = strings.TrimPrefix(c.Request.Header.Get("Authorization"), "bearer ")
} else { } else {
token = c.Query("token") token = c.Query("token")
} }
if _, err := utils.VerifyToken(token); err != nil { if mc, err := utils.VerifyToken(token); err != nil {
rErr(c, -2, "token校验失败", err) rErr(c, -2, "token校验失败", err)
return return
}
if username, err := getUser(c); err != nil {
rErr(c, -1, "无法获取user信息", err)
} else { } else {
c.Set(eum.CtxUserName, username) c.Set(eum.CtxUserName, mc.Username)
c.Set(eum.CtxRole, repository.UserRepository.GetUserByName(username).Role) c.Set(eum.CtxRole, repository.UserRepository.GetUserByName(mc.Username).Role)
} }
} }
c.Next() c.Next()
} }
} }
func getUser(ctx *gin.Context) (string, error) {
var token string
if ctx.Request.Header.Get("token") != "" {
token = ctx.Request.Header.Get("token")
} else {
token = ctx.Query("token")
}
if mc, err := utils.VerifyToken(token); err == nil && mc != nil {
return mc.Username, nil
} else {
return "", errors.Join(errors.New("用户信息获取失败"), err)
}
}