avoid redudant scheduler

Signed-off-by: husharp <jinhao.hu@pingcap.com>
tikv · Jun 5, 2024 · a312a07 · a312a07
1 parent 301fabb
commit a312a07
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/tools/pd-simulator/main.go b/tools/pd-simulator/main.go
@@ -154,6 +154,8 @@ func simStart(pdAddr, statusAddress string, simCase string, simConfig *sc.SimCon
 	tick := time.NewTicker(tickInterval)
 	defer tick.Stop()
 	sc := make(chan os.Signal, 1)
+	// halt scheduling
+	simulator.ChooseToHaltPDSchedule(true)
 	signal.Notify(sc,
 		syscall.SIGHUP,
 		syscall.SIGINT,

diff --git a/tools/pd-simulator/simulator/client.go b/tools/pd-simulator/simulator/client.go
@@ -16,6 +16,7 @@ package simulator
 
 import (
 	"context"
+	"strconv"
 	"strings"
 	"sync"
 	"time"
@@ -366,3 +367,9 @@ func (c *client) requestHeader() *pdpb.RequestHeader {
 		ClusterId: c.clusterID,
 	}
 }
+
+func ChooseToHaltPDSchedule(halt bool) {
+	PDHTTPClient.SetConfig(context.Background(), map[string]any{
+		"schedule.halt-scheduling": strconv.FormatBool(halt),
+	})
+}
diff --git a/tools/pd-simulator/simulator/node.go b/tools/pd-simulator/simulator/node.go
@@ -172,6 +172,8 @@ func (n *Node) stepTask() {
 	}
 }
 
+var schedulerCheck sync.Once
+
 func (n *Node) stepHeartBeat() {
 	config := n.raftEngine.storeConfig
 
@@ -182,6 +184,7 @@ func (n *Node) stepHeartBeat() {
 	period = uint64(config.RaftStore.RegionHeartBeatInterval.Duration / config.SimTickInterval.Duration)
 	if n.tick%period == 0 {
 		n.regionHeartBeat()
+		schedulerCheck.Do(func() { ChooseToHaltPDSchedule(false) })
 	}
 }