tk/Ocelot
1
0
Fork 0
mirror of https://github.com/nsnail/Ocelot.git synced 2025-04-22 10:12:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

RoundRobin loadblancer cause config error https://github.com/TomPallister/Ocelot/issues/103

Browse Source
This commit is contained in:
geffzhang 2017-06-03 22:32:47 +08:00
parent 54db27100d
commit 02162dd7a6
3 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Ocelot/LoadBalancer/LoadBalancers/LoadBalancerFactory.cs
View File

@ -19,7 +19,7 @@ namespace Ocelot.LoadBalancer.LoadBalancers
switch (reRoute.LoadBalancer)
{
case "RoundRobin":
return new RoundRobinLoadBalancer(await serviceProvider.Get());
return new RoundRobinLoadBalancer(async () => await serviceProvider.Get());
case "LeastConnection":
return new LeastConnectionLoadBalancer(async () => await serviceProvider.Get(), reRoute.ServiceProviderConfiguraion.ServiceName);
default:

12
src/Ocelot/LoadBalancer/LoadBalancers/RoundRobinLoadBalancer.cs
View File

@ -2,27 +2,31 @@
using System.Threading.Tasks;
using Ocelot.Responses;
using Ocelot.Values;
using System;
namespace Ocelot.LoadBalancer.LoadBalancers
{
public class RoundRobinLoadBalancer : ILoadBalancer
{
private readonly List<Service> _services;
private readonly Func<Task<List<Service>>> _services;
private int _last;
public RoundRobinLoadBalancer(List<Service> services)
public RoundRobinLoadBalancer(Func<Task<List<Service>>> services)
{
_services = services;
}
public async Task<Response<HostAndPort>> Lease()
{
if (_last >= _services.Count)
var services = await _services.Invoke();
if (_last >= services.Count)
{
_last = 0;
}
var next = await Task.FromResult(_services[_last]);
var next = await Task.FromResult(services[_last]);
_last++;
return new OkResponse<HostAndPort>(next.HostAndPort);
}

3
test/Ocelot.UnitTests/LoadBalancer/RoundRobinTests.cs
View File

@ -6,6 +6,7 @@ using Ocelot.Values;
using Shouldly;
using TestStack.BDDfy;
using Xunit;
using System.Threading.Tasks;
namespace Ocelot.UnitTests.LoadBalancer
{
@ -24,7 +25,7 @@ namespace Ocelot.UnitTests.LoadBalancer
new Service("product", new HostAndPort("127.0.0.1", 5001), string.Empty, string.Empty, new string[0])
};
_roundRobin = new RoundRobinLoadBalancer(_services);
_roundRobin = new RoundRobinLoadBalancer(() => Task.FromResult(_services));
}
[Fact]