Skip to content

Commit 2d8bf58

Browse files
committed
updated with classic client options to keep backward compatibility
1 parent 75183e3 commit 2d8bf58

File tree

3 files changed

+10
-9
lines changed

3 files changed

+10
-9
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,7 @@ public Builder setMaxConnections(int maxConnections) {
367367
* @return
368368
*/
369369
public Builder setConnectionTTL(long timeout, ChronoUnit unit) {
370-
this.configuration.put("connection_ttl", String.valueOf(Duration.of(timeout, unit).toMillis()));
370+
this.configuration.put(ClickHouseClientOption.CONNECTION_TTL.getKey(), String.valueOf(Duration.of(timeout, unit).toMillis()));
371371
return this;
372372
}
373373

@@ -381,7 +381,7 @@ public Builder setConnectionTTL(long timeout, ChronoUnit unit) {
381381
* @return
382382
*/
383383
public Builder setKeepAliveTimeout(long timeout, ChronoUnit unit) {
384-
this.configuration.put("connection_keep_alive_timeout", String.valueOf(Duration.of(timeout, unit).toMillis()));
384+
this.configuration.put(ClickHouseHttpOption.KEEP_ALIVE_TIMEOUT.getKey(), String.valueOf(Duration.of(timeout, unit).toMillis()));
385385
return this;
386386
}
387387

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,8 @@ public SSLContext createSSLContext() {
134134

135135
private ConnectionConfig createConnectionConfig() {
136136
ConnectionConfig.Builder connConfig = ConnectionConfig.custom();
137-
connConfig.setTimeToLive(MapUtils.getLong(chConfiguration, "connection_ttl"), TimeUnit.MILLISECONDS);
137+
connConfig.setTimeToLive(MapUtils.getLong(chConfiguration, ClickHouseClientOption.CONNECTION_TTL.getKey()),
138+
TimeUnit.MILLISECONDS);
138139
connConfig.setConnectTimeout(MapUtils.getLong(chConfiguration, ClickHouseClientOption.CONNECTION_TIMEOUT.getKey()),
139140
TimeUnit.MILLISECONDS);
140141
connConfig.setValidateAfterInactivity(CONNECTION_INACTIVITY_CHECK, TimeUnit.MILLISECONDS); // non-configurable for now
@@ -221,7 +222,7 @@ public CloseableHttpClient createHttpClient() {
221222
} else {
222223
clientBuilder.setConnectionManager(basicConnectionManager(sslContext, socketConfig));
223224
}
224-
long keepAliveTimeout = MapUtils.getLong(chConfiguration, "connection_keep_alive_timeout");
225+
long keepAliveTimeout = MapUtils.getLong(chConfiguration, ClickHouseHttpOption.KEEP_ALIVE_TIMEOUT.getKey());
225226
if (keepAliveTimeout > 0) {
226227
clientBuilder.setKeepAliveStrategy((response, context) -> TimeValue.ofMilliseconds(keepAliveTimeout));
227228
}

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

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ public class ConnectionManagementTests extends BaseIntegrationTest{
2626
@Test(groups = {"integration"},dataProvider = "testConnectionTTLProvider")
2727
@SuppressWarnings("java:S2925")
2828
public void testConnectionTTL(Long connectionTtl, Long keepAlive, int openSockets) throws Exception {
29+
if (isCloud()) {
30+
return; // skip cloud tests because of wiremock proxy. TODO: fix it
31+
}
2932
ClickHouseNode server = getServer(ClickHouseProtocol.HTTP);
3033

3134
int proxyPort = new Random().nextInt(1000) + 10000;
@@ -46,7 +49,6 @@ public void testConnectionTTL(Long connectionTtl, Long keepAlive, int openSocket
4649
.setUsername("default")
4750
.setPassword(getPassword())
4851
.useNewImplementation(true)
49-
// .useNewImplementation(System.getProperty("client.tests.useNewImplementation", "false").equals("true"))
5052
.addProxy(ProxyType.HTTP, "localhost", proxyPort);
5153
if (connectionTtl != null) {
5254
clientBuilder.setConnectionTTL(connectionTtl, ChronoUnit.MILLIS);
@@ -96,23 +98,21 @@ private static class ConnectionCounterListener implements WiremockNetworkTraffic
9698
@Override
9799
public void opened(Socket socket) {
98100
opened.incrementAndGet();
99-
System.out.println("Opened: " + socket);
100101
}
101102

102103
@Override
103104
public void incoming(Socket socket, ByteBuffer bytes) {
104-
105+
// ignore
105106
}
106107

107108
@Override
108109
public void outgoing(Socket socket, ByteBuffer bytes) {
109-
110+
// ignore
110111
}
111112

112113
@Override
113114
public void closed(Socket socket) {
114115
closed.incrementAndGet();
115-
System.out.println("Closed: " + socket);
116116
}
117117
}
118118

0 commit comments

Comments
 (0)