br: cancel context when timer stop (#56948) (#57006)

ref #53480
pingcap · Oct 30, 2024 · 1838458 · 1838458
1 parent 1c3d94c
commit 1838458
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/br/pkg/backup/BUILD.bazel b/br/pkg/backup/BUILD.bazel
@@ -68,7 +68,7 @@ go_test(
     embed = [":backup"],
     flaky = True,
     race = "on",
-    shard_count = 14,
+    shard_count = 15,
     deps = [
         "//br/pkg/conn",
         "//br/pkg/gluetidb/mock",

diff --git a/br/pkg/backup/store.go b/br/pkg/backup/store.go
@@ -79,6 +79,7 @@ func (trecv *timeoutRecv) Refresh() {
 func (trecv *timeoutRecv) Stop() {
 	close(trecv.refresh)
 	trecv.wg.Wait()
+	trecv.cancel(nil)
 }
 
 var TimeoutOneResponse = time.Hour

diff --git a/br/pkg/backup/store_test.go b/br/pkg/backup/store_test.go
@@ -96,3 +96,13 @@ func TestTimeoutRecvCancel(t *testing.T) {
 	cancel()
 	trecv.wg.Wait()
 }
+
+func TestTimeoutRecvCanceled(t *testing.T) {
+	ctx := context.Background()
+	cctx, cancel := context.WithCancel(ctx)
+	defer cancel()
+
+	tctx, trecv := StartTimeoutRecv(cctx, time.Hour)
+	trecv.Stop()
+	require.Equal(t, "context canceled", tctx.Err().Error())
+}