Add possibility to set Connection to Close ( request.Headers.ConnectionClose = ConnectionClose ) in order to prevent persistent connections.

Author: Bjarte K. Helland

src/Ocelot/Configuration/Creator/ConnectionCloseCreator.cs

@@ -0,0 +1,14 @@
+using Ocelot.Configuration.File;
+
+namespace Ocelot.Configuration.Creator
+{
+    public class ConnectionCloseCreator : IConnectionCloseCreator
+    {
+        public bool Create(bool fileRouteConnectionClose, FileGlobalConfiguration globalConfiguration)
+        {
+            var globalConnectionClose = globalConfiguration.ConnectionClose;
+
+            return fileRouteConnectionClose || globalConnectionClose;
+        }
+    }
+}

src/Ocelot/Configuration/Creator/IConnectionCloseCreator.cs

@@ -0,0 +1,9 @@
+using Ocelot.Configuration.File;
+
+namespace Ocelot.Configuration.Creator
+{
+    public interface IConnectionCloseCreator
+    {
+        bool Create(bool fileRouteConnectionClose, FileGlobalConfiguration globalConfiguration);
+    }
+}

src/Ocelot/Configuration/Creator/RoutesCreator.cs

@@ -26,6 +26,7 @@ public class RoutesCreator : IRoutesCreator
         private readonly IRouteKeyCreator _routeKeyCreator;
         private readonly ISecurityOptionsCreator _securityOptionsCreator;
         private readonly IVersionCreator _versionCreator;
+        private readonly IConnectionCloseCreator _connectionCloseCreator;
 
         public RoutesCreator(
             IClaimsToThingCreator claimsToThingCreator,
@@ -42,7 +43,8 @@ public RoutesCreator(
             ILoadBalancerOptionsCreator loadBalancerOptionsCreator,
             IRouteKeyCreator routeKeyCreator,
             ISecurityOptionsCreator securityOptionsCreator,
-            IVersionCreator versionCreator
+            IVersionCreator versionCreator,
+            IConnectionCloseCreator connectionCloseCreator
             )
         {
             _routeKeyCreator = routeKeyCreator;
@@ -61,6 +63,7 @@ IVersionCreator versionCreator
             _loadBalancerOptionsCreator = loadBalancerOptionsCreator;
             _securityOptionsCreator = securityOptionsCreator;
             _versionCreator = versionCreator;
+            _connectionCloseCreator = connectionCloseCreator;
         }
 
         public List<Route> Create(FileConfiguration fileConfiguration)
@@ -112,6 +115,8 @@ private DownstreamRoute SetUpDownstreamRoute(FileRoute fileRoute, FileGlobalConf
 
             var downstreamHttpVersion = _versionCreator.Create(fileRoute.DownstreamHttpVersion);
 
+            var connectionClose = _connectionCloseCreator.Create(fileRoute.ConnectionClose, globalConfiguration);
+
             var route = new DownstreamRouteBuilder()
                 .WithKey(fileRoute.Key)
                 .WithDownstreamPathTemplate(fileRoute.DownstreamPathTemplate)
@@ -148,6 +153,7 @@ private DownstreamRoute SetUpDownstreamRoute(FileRoute fileRoute, FileGlobalConf
                 .WithSecurityOptions(securityOptions)
                 .WithDownstreamHttpVersion(downstreamHttpVersion)
                 .WithDownStreamHttpMethod(fileRoute.DownstreamHttpMethod)
+                .WithConnectionClose(connectionClose)
                 .Build();
 
             return route;

src/Ocelot/Configuration/DownstreamRoute.cs

@@ -43,7 +43,8 @@ public DownstreamRoute(
             bool dangerousAcceptAnyServerCertificateValidator,
             SecurityOptions securityOptions,
             string downstreamHttpMethod,
-            Version downstreamHttpVersion)
+            Version downstreamHttpVersion, 
+            bool connectionClose)
         {
             DangerousAcceptAnyServerCertificateValidator = dangerousAcceptAnyServerCertificateValidator;
             AddHeadersToDownstream = addHeadersToDownstream;
@@ -79,6 +80,7 @@ public DownstreamRoute(
             SecurityOptions = securityOptions;
             DownstreamHttpMethod = downstreamHttpMethod;
             DownstreamHttpVersion = downstreamHttpVersion;
+            ConnectionClose = connectionClose;
         }
 
         public string Key { get; }
@@ -114,6 +116,7 @@ public DownstreamRoute(
         public bool DangerousAcceptAnyServerCertificateValidator { get; }
         public SecurityOptions SecurityOptions { get; }
         public string DownstreamHttpMethod { get; }
-        public Version DownstreamHttpVersion { get; }
+        public Version DownstreamHttpVersion { get;  }
+        public bool ConnectionClose { get; }
     }
 }

src/Ocelot/Configuration/File/FileGlobalConfiguration.cs

@@ -9,6 +9,7 @@ public FileGlobalConfiguration()
             LoadBalancerOptions = new FileLoadBalancerOptions();
             QoSOptions = new FileQoSOptions();
             HttpHandlerOptions = new FileHttpHandlerOptions();
+            ConnectionClose = false;
         }
 
         public string RequestIdKey { get; set; }
@@ -28,5 +29,7 @@ public FileGlobalConfiguration()
         public FileHttpHandlerOptions HttpHandlerOptions { get; set; }
 
         public string DownstreamHttpVersion { get; set; }
+
+        public bool ConnectionClose { get; set; }
     }
 }

