fix: fix needRefreshToken logic

tuihub · Sep 25, 2024 · 7efe264 · 7efe264
1 parent a2f6162
commit 7efe264
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wrapper.go b/wrapper.go
@@ -105,11 +105,11 @@ func (s *serviceWrapper) EnablePorter(ctx context.Context, req *pb.EnablePorterR
 	if resp, err := s.LibrarianPorterServiceServer.EnablePorter(ctx, req); isUnimplementedError(err) {
 		return &pb.EnablePorterResponse{
 			StatusMessage:    "",
-			NeedRefreshToken: needRefreshToken || resp.GetNeedRefreshToken(),
+			NeedRefreshToken: needRefreshToken,
 			EnablesSummary:   nil,
 		}, nil
 	} else {
-		resp.NeedRefreshToken = needRefreshToken
+		resp.NeedRefreshToken = needRefreshToken || resp.GetNeedRefreshToken()
 		return resp, err
 	}
 }