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

feat: refactor kickout, replaced

Browse Source
This commit is contained in:
weloe
2023-10-22 05:28:00 +08:00
parent bb1845fd3c
commit 272189517b
2 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
enforcer_manager_api.go
View File

@@ -1,6 +1,7 @@
package token_go
import (
"container/list"
"fmt"
"github.com/weloe/token-go/constant"
"github.com/weloe/token-go/model"
@@ -8,11 +9,16 @@ import (
)
// Replaced replace other user
func (e *Enforcer) Replaced(id string, device string) error {
func (e *Enforcer) Replaced(id string, device ...string) error {
var err error
if session := e.GetSession(id); session != nil {
// get by login device
tokenSignList := session.GetFilterTokenSign(device)
var tokenSignList *list.List
if len(device) == 0 {
tokenSignList = session.GetTokenSignListCopy()
} else {
// get by login device
tokenSignList = session.GetFilterTokenSign(device[0])
}
// sign account replaced
for element := tokenSignList.Front(); element != nil; element = element.Next() {
if tokenSign, ok := element.Value.(*model.TokenSign); ok {
@@ -42,11 +48,17 @@ func (e *Enforcer) Replaced(id string, device string) error {
}
// Kickout kickout user
func (e *Enforcer) Kickout(id string, device string) error {
func (e *Enforcer) Kickout(id string, device ...string) error {
session := e.GetSession(id)
if session != nil {
// get by login device
tokenSignList := session.GetFilterTokenSign(device)
var tokenSignList *list.List
if len(device) == 0 {
tokenSignList = session.GetTokenSignListCopy()
} else {
// get by login device
tokenSignList = session.GetFilterTokenSign(device[0])
}
// sign account kicked
for element := tokenSignList.Front(); element != nil; element = element.Next() {
if tokenSign, ok := element.Value.(*model.TokenSign); ok {