src/Ocelot/Configuration/File/FileRoute.cs

@@ -24,6 +24,7 @@ public FileRoute()
             LoadBalancerOptions = new FileLoadBalancerOptions();
             SecurityOptions = new FileSecurityOptions();
             Priority = 1;
+            ConnectionClose = false;
         }
 
         public string DownstreamPathTemplate { get; set; }
@@ -56,6 +57,7 @@ public FileRoute()
         public int Timeout { get; set; }
         public bool DangerousAcceptAnyServerCertificateValidator { get; set; }
         public FileSecurityOptions SecurityOptions { get; set; }
-        public string DownstreamHttpVersion { get; set; }
+        public string DownstreamHttpVersion { get; set; }
+        public bool ConnectionClose { get; set; }
     }
 }

src/Ocelot/DependencyInjection/OcelotBuilder.cs

@@ -100,6 +100,7 @@ public OcelotBuilder(IServiceCollection services, IConfiguration configurationRo
             Services.TryAddSingleton<IAuthenticationOptionsCreator, AuthenticationOptionsCreator>();
             Services.TryAddSingleton<IUpstreamTemplatePatternCreator, UpstreamTemplatePatternCreator>();
             Services.TryAddSingleton<IRequestIdKeyCreator, RequestIdKeyCreator>();
+            Services.TryAddSingleton<IConnectionCloseCreator, ConnectionCloseCreator>();
             Services.TryAddSingleton<IServiceProviderConfigurationCreator, ServiceProviderConfigurationCreator>();
             Services.TryAddSingleton<IQoSOptionsCreator, QoSOptionsCreator>();
             Services.TryAddSingleton<IRouteOptionsCreator, RouteOptionsCreator>();

src/Ocelot/Requester/HttpClientBuilder.cs

@@ -65,7 +65,7 @@ public IHttpClient Create(DownstreamRoute downstreamRoute)
                 Timeout = timeout,
             };
 
-            _client = new HttpClientWrapper(_httpClient);
+            _client = new HttpClientWrapper(_httpClient, downstreamRoute.ConnectionClose);
 
             return _client;
         }
@@ -87,6 +87,7 @@ private static HttpClientHandler UseNonCookiesHandler(DownstreamRoute downstream
                 UseCookies = downstreamRoute.HttpHandlerOptions.UseCookieContainer,
                 UseProxy = downstreamRoute.HttpHandlerOptions.UseProxy,
                 MaxConnectionsPerServer = downstreamRoute.HttpHandlerOptions.MaxConnectionsPerServer,
+                
             };
         }
 

src/Ocelot/Requester/HttpClientWrapper.cs

@@ -11,13 +11,17 @@ public class HttpClientWrapper : IHttpClient
     {
         public HttpClient Client { get; }
 
-        public HttpClientWrapper(HttpClient client)
+        public bool ConnectionClose { get; }
+
+        public HttpClientWrapper(HttpClient client, bool connectionClose = false)
         {
             Client = client;
+            ConnectionClose = connectionClose;
         }
 
         public Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken = default)
         {
+            request.Headers.ConnectionClose = ConnectionClose;
             return Client.SendAsync(request, cancellationToken);
         }
     }