disttask: refine dispatcher (#45460)

ref #41495
pingcap · Aug 1, 2023 · e5e507a · e5e507a
1 parent 67e8f90
commit e5e507a
Showing 9 changed files with 437 additions and 304 deletions.
diff --git a/disttask/framework/BUILD.bazel b/disttask/framework/BUILD.bazel
@@ -9,7 +9,7 @@ go_test(
     ],
     flaky = True,
     race = "on",
-    shard_count = 9,
+    shard_count = 10,
     deps = [
         "//disttask/framework/dispatcher",
         "//disttask/framework/proto",

diff --git a/disttask/framework/dispatcher/BUILD.bazel b/disttask/framework/dispatcher/BUILD.bazel
@@ -4,6 +4,7 @@ go_library(
     name = "dispatcher",
     srcs = [
         "dispatcher.go",
+        "dispatcher_manager.go",
         "dispatcher_mock.go",
         "register.go",
     ],