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 {
session = model.NewSession("0", "account-session", id)
}
session.AddTokenSign(&model.TokenSign{
session.AddDistinctValueTokenSign(&model.TokenSign{
Value: tokenValue,
Device: device,
})

View File

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

View File

@@ -93,6 +93,11 @@ func (s *Session) GetTokenSign(tokenValue string) *TokenSign {
// AddTokenSign add 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 {
return
}