Merge #129263

129263: rac2: implement wait for eval r=sumeerbhola a=kvoli Implement `RangeController.WaitForEval(..)`, which will be used to wait on a configurable number of streams to have available tokens. Note that the functionality is currently incomplete without replica state management, done in `HandleRaftReady`. Resolves: #128910 Release note: None Co-authored-by: Austen McClernon <[email protected]>
cockroachdb · Aug 21, 2024 · c8c495d · c8c495d
2 parents 77c22fb + 727ae59
commit c8c495d
Show file tree

Hide file tree

Showing 6 changed files with 1,244 additions and 3 deletions.
diff --git a/pkg/kv/kvserver/kvflowcontrol/rac2/BUILD.bazel b/pkg/kv/kvserver/kvflowcontrol/rac2/BUILD.bazel
@@ -13,6 +13,7 @@ go_library(
     deps = [
         "//pkg/kv/kvserver/kvflowcontrol",
         "//pkg/raft/raftpb",
+        "//pkg/raft/tracker",
         "//pkg/roachpb",
         "//pkg/settings/cluster",
         "//pkg/util/admission/admissionpb",
@@ -28,15 +29,19 @@ go_test(
     name = "rac2_test",
     srcs = [
         "priority_test.go",
+        "range_controller_test.go",
         "token_counter_test.go",
     ],
     data = glob(["testdata/**"]),
     embed = [":rac2"],
     deps = [
         "//pkg/kv/kvserver/kvflowcontrol",
         "//pkg/raft/raftpb",
+        "//pkg/raft/tracker",
+        "//pkg/roachpb",
         "//pkg/settings/cluster",
         "//pkg/util/admission/admissionpb",
+        "//pkg/util/humanizeutil",
         "//pkg/util/leaktest",
         "//pkg/util/log",
         "//pkg/util/syncutil",