tk/Ocelot
1
0
Fork 0
mirror of https://github.com/nsnail/Ocelot.git synced 2025-06-19 07:18:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

tests for error handling on load balancing middleware

Browse Source
This commit is contained in:
Tom Gardham-Pallister
2017-02-06 20:22:09 +00:00
parent c3e60ac08a
commit a4495b8fa9
2 changed files with 83 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/Ocelot/LoadBalancer/Middleware/LoadBalancingMiddleware.cs
View File

@ -31,16 +31,18 @@ namespace Ocelot.LoadBalancer.Middleware
{
_logger.LogDebug("started calling load balancing middleware");
var loadBalancer = _loadBalancerHouse.Get(DownstreamRoute.ReRoute.LoadBalancerKey);
if(loadBalancer.IsError)
var getLoadBalancer = _loadBalancerHouse.Get(DownstreamRoute.ReRoute.LoadBalancerKey);
if(getLoadBalancer.IsError)
{
//set errors and return
SetPipelineError(getLoadBalancer.Errors);
return;
}
var hostAndPort = await loadBalancer.Data.Lease();
var hostAndPort = await getLoadBalancer.Data.Lease();
if(hostAndPort.IsError)
{
//set errors and return
{
SetPipelineError(hostAndPort.Errors);
return;
}
SetHostAndPortForThisRequest(hostAndPort.Data);
@ -51,11 +53,11 @@ namespace Ocelot.LoadBalancer.Middleware
{
await _next.Invoke(context);
loadBalancer.Data.Release(hostAndPort.Data);
getLoadBalancer.Data.Release(hostAndPort.Data);
}
catch (Exception)
{
loadBalancer.Data.Release(hostAndPort.Data);
getLoadBalancer.Data.Release(hostAndPort.Data);
_logger.LogDebug("error calling next middleware, exception will be thrown to global handler");
throw;
}