From 182f2a921f8ea45d1027aab1b2af1a204d057de5 Mon Sep 17 00:00:00 2001
From: 3AceShowHand <jinl1037@hotmail.com>
Date: Fri, 27 May 2022 17:35:11 +0800
Subject: [PATCH] handle heartbeat with stopping.

---
 cdc/scheduler/internal/tp/agent.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/cdc/scheduler/internal/tp/agent.go b/cdc/scheduler/internal/tp/agent.go
index 6da983a2935..ba3a44a7565 100644
--- 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(),