mirror of
https://github.com/Buriburizaem0n/nezha_domains.git
synced 2026-02-04 12:40:07 +00:00
⚡️ refactor: 小幅提升性能
This commit is contained in:
@@ -22,8 +22,6 @@ type AuthorizeOption struct {
|
||||
|
||||
func Authorize(opt AuthorizeOption) func(*gin.Context) {
|
||||
return func(c *gin.Context) {
|
||||
token, err := c.Cookie(dao.Conf.Site.CookieName)
|
||||
token = strings.TrimSpace(token)
|
||||
var code uint64 = http.StatusForbidden
|
||||
if opt.Guest {
|
||||
code = http.StatusBadRequest
|
||||
@@ -35,18 +33,21 @@ func Authorize(opt AuthorizeOption) func(*gin.Context) {
|
||||
Link: opt.Redirect,
|
||||
Btn: opt.Btn,
|
||||
}
|
||||
if token != "" {
|
||||
|
||||
}
|
||||
var isLogin bool
|
||||
var u model.User
|
||||
err = dao.DB.Where("token = ?", token).First(&u).Error
|
||||
if err == nil {
|
||||
isLogin = u.TokenExpired.After(time.Now())
|
||||
}
|
||||
if isLogin {
|
||||
c.Set(model.CtxKeyAuthorizedUser, &u)
|
||||
|
||||
token, _ := c.Cookie(dao.Conf.Site.CookieName)
|
||||
token = strings.TrimSpace(token)
|
||||
if token != "" {
|
||||
var u model.User
|
||||
if err := dao.DB.Where("token = ?", token).First(&u).Error; err == nil {
|
||||
isLogin = u.TokenExpired.After(time.Now())
|
||||
}
|
||||
if isLogin {
|
||||
c.Set(model.CtxKeyAuthorizedUser, &u)
|
||||
}
|
||||
}
|
||||
|
||||
// 已登录且只能游客访问
|
||||
if isLogin && opt.Guest {
|
||||
ShowErrorPage(c, commonErr, opt.IsPage)
|
||||
|
||||
Reference in New Issue
Block a user