address the comment

Signed-off-by: yisaer <[email protected]>
pingcap · Aug 24, 2022 · 4451752 · 4451752
1 parent 5d48e3d
commit 4451752
Showing 1 changed file with 4 additions and 13 deletions.
diff --git a/executor/plan_replayer.go b/executor/plan_replayer.go
@@ -25,6 +25,7 @@ import (
 	"io"
 	"os"
 	"path/filepath"
+	"strconv"
 	"strings"
 	"time"
 
@@ -326,19 +327,10 @@ func dumpTiFlashReplica(ctx sessionctx.Context, zw *zip.Writer, pairs map[tableN
 			continue
 		}
 		if t.Meta().TiFlashReplica != nil && t.Meta().TiFlashReplica.Count > 0 {
-			sql := fmt.Sprintf("SELECT TABLE_SCHEMA,TABLE_NAME,REPLICA_COUNT FROM INFORMATION_SCHEMA.TIFLASH_REPLICA WHERE TABLE_SCHEMA ='%s' AND TABLE_NAME='%s' AND REPLICA_COUNT > 0",
-				dbName.L, tableName.L)
-			recordSet, err := ctx.(sqlexec.SQLExecutor).ExecuteInternal(context.Background(), sql)
-			if err != nil {
-				return err
-			}
-			sRows, err := resultSetToStringSlice(context.Background(), recordSet)
-			if err != nil {
-				return err
-			}
-			for _, row := range sRows {
-				fmt.Fprintf(bf, "%s\n", strings.Join(row, "\t"))
+			row := []string{
+				pair.DBName, pair.TableName, strconv.FormatUint(t.Meta().TiFlashReplica.Count, 10),
 			}
+			fmt.Fprintf(bf, "%s\n", strings.Join(row, "\t"))
 		}
 	}
 	return nil
@@ -633,7 +625,6 @@ func (e *PlanReplayerLoadExec) Next(ctx context.Context, req *chunk.Chunk) error
 func loadSetTiFlashReplica(ctx sessionctx.Context, z *zip.Reader) error {
 	for _, zipFile := range z.File {
 		if strings.Compare(zipFile.Name, tiFlashReplicasFile) == 0 {
-			//varMap := make(map[string]string)
 			v, err := zipFile.Open()
 			if err != nil {
 				return errors.AddStack(err)