From 7efe2647c13f1b38002428705142c959e43a7874 Mon Sep 17 00:00:00 2001
From: MuZhou233 <muzhou233@outlook.com>
Date: Wed, 25 Sep 2024 18:13:02 +0800
Subject: [PATCH] fix: fix needRefreshToken logic

---
 wrapper.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/wrapper.go b/wrapper.go
index bd31e79..dc7e365 100644
--- 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
 	}
 }