fix: too many tokens

openimsdk · Oct 28, 2024 · afe9b8f · afe9b8f
1 parent bdbb4d4
commit afe9b8f
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/pkg/common/db/cache/token.go b/pkg/common/db/cache/token.go
@@ -27,7 +27,7 @@ import (
 
 const (
 	chatToken       = "CHAT_UID_TOKEN_STATUS:"
-	userMaxTokenNum = 20
+	userMaxTokenNum = 10
 )
 
 type TokenInterface interface {
@@ -96,15 +96,19 @@ func (t *TokenCacheRedis) SetTokenExpire(ctx context.Context, userID string, tok
 		}
 	}
 	var sorted bool
+	var index int
 	for i := len(mm) - len(fields); i > userMaxTokenNum; i-- {
 		if !sorted {
 			sorted = true
 			sort.Sort(ts)
 		}
-		fields = append(fields, ts[i].Token)
+		fields = append(fields, ts[index].Token)
+		index++
 	}
-	if err := t.rdb.HDel(ctx, key, fields...).Err(); err != nil {
-		return errs.Wrap(err)
+	if len(fields) > 0 {
+		if err := t.rdb.HDel(ctx, key, fields...).Err(); err != nil {
+			return errs.Wrap(err)
+		}
 	}
 	return nil
 }