From 6ead3716f08623363f33c3c2ed445e7922b9a041 Mon Sep 17 00:00:00 2001
From: "mikhail.koltunov" <mikhail.koltunov@dif.tech>
Date: Fri, 23 Feb 2024 18:14:24 +0600
Subject: [PATCH] Add configuration for the authorization http client

---
 .../kafka/ui/config/HttpClientProperties.java | 17 ++++++
 .../kafka/ui/config/SecurityAuthConfig.java   | 59 +++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/HttpClientProperties.java
 create mode 100644 kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/SecurityAuthConfig.java

diff --git a/kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/HttpClientProperties.java b/kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/HttpClientProperties.java
new file mode 100644
index 00000000000..1c572b74550
--- /dev/null
+++ b/kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/HttpClientProperties.java
@@ -0,0 +1,17 @@
+package com.provectus.kafka.ui.config;
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+@ConfigurationProperties("httpclient")
+@Data
+public class HttpClientProperties {
+
+  Integer connectTimeoutMillis;
+  Boolean socketKeepAlive;
+  Integer tcpKeepIdle;
+  Integer tcpKeepInterval;
+  Integer tcpKeepCount;
+}
diff --git a/kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/SecurityAuthConfig.java b/kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/SecurityAuthConfig.java
new file mode 100644
index 00000000000..4c6865f9b89
--- /dev/null
+++ b/kafka-ui-api/src/main/java/com/provectus/kafka/ui/config/SecurityAuthConfig.java
@@ -0,0 +1,59 @@
+package com.provectus.kafka.ui.config;
+
+import io.netty.channel.ChannelOption;
+import io.netty.channel.socket.nio.NioChannelOption;
+import jdk.net.ExtendedSocketOptions;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.client.reactive.ReactorClientHttpConnector;
+import org.springframework.security.oauth2.client.endpoint.OAuth2AuthorizationCodeGrantRequest;
+import org.springframework.security.oauth2.client.endpoint.ReactiveOAuth2AccessTokenResponseClient;
+import org.springframework.security.oauth2.client.endpoint.WebClientReactiveAuthorizationCodeTokenResponseClient;
+import org.springframework.web.reactive.function.client.WebClient;
+import reactor.netty.http.client.HttpClient;
+
+@Configuration
+public class SecurityAuthConfig {
+
+  public static class ConfigurableReactiveAuthClient extends WebClientReactiveAuthorizationCodeTokenResponseClient {
+
+    public ConfigurableReactiveAuthClient(HttpClient httpClient) {
+      var connector = new ReactorClientHttpConnector(httpClient);
+      setWebClient(WebClient.builder().clientConnector(connector).build());
+    }
+  }
+
+  @Bean
+  public ReactiveOAuth2AccessTokenResponseClient<OAuth2AuthorizationCodeGrantRequest> authClient(
+      HttpClientProperties properties) {
+    HttpClient httpClient = httpClientWithProperties(properties);
+    return new ConfigurableReactiveAuthClient(httpClient);
+  }
+
+  private HttpClient httpClientWithProperties(HttpClientProperties properties) {
+    HttpClient httpClient = HttpClient.create();
+    Integer connectTimeoutMillis = properties.getConnectTimeoutMillis();
+    Boolean socketKeepAlive = properties.getSocketKeepAlive();
+    if (connectTimeoutMillis != null) {
+      httpClient = httpClient.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, connectTimeoutMillis);
+    }
+    Integer tcpKeepIdle = properties.getTcpKeepIdle();
+    Integer tcpKeepInterval = properties.getTcpKeepInterval();
+
+    if (socketKeepAlive != null) {
+      httpClient = httpClient.option(ChannelOption.SO_KEEPALIVE, socketKeepAlive);
+    }
+    if (tcpKeepIdle != null) {
+      httpClient = httpClient.option(NioChannelOption.of(ExtendedSocketOptions.TCP_KEEPIDLE), tcpKeepIdle);
+    }
+    if (tcpKeepInterval != null) {
+      httpClient = httpClient.option(NioChannelOption.of(ExtendedSocketOptions.TCP_KEEPINTERVAL), tcpKeepInterval);
+    }
+    Integer tcpKeepCount = properties.getTcpKeepCount();
+    if (tcpKeepCount != null) {
+      httpClient = httpClient.option(NioChannelOption.of(ExtendedSocketOptions.TCP_KEEPCOUNT), tcpKeepCount);
+    }
+    return httpClient;
+  }
+
+}