close session after restoring tiflash-replica

Signed-off-by: 5kbpers <[email protected]>
pingcap · Mar 17, 2020 · eb6db46 · eb6db46
1 parent 9a98ffc
commit eb6db46
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cmd/restore.go b/cmd/restore.go
@@ -70,10 +70,12 @@ func runRestoreTiflashReplicaCommand(command *cobra.Command) error {
 	if err != nil {
 		return err
 	}
+	defer store.Close()
 	db, err := restore.NewDB(tidbGlue, store)
 	if err != nil {
 		return err
 	}
+	defer db.Close()
 
 	for _, d := range dbs {
 		for _, t := range d.Tables {