diff --git a/internal/util/grpcclient/client.go b/internal/util/grpcclient/client.go
index 9a60d59150..71a031fa79 100644
--- a/internal/util/grpcclient/client.go
+++ b/internal/util/grpcclient/client.go
@@ -106,8 +106,10 @@ type ClientBase[T interface {
 	encryption bool
 	addr       atomic.String
 	// conn                   *grpc.ClientConn
-	grpcClientMtx          sync.RWMutex
-	role                   string
+	grpcClientMtx sync.RWMutex
+	role          string
+	isNode        bool // pre-calculated is node flag
+
 	ClientMaxSendSize      int
 	ClientMaxRecvSize      int
 	CompressionEnabled     bool
@@ -159,6 +161,12 @@ func NewClientBase[T interface {
 // SetRole sets role of client
 func (c *ClientBase[T]) SetRole(role string) {
 	c.role = role
+	if strings.HasPrefix(role, typeutil.DataNodeRole) ||
+		strings.HasPrefix(role, typeutil.IndexNodeRole) ||
+		strings.HasPrefix(role, typeutil.QueryNodeRole) ||
+		strings.HasPrefix(role, typeutil.ProxyRole) {
+		c.isNode = true
+	}
 }
 
 // GetRole returns role of client
@@ -417,8 +425,7 @@ func (c *ClientBase[T]) checkGrpcErr(ctx context.Context, err error) (needRetry,
 }
 
 func (c *ClientBase[T]) checkNodeSessionExist(ctx context.Context) (bool, error) {
-	switch c.GetRole() {
-	case typeutil.DataNodeRole, typeutil.IndexNodeRole, typeutil.QueryNodeRole, typeutil.ProxyRole:
+	if c.isNode {
 		err := c.verifySession(ctx)
 		if errors.Is(err, merr.ErrNodeNotFound) {
 			log.Warn("failed to verify node session", zap.Error(err))
diff --git a/internal/util/grpcclient/client_test.go b/internal/util/grpcclient/client_test.go
index e4a4ee3ea9..fa691e61e7 100644
--- a/internal/util/grpcclient/client_test.go
+++ b/internal/util/grpcclient/client_test.go
@@ -98,6 +98,7 @@ func TestClientBase_NodeSessionNotExist(t *testing.T) {
 	base := ClientBase[*mockClient]{
 		maxCancelError: 10,
 		MaxAttempts:    3,
+		isNode:         true,
 	}
 	base.SetGetAddrFunc(func() (string, error) {
 		return "", errors.New("mocked address error")
@@ -148,6 +149,7 @@ func testCall(t *testing.T, compressed bool) {
 	base := ClientBase[*mockClient]{
 		maxCancelError: 10,
 		MaxAttempts:    3,
+		isNode:         true,
 	}
 	base.CompressionEnabled = compressed
 	initClient := func() {