Sun
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package middleware
|
|
|
|
import (
|
|
"sun-panel/api/api_v1/common/apiReturn"
|
|
"sun-panel/global"
|
|
"sun-panel/models"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func LoginInterceptor(c *gin.Context) {
|
|
|
|
// 继续执行后续的操作，再回来
|
|
// c.Next()
|
|
|
|
// 获得token
|
|
cToken := c.GetHeader("token")
|
|
|
|
// 没有token信息视为未登录
|
|
if cToken == "" {
|
|
apiReturn.ErrorCode(c, 1000, global.Lang.Get("login.err_not_login"), nil)
|
|
c.Abort() // 终止执行后续的操作，一般配合return使用
|
|
return
|
|
}
|
|
|
|
token := ""
|
|
{
|
|
var ok bool
|
|
token, ok = global.CUserToken.Get(cToken)
|
|
// 可能已经安全退出或者很久没有使用已过期
|
|
if !ok || token == "" {
|
|
apiReturn.ErrorCode(c, 1001, global.Lang.Get("login.err_not_login"), nil)
|
|
c.Abort() // 终止执行后续的操作，一般配合return使用
|
|
return
|
|
}
|
|
}
|
|
|
|
// 直接返回缓存的用户信息
|
|
if userInfo, success := global.UserToken.Get(token); success {
|
|
c.Set("userInfo", userInfo)
|
|
return
|
|
}
|
|
|
|
global.Logger.Debug("准备查询数据库的用户资料", token)
|
|
|
|
mUser := models.User{}
|
|
// 去库中查询是否存在该用户；否则返回错误
|
|
if info, err := mUser.GetUserInfoByToken(token); err != nil || info.Token == "" || info.ID == 0 {
|
|
apiReturn.ErrorCode(c, 1001, global.Lang.Get("login.err_token_expire"), nil)
|
|
c.Abort()
|
|
return
|
|
} else {
|
|
// 通过 设置当前用户信息
|
|
global.UserToken.SetDefault(info.Token, info)
|
|
global.CUserToken.SetDefault(cToken, token)
|
|
c.Set("userInfo", info)
|
|
}
|
|
|
|
}
|
|
|
|
// 不验证缓存直接验证库省去没有缓存每次都要手动登录的问题
|
|
func LoginInterceptorDev(c *gin.Context) {
|
|
|
|
// 获得token
|
|
token := c.GetHeader("token")
|
|
mUser := models.User{}
|
|
|
|
// 去库中查询是否存在该用户；否则返回错误
|
|
if info, err := mUser.GetUserInfoByToken(token); err != nil || info.ID == 0 {
|
|
apiReturn.ErrorCode(c, 1001, global.Lang.Get("login.err_token_expire"), nil)
|
|
c.Abort()
|
|
return
|
|
} else {
|
|
// 通过
|
|
// 设置当前用户信息
|
|
c.Set("userInfo", info)
|
|
}
|
|
}
|