Make load balancing configurable

crates/ott-balancer/src/balancer.rs

@@ -196,16 +196,7 @@ pub struct BalancerContext {
 }
 impl BalancerContext {
     pub fn new() -> Self {
-        if BalancerConfig::get().selection_strategy.is_some() {
-            BalancerContext {
-                monolith_selection: BalancerConfig::get()
-                    .selection_strategy
-                    .expect("selection strategy not set"),
-                ..BalancerContext::default()
-            }
-        } else {
-            BalancerContext::default()
-        }
+        Default::default()
     }
 
     #[instrument(skip(self, client), err, fields(client_id = %client.id, room = %client.room))]

crates/ott-balancer/src/lib.rs

@@ -57,6 +57,19 @@ pub async fn run() -> anyhow::Result<()> {
 
     let config = BalancerConfig::get();
 
+    if config.selection_strategy.is_some() {
+        info!("Using selection strategy: {:?}", config.selection_strategy);
+        BalancerContext {
+            monolith_selection: config
+                .selection_strategy
+                .expect("selection strategy not set"),
+            ..BalancerContext::default()
+        }
+    } else {
+        info!("Using default selection strategy");
+        BalancerContext::default()
+    };
+
     let console_layer = if args.console {
         let console_layer = if args.remote_console {
             console_subscriber::ConsoleLayer::builder()