Added support for balancer class. (#16)

Intreecom · Nov 25, 2024 · 0f59891 · 0f59891
1 parent 152acf9
commit 0f59891
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/consts.rs b/src/consts.rs
@@ -23,3 +23,4 @@ pub const DEFAULT_LB_ALGORITHM: &str = "least-connections";
 pub const DEFAULT_LB_BALANCER_TYPE: &str = "lb11";
 
 pub const FINALIZER_NAME: &str = "robotlb/finalizer";
+pub const ROBOTLB_LB_CLASS: &str = "robotlb";
diff --git a/src/main.rs b/src/main.rs
@@ -134,6 +134,17 @@ pub async fn reconcile_service(
         return Err(RobotLBError::SkipService);
     }
 
+    let lb_type = svc
+        .spec
+        .as_ref()
+        .and_then(|s| s.load_balancer_class.as_ref())
+        .map(String::as_str)
+        .unwrap_or(consts::ROBOTLB_LB_CLASS);
+    if lb_type != consts::ROBOTLB_LB_CLASS {
+        tracing::debug!("Load balancer class is not robotlb. Skipping...");
+        return Err(RobotLBError::SkipService);
+    }
+
     tracing::info!("Starting service reconcilation");
 
     let lb = LoadBalancer::try_from_svc(&svc, &context)?;
Original file line number	Diff line number	Diff line change
Expand Up		@@ -23,3 +23,4 @@ pub const DEFAULT_LB_ALGORITHM: &str = "least-connections";
		pub const DEFAULT_LB_BALANCER_TYPE: &str = "lb11";

		pub const FINALIZER_NAME: &str = "robotlb/finalizer";
		pub const ROBOTLB_LB_CLASS: &str = "robotlb";