handle heartbeat with stopping.

pingcap · May 27, 2022 · 182f2a9 · 182f2a9
1 parent d670016
commit 182f2a9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cdc/scheduler/internal/tp/agent.go b/cdc/scheduler/internal/tp/agent.go
@@ -241,9 +241,8 @@ func (a *agent) collectTableStatus() []schedulepb.TableStatus {
 func (a *agent) handleMessageHeartbeat() *schedulepb.Message {
 	tables := a.collectTableStatus()
 	response := &schedulepb.HeartbeatResponse{
-		Tables: tables,
-		// todo (Ling Jin): how to set `IsStopping`
-		IsStopping: false,
+		Tables:     tables,
+		IsStopping: a.stopping,
 	}
 	return &schedulepb.Message{
 		Header:            a.newMessageHeader(),