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()
|
|||
|
|
}
|
|||
|
|
}
|