71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
package middleware
|
||
|
||
import (
|
||
"Lee-WineList/core"
|
||
"Lee-WineList/oauth2"
|
||
"git.echol.cn/loser/logger/log"
|
||
"github.com/gin-gonic/gin"
|
||
"net/http"
|
||
"strings"
|
||
)
|
||
|
||
// AuthorizeToken 验证OAuth2生成的Token
|
||
func AuthorizeToken() gin.HandlerFunc {
|
||
return func(ctx *gin.Context) {
|
||
// 判断有无token
|
||
tokenStr := ctx.GetHeader("Authorization")
|
||
if tokenStr == "" || !strings.HasPrefix(tokenStr, "Bearer ") {
|
||
core.R(ctx).FailWithMessageAndCode("请先登录", http.StatusUnauthorized)
|
||
ctx.Abort()
|
||
return
|
||
}
|
||
// 先取出用户Token
|
||
token, err := oauth2.OAuthServer.ValidationBearerToken(ctx.Request)
|
||
if err != nil {
|
||
log.Errorf("获取Token失败，错误：%s", err.Error())
|
||
core.R(ctx).FailWithMessageAndCode("登录已失效或已在其他地方登录", http.StatusUnauthorized)
|
||
ctx.Abort()
|
||
return
|
||
}
|
||
// 把UserId字段反序列化成map
|
||
//info := make(map[string]string)
|
||
//if err = json.Unmarshal([]byte(token.GetUserID()), &info); err != nil {
|
||
// core.R(ctx).FailWithMessageAndCode("Token数据解析失败", http.StatusUnauthorized)
|
||
// ctx.Abort()
|
||
// return
|
||
//}
|
||
//go func() {
|
||
// // 异步记录用户在线情况，十分钟没操作就是不在线了
|
||
// rdsKey := "oauth:online:" + info["userId"]
|
||
// global.RedisConn.Set(context.Background(), rdsKey, "1", 10*time.Minute)
|
||
//}()
|
||
// 判断通过，允许放行
|
||
ctx.Request.Header.Add("userId", token.GetUserID())
|
||
ctx.Set("userId", token.GetUserID())
|
||
ctx.Next()
|
||
}
|
||
}
|
||
|
||
// DealLoginUserId 处理登录用户Id
|
||
func DealLoginUserId() gin.HandlerFunc {
|
||
return func(ctx *gin.Context) {
|
||
// 判断有无token
|
||
tokenStr := ctx.GetHeader("Authorization")
|
||
if tokenStr == "" || !strings.HasPrefix(tokenStr, "Bearer ") {
|
||
//ctx.Next()
|
||
return
|
||
}
|
||
// 先取出用户Token
|
||
token, err := oauth2.OAuthServer.ValidationBearerToken(ctx.Request)
|
||
if err != nil {
|
||
//ctx.Next()
|
||
return
|
||
}
|
||
//log.Debugf("本次请求存在正常Token: %v", tokenStr)
|
||
// 判断通过，允许放行
|
||
ctx.Request.Header.Add("userId", token.GetUserID())
|
||
ctx.Set("userId", token.GetUserID())
|
||
//ctx.Next()
|
||
}
|
||
}
|