added option to disable cookies

Author: chernser

client-v2/src/main/java/com/clickhouse/client/api/Client.java

@@ -450,6 +450,12 @@ public Builder useNewImplementation(boolean useNewImplementation) {
             return this;
         }
 
+        public Builder setHttpCookiesEnabled(boolean enabled) {
+            //TODO: extract to settings string constants
+            this.configuration.put("client.http.cookies_enabled", String.valueOf(enabled));
+            return this;
+        }
+
         public Client build() {
             // check if endpoint are empty. so can not initiate client
             if (this.endpoints.isEmpty()) {

client-v2/src/main/java/com/clickhouse/client/api/internal/HttpAPIClientHelper.java

@@ -90,6 +90,11 @@ public CloseableHttpClient createHttpClient() {
         } else if (proxyType == ProxyType.SOCKS) {
             soCfgBuilder.setSocksProxyAddress(new InetSocketAddress(proxyHost, Integer.parseInt(proxyPort)));
         }
+
+        if (chConfiguration.getOrDefault("client.http.cookies_enabled", "true")
+                .equalsIgnoreCase("false")) {
+            clientBuilder.disableCookieManagement();
+        }
         clientBuilder.setDefaultCredentialsProvider(credProviderBuilder.build());
         return clientBuilder.build();
     }

client-v2/src/test/java/com/clickhouse/client/ProxyTests.java

@@ -24,6 +24,7 @@
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 
 import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
@@ -132,6 +133,37 @@ public void testProxyWithCookies() {
         }
     }
 
+    @Test(groups = { "integration" })
+    public void testProxyWithDisabledCookies() {
+        client.set(clientBuilder(initProxy(), true).setHttpCookiesEnabled(false).build());
+        final int targetPort = getServer(ClickHouseProtocol.HTTP).getPort();
+
+        proxy.get().addStubMapping(post(urlMatching("/.*"))
+                .inScenario("routeCookies")
+                .whenScenarioStateIs(Scenario.STARTED)
+                .willReturn(aResponse().withHeader(HttpHeaders.SET_COOKIE, "routeName=routeA")
+                        .proxiedFrom("http://localhost:" + targetPort)).willSetStateTo("cookies").build());
+
+        proxy.get().addStubMapping(post(urlMatching("/.*"))
+                .inScenario("routeCookies")
+                .whenScenarioStateIs("cookies")
+                .withHeader(HttpHeaders.COOKIE, equalTo("routeName=routeA"))
+                .willReturn(aResponse().proxiedFrom("http://localhost:" + targetPort)).build());
+
+        try {
+            client.get().execute("select 1").get();
+        } catch (Exception e) {
+            fail("Should not have thrown exception.", e);
+        }
+        try {
+            client.get().execute("select 1").get();
+        } catch (ExecutionException e) {
+            Assert.assertTrue(e.getCause() instanceof ClientException);
+        } catch (Exception e) {
+            fail("Should have thrown exception.", e);
+        }
+    }
+
     private Client.Builder clientBuilder(int proxyPort, boolean onlyNewImplementation) {
         return new Client.Builder()
                 .addEndpoint(Protocol.HTTP, "clickhouse", 8123, false)