Add 'Name' property to ILoadBalancer, for future use in LoadBalancerF…

…actory.
ThreeMammals · Mar 15, 2020 · 789782c · 789782c
1 parent 347ea72
commit 789782c
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 0 deletions.
diff --git a/src/Ocelot/LoadBalancer/LoadBalancers/CookieStickySessions.cs b/src/Ocelot/LoadBalancer/LoadBalancers/CookieStickySessions.cs
@@ -84,5 +84,7 @@ public async Task<Response<ServiceHostAndPort>> Lease(DownstreamContext context)
         public void Release(ServiceHostAndPort hostAndPort)
         {
         }
+
+        public string Name => GetType().Name;
     }
 }
diff --git a/src/Ocelot/LoadBalancer/LoadBalancers/ILoadBalancer.cs b/src/Ocelot/LoadBalancer/LoadBalancers/ILoadBalancer.cs
@@ -10,5 +10,7 @@ public interface ILoadBalancer
         Task<Response<ServiceHostAndPort>> Lease(DownstreamContext context);
 
         void Release(ServiceHostAndPort hostAndPort);
+
+        string Name { get; }
     }
 }
diff --git a/src/Ocelot/LoadBalancer/LoadBalancers/LeastConnection.cs b/src/Ocelot/LoadBalancer/LoadBalancers/LeastConnection.cs
@@ -71,6 +71,8 @@ public void Release(ServiceHostAndPort hostAndPort)
             }
         }
 
+        public string Name => GetType().Name;
+
         private Lease AddConnection(Lease lease)
         {
             return new Lease(lease.HostAndPort, lease.Connections + 1);

diff --git a/src/Ocelot/LoadBalancer/LoadBalancers/NoLoadBalancer.cs b/src/Ocelot/LoadBalancer/LoadBalancers/NoLoadBalancer.cs
@@ -33,5 +33,7 @@ public async Task<Response<ServiceHostAndPort>> Lease(DownstreamContext downstre
         public void Release(ServiceHostAndPort hostAndPort)
         {
         }
+
+        public string Name => GetType().Name;
     }
 }
diff --git a/src/Ocelot/LoadBalancer/LoadBalancers/RoundRobin.cs b/src/Ocelot/LoadBalancer/LoadBalancers/RoundRobin.cs
@@ -38,5 +38,7 @@ public async Task<Response<ServiceHostAndPort>> Lease(DownstreamContext downstre
         public void Release(ServiceHostAndPort hostAndPort)
         {
         }
+
+        public string Name => GetType().Name;
     }
 }
diff --git a/test/Ocelot.UnitTests/LoadBalancer/LoadBalancerHouseTests.cs b/test/Ocelot.UnitTests/LoadBalancer/LoadBalancerHouseTests.cs
@@ -164,6 +164,8 @@ public void Release(ServiceHostAndPort hostAndPort)
             {
                 throw new NotImplementedException();
             }
+
+            public string Name => GetType().Name;
         }
 
         private class FakeRoundRobinLoadBalancer : ILoadBalancer
@@ -177,6 +179,8 @@ public void Release(ServiceHostAndPort hostAndPort)
             {
                 throw new NotImplementedException();
             }
+
+            public string Name => GetType().Name;
         }
     }
 }