fix switch mode bug in restore (#473)

pingcap · Aug 26, 2020 · 3234e8d · 3234e8d
1 parent 64271cf
commit 3234e8d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/restore/client.go b/pkg/restore/client.go
@@ -714,6 +714,14 @@ func (rc *Client) SwitchToImportMode(ctx context.Context) {
 	// so we need ping tikv in less than 10 minute
 	go func() {
 		tick := time.NewTicker(rc.switchModeInterval)
+		defer tick.Stop()
+
+		// [important!] switch tikv mode into import at the beginning
+		log.Info("switch to import mode at beginning")
+		err := rc.switchTiKVMode(ctx, import_sstpb.SwitchMode_Import)
+		if err != nil {
+			log.Warn("switch to import mode failed", zap.Error(err))
+		}
 
 		for {
 			select {