feat: update method name, add method AddTokenSign

This commit is contained in:
weloe
2023-10-31 23:01:47 +08:00
parent 3ac000db86
commit ddcf703835
3 changed files with 8 additions and 3 deletions

View File

@@ -214,7 +214,7 @@ func (e *Enforcer) LoginByModel(id string, loginModel *model.Login, ctx ctx.Cont
if session = e.GetSession(id); session == nil { if session = e.GetSession(id); session == nil {
session = model.NewSession("0", "account-session", id) session = model.NewSession("0", "account-session", id)
} }
session.AddTokenSign(&model.TokenSign{ session.AddDistinctValueTokenSign(&model.TokenSign{
Value: tokenValue, Value: tokenValue,
Device: device, Device: device,
}) })

View File

@@ -440,11 +440,11 @@ func TestEnforcer_JsonAdapter(t *testing.T) {
t.Fatalf("NewEnforcer() failed: %v", err) t.Fatalf("NewEnforcer() failed: %v", err)
} }
newSession := model.NewSession("1", "2", "3") newSession := model.NewSession("1", "2", "3")
newSession.AddTokenSign(&model.TokenSign{ newSession.AddDistinctValueTokenSign(&model.TokenSign{
Value: "2", Value: "2",
Device: "device", Device: "device",
}) })
newSession.AddTokenSign(&model.TokenSign{ newSession.AddDistinctValueTokenSign(&model.TokenSign{
Value: "3", Value: "3",
Device: "device", Device: "device",
}) })

View File

@@ -93,6 +93,11 @@ func (s *Session) GetTokenSign(tokenValue string) *TokenSign {
// AddTokenSign add TokenSign // AddTokenSign add TokenSign
func (s *Session) AddTokenSign(sign *TokenSign) { func (s *Session) AddTokenSign(sign *TokenSign) {
s.TokenSignList = append(s.TokenSignList, sign)
}
// AddDistinctValueTokenSign check tokenValue ,then add TokenSign
func (s *Session) AddDistinctValueTokenSign(sign *TokenSign) {
if s.GetTokenSign(sign.Value) != nil { if s.GetTokenSign(sign.Value) != nil {
return return
} }