fixes #2209 update MrasHandler to set keepalive timeout to 10 seconds (…

…#2210)
networknt · Apr 12, 2024 · 19d6aa1 · 19d6aa1
1 parent 4558d06
commit 19d6aa1
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/ingress-proxy/src/main/java/com/networknt/proxy/mras/MrasHandler.java b/ingress-proxy/src/main/java/com/networknt/proxy/mras/MrasHandler.java
@@ -319,10 +319,13 @@ private void invokeApi(HttpServerExchange exchange, String serviceHost, String r
                 }
                 // this a workaround to bypass the hostname verification in jdk11 http client.
                 Map<String, Object> tlsMap = (Map<String, Object>)ClientConfig.get().getMappedConfig().get(Http2Client.TLS);
+                final Properties props = System.getProperties();
                 if(tlsMap != null && !Boolean.TRUE.equals(tlsMap.get(TLSConfig.VERIFY_HOSTNAME))) {
-                    final Properties props = System.getProperties();
                     props.setProperty("jdk.internal.httpclient.disableHostnameVerification", Boolean.TRUE.toString());
                 }
+                props.setProperty("jdk.httpclient.keepalive.timeout", "10");
+                props.setProperty("jdk.httpclient.connectionPoolSize", "10");
+
                 client = clientBuilder.build();
             } catch (IOException e) {
                 logger.error("Cannot create HttpClient:", e);
@@ -367,10 +370,12 @@ private Result<TokenResponse> getAccessToken() throws Exception {
                 }
                 // this a workaround to bypass the hostname verification in jdk11 http client.
                 Map<String, Object> tlsMap = (Map<String, Object>)ClientConfig.get().getMappedConfig().get(Http2Client.TLS);
+                final Properties props = System.getProperties();
                 if(tlsMap != null && !Boolean.TRUE.equals(tlsMap.get(TLSConfig.VERIFY_HOSTNAME))) {
-                    final Properties props = System.getProperties();
                     props.setProperty("jdk.internal.httpclient.disableHostnameVerification", Boolean.TRUE.toString());
                 }
+                props.setProperty("jdk.httpclient.keepalive.timeout", "10");
+                props.setProperty("jdk.httpclient.connectionPoolSize", "10");
                 client = clientBuilder.build();
             } catch (IOException e) {
                 logger.error("Cannot create HttpClient:", e);
@@ -440,10 +445,13 @@ private Result<TokenResponse> getMicrosoftToken() throws Exception {
                 }
                 // this a workaround to bypass the hostname verification in jdk11 http client.
                 Map<String, Object> tlsMap = (Map<String, Object>)ClientConfig.get().getMappedConfig().get(Http2Client.TLS);
+                final Properties props = System.getProperties();
                 if(tlsMap != null && !Boolean.TRUE.equals(tlsMap.get(TLSConfig.VERIFY_HOSTNAME))) {
-                    final Properties props = System.getProperties();
                     props.setProperty("jdk.internal.httpclient.disableHostnameVerification", Boolean.TRUE.toString());
                 }
+                props.setProperty("jdk.httpclient.keepalive.timeout", "10");
+                props.setProperty("jdk.httpclient.connectionPoolSize", "10");
+
                 clientMicrosoft = clientBuilder.build();
             } catch (IOException e) {
                 logger.error("Cannot create HttpClient:", e);