HI-VPN/hi-server
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update: Successful authentication records the identifier.

Browse Source
This commit is contained in:
EUForest 2025-11-06 15:33:14 +08:00
parent de6661cc16
commit 066f5d6538
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/middleware/authMiddleware.go
View File

@ -42,8 +42,11 @@ func AuthMiddleware(svc *svc.ServiceContext) func(c *gin.Context) {
} }
loginType := "" loginType := ""
if claims["LoginType"] != nil { if claims["CtxLoginType"] != nil {
loginType = claims["LoginType"].(string) loginType = claims["CtxLoginType"].(string)
}
if claims["identifier"] != nil {
ctx = context.WithValue(ctx, constant.CtxKeyIdentifier, claims["identifier"].(string))
} }
// get user id from token // get user id from token
userId := int64(claims["UserId"].(float64)) userId := int64(claims["UserId"].(float64))
@ -82,9 +85,10 @@ func AuthMiddleware(svc *svc.ServiceContext) func(c *gin.Context) {
c.Abort() c.Abort()
return return
} }
ctx = context.WithValue(ctx, constant.LoginType, loginType) ctx = context.WithValue(ctx, constant.CtxLoginType, loginType)
ctx = context.WithValue(ctx, constant.CtxKeyUser, userInfo) ctx = context.WithValue(ctx, constant.CtxKeyUser, userInfo)
ctx = context.WithValue(ctx, constant.CtxKeySessionID, sessionId) ctx = context.WithValue(ctx, constant.CtxKeySessionID, sessionId)
c.Request = c.Request.WithContext(ctx) c.Request = c.Request.WithContext(ctx)
c.Next() c.Next()
} }