forked from lxh/go-wxhelper
37 lines
935 B
Go
37 lines
935 B
Go
package middleware
|
||
|
||
import (
|
||
"gitee.ltd/lxh/logger/log"
|
||
"github.com/gin-gonic/gin"
|
||
"net/http"
|
||
"strings"
|
||
"wechat-robot/pkg/auth"
|
||
"wechat-robot/pkg/response"
|
||
)
|
||
|
||
// AuthorizeToken
|
||
// @description: 验证OAuth2生成的Token
|
||
// @return gin.HandlerFunc
|
||
func AuthorizeToken() gin.HandlerFunc {
|
||
return func(ctx *gin.Context) {
|
||
// 判断有无token
|
||
tokenStr := ctx.GetHeader("Authorization")
|
||
if tokenStr == "" || !strings.HasPrefix(tokenStr, "Bearer ") {
|
||
response.New(ctx).SetMsg("请先登录").Fail()
|
||
ctx.Abort()
|
||
return
|
||
}
|
||
// 先取出用户Token
|
||
token, err := auth.OAuthServer.ValidationBearerToken(ctx.Request)
|
||
if err != nil {
|
||
log.Errorf("获取Token失败，错误：%s", err.Error())
|
||
response.New(ctx).SetCode(http.StatusUnauthorized).SetMsg("登录已失效或已在其他地方登录").Fail()
|
||
ctx.Abort()
|
||
return
|
||
}
|
||
|
||
// 判断通过，允许放行
|
||
ctx.Set("userId", token.GetUserID())
|
||
}
|
||
}
|