Backend method for the load balancer

ankitects · dae · Mar 26, 2025 · Mar 12, 2025 · Mar 12, 2025 · Mar 14, 2025
commit 5ea2cec3a9be50d68c15fd1b80ede63de4d21ddb
diff --git a/proto/anki/collection.proto b/proto/anki/collection.proto
@@ -19,7 +19,7 @@ service CollectionService {
   rpc MergeUndoEntries(generic.UInt32) returns (OpChanges);
   rpc LatestProgress(generic.Empty) returns (Progress);
   rpc SetWantsAbort(generic.Empty) returns (generic.Empty);
-  rpc ClearStudyQueues(generic.Empty) returns (generic.Empty);
+  rpc SetLoadBalancerEnabled(generic.Bool) returns (generic.Empty);
 }
 
 // Implicitly includes any of the above methods that are not listed in the

diff --git a/pylib/anki/collection.py b/pylib/anki/collection.py
@@ -986,8 +986,7 @@ def _get_enable_load_balancer(self) -> bool:
         return self.get_config_bool(Config.Bool.LOAD_BALANCER_ENABLED)
 
     def _set_enable_load_balancer(self, value: bool) -> None:
-        self._backend.clear_study_queues()
-        self.set_config_bool(Config.Bool.LOAD_BALANCER_ENABLED, value)
+        self._backend.set_load_balancer_enabled(value)
 
     load_balancer_enabled = property(
         fget=_get_enable_load_balancer, fset=_set_enable_load_balancer

diff --git a/rslib/src/collection/service.rs b/rslib/src/collection/service.rs
@@ -3,6 +3,7 @@
 use anki_proto::generic;
 
 use crate::collection::Collection;
+use crate::config::BoolKey;
 use crate::error;
 use crate::progress::progress_to_proto;
 
@@ -50,8 +51,9 @@ impl crate::services::CollectionService for Collection {
         Ok(())
     }
 
-    fn clear_study_queues(&mut self) -> error::Result<()> {
+    fn set_load_balancer_enabled(&mut self, input: generic::Bool) -> error::Result<()> {
         self.clear_study_queues();
+        self.set_config_bool(BoolKey::LoadBalancerEnabled, input.val, false)?;
         Ok(())
     }
 }