br: optimize communication with tikv stores. (pingcap#31722)

Signed-off-by: Gaoming Zhang <zhanggaoming028@gmail.com>
MoCuishle28 · Jun 28, 2022 · 36b55d6 · 36b55d6
1 parent 6a9a6f1
commit 36b55d6
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/br/pkg/restore/client.go b/br/pkg/restore/client.go
@@ -1090,14 +1090,13 @@ func (rc *Client) switchTiKVMode(ctx context.Context, mode import_sstpb.SwitchMo
 			return errors.Trace(err)
 		}
 
-		opt := grpc.WithInsecure()
-		if rc.tlsConf != nil {
-			opt = grpc.WithTransportCredentials(credentials.NewTLS(rc.tlsConf))
-		}
-
 		finalStore := store
 		rc.workerPool.ApplyOnErrorGroup(eg,
 			func() error {
+				opt := grpc.WithInsecure()
+				if rc.tlsConf != nil {
+					opt = grpc.WithTransportCredentials(credentials.NewTLS(rc.tlsConf))
+				}
 				gctx, cancel := context.WithTimeout(ectx, time.Second*5)
 				connection, err := grpc.DialContext(
 					gctx,