From 6b41879387b97a4ba1bcab6e87e03b3b1ff39325 Mon Sep 17 00:00:00 2001 From: Monet Lee Date: Wed, 12 Jun 2024 15:33:55 +0800 Subject: [PATCH] fix uncorrect contents. --- internal/rpc/auth/auth.go | 31 ++++++++++++++++--------------- 1 file changed, 16 insertions(+), 15 deletions(-) diff --git a/internal/rpc/auth/auth.go b/internal/rpc/auth/auth.go index ce883c3c0..486611e71 100644 --- a/internal/rpc/auth/auth.go +++ b/internal/rpc/auth/auth.go @@ -172,22 +172,23 @@ func (s *authServer) forceKickOff(ctx context.Context, userID string, platformID if err != nil { log.ZError(ctx, "forceKickOff", err, "kickReq", kickReq) } - m, err := s.authDatabase.GetTokensWithoutError(ctx, userID, int(platformID)) - if err != nil && err != redis.Nil { - return err - } - if m == nil { - return errs.New("token map is empty").Wrap() - } - for k := range m { - m[k] = constant.KickedToken - log.ZDebug(ctx, "set token map is ", "token map", m, "userID", - userID, "token", k) + } + + m, err := s.authDatabase.GetTokensWithoutError(ctx, userID, int(platformID)) + if err != nil && err != redis.Nil { + return err + } + if m == nil { + return errs.New("token map is empty").Wrap() + } + for k := range m { + m[k] = constant.KickedToken + log.ZDebug(ctx, "set token map is ", "token map", m, "userID", + userID, "token", k) - err = s.authDatabase.SetTokenMapByUidPid(ctx, userID, int(platformID), m) - if err != nil { - return err - } + err = s.authDatabase.SetTokenMapByUidPid(ctx, userID, int(platformID), m) + if err != nil { + return err } } return nil