package/token-go
1
0
Fork 0
mirror of https://github.com/weloe/token-go.git synced 2025-10-05 07:26:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix set cookie error, lint error and update default token name to ''Tokengo'

Browse Source
This commit is contained in:
weloe
2023-06-06 05:32:25 +08:00
parent f5939dce57
commit a9d04d1e72
4 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
enforcer_internal_api.go
View File

@@ -5,6 +5,7 @@ import (
"github.com/weloe/token-go/constant"
"github.com/weloe/token-go/ctx"
"github.com/weloe/token-go/model"
"math"
"strconv"
)
@@ -56,10 +57,20 @@ func (e *Enforcer) ResponseToken(tokenValue string, loginModel *model.Login, ctx
// set token to cookie
if tokenConfig.IsReadCookie {
cookieTimeout := tokenConfig.Timeout
if loginModel.IsLastingCookie {
var cookieTimeout int64
if !loginModel.IsLastingCookie {
cookieTimeout = -1
} else {
if loginModel.Timeout != 0 {
cookieTimeout = loginModel.Timeout
} else {
cookieTimeout = tokenConfig.Timeout
}
if cookieTimeout == constant.NeverExpire {
cookieTimeout = math.MaxInt64
}
}
// add cookie use tokenConfig.CookieConfig
ctx.Response().AddCookie(tokenConfig.TokenName,
tokenValue,
@@ -71,6 +82,7 @@ func (e *Enforcer) ResponseToken(tokenValue string, loginModel *model.Login, ctx
// set token to header
if loginModel.IsWriteHeader {
ctx.Response().SetHeader(tokenConfig.TokenName, tokenValue)
ctx.Response().AddHeader(constant.AccessControlExposeHeaders, tokenConfig.TokenName)
}
return nil