Implemented locality weighted load balancing ( tests)

allegro · Jan 17, 2024 · 3498959 · 3498959
1 parent 918ccd5
commit 3498959
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/...main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/infrastructure/ControlPlaneConfig.kt b/...main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/infrastructure/ControlPlaneConfig.kt
@@ -75,7 +75,7 @@ class ControlPlaneConfig {
     ): ControlPlane =
         ControlPlane.builder(properties, meterRegistry)
             .withMetrics(metrics)
-            .withCurrentZone(localDatacenter(consulProperties))
+            .withCurrentZone(localDatacenter(consulProperties, properties))
             .withEnvoyHttpFilters(envoyHttpFilters)
             .build(globalStateChanges.combined())
 
@@ -115,11 +115,12 @@ class ControlPlaneConfig {
     fun localClusterStateChanges(
         consulServiceChanges: ConsulServiceChanges,
         consulProperties: ConsulProperties,
-        transformers: List<ServiceInstancesTransformer>
+        transformers: List<ServiceInstancesTransformer>,
+        envoyControlProperties: EnvoyControlProperties
     ): LocalClusterStateChanges = ConsulLocalClusterStateChanges(
         consulServiceChanges,
         Locality.LOCAL,
-        localDatacenter(consulProperties),
+        localDatacenter(consulProperties, envoyControlProperties),
         transformers
     )
 
@@ -173,7 +174,9 @@ class ControlPlaneConfig {
         return EnvoyHttpFilters.defaultFilters(properties.envoy.snapshot)
     }
 
-    fun localDatacenter(properties: ConsulProperties): String {
+    fun localDatacenter(properties: ConsulProperties, envoyControlProperties: EnvoyControlProperties): String {
+        logger.info("LoadBalancing priorities: ${envoyControlProperties.envoy
+            .snapshot.loadBalancing.priorities.zonePriorities}")
         return (ConsulClient(properties.host, properties.port).agentSelf.value?.config?.datacenter ?: "local")
             .also { logger.info("Resolved local datacenter name: $it") }
     }