-
Notifications
You must be signed in to change notification settings - Fork 530
/
ApacheHttpConnectionImplTest.java
118 lines (102 loc) · 5.38 KB
/
ApacheHttpConnectionImplTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package com.clickhouse.client.http;
import com.clickhouse.client.ClickHouseClient;
import com.clickhouse.client.ClickHouseConfig;
import com.clickhouse.client.ClickHouseNode;
import com.clickhouse.client.ClickHouseProtocol;
import com.clickhouse.client.ClickHouseRequest;
import com.clickhouse.client.ClickHouseResponse;
import com.clickhouse.client.ClickHouseSocketFactory;
import com.clickhouse.client.config.ClickHouseClientOption;
import com.clickhouse.client.http.config.ClickHouseHttpOption;
import com.clickhouse.client.http.config.HttpConnectionProvider;
import com.clickhouse.config.ClickHouseOption;
import com.clickhouse.data.ClickHouseUtils;
import java.io.IOException;
import java.io.Serializable;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.hc.client5.http.socket.PlainConnectionSocketFactory;
import org.testng.Assert;
import org.testng.annotations.Test;
public class ApacheHttpConnectionImplTest extends ClickHouseHttpClientTest {
public static class CustomSocketFactory implements ClickHouseSocketFactory {
private static final AtomicBoolean created = new AtomicBoolean();
@Override
public <T> T create(ClickHouseConfig config, Class<T> clazz) throws IOException, UnsupportedOperationException {
if (config == null || clazz == null) {
throw new IllegalArgumentException("Non-null configuration and class are required");
} else if (List.class.isAssignableFrom(clazz)) {
return clazz.cast(Collections.singletonList(config));
} else if (PlainConnectionSocketFactory.class.isAssignableFrom(clazz)) {
if (created.compareAndSet(false, true)) {
return ApacheHttpConnectionImpl.ApacheHttpClientSocketFactory.instance.create(config, clazz);
} else {
throw new IOException("socket factory has already created");
}
}
throw new UnsupportedOperationException(ClickHouseUtils.format("Class %s is not supported", clazz));
}
@Override
public boolean supports(Class<?> clazz) {
return List.class.isAssignableFrom(clazz) || PlainConnectionSocketFactory.class.isAssignableFrom(clazz);
}
}
@Override
protected Map<ClickHouseOption, Serializable> getClientOptions() {
return Collections.singletonMap(ClickHouseHttpOption.CONNECTION_PROVIDER,
HttpConnectionProvider.APACHE_HTTP_CLIENT);
}
@Test(groups = { "integration" })
public void testConnection() throws Exception {
ClickHouseNode server = getServer(ClickHouseProtocol.HTTP);
try (ClickHouseClient client = ClickHouseClient.newInstance()) {
ClickHouseRequest<?> req1 = newRequest(client, server);
try (ClickHouseResponse resp = req1.query("select 1").executeAndWait()) {
Assert.assertEquals(resp.firstRecord().getValue(0).asString(), "1");
}
// req2 will use same connection with req1
ClickHouseRequest<?> req2 = newRequest(client, server);
try (ClickHouseResponse resp = req2.query("select 1").executeAndWait()) {
Assert.assertEquals(resp.firstRecord().getValue(0).asString(), "1");
}
}
}
@Test(groups = { "integration" })
public void testCustomOptions() throws Exception {
Map<String, String> customOptions = new HashMap<>();
customOptions.put(ClickHouseHttpOption.CONNECTION_PROVIDER.getKey(),
HttpConnectionProvider.APACHE_HTTP_CLIENT.name());
customOptions.put("", "nothing");
customOptions.put("my.custom.option.1", "one");
ClickHouseNode server = getServer(getProtocol(), customOptions);
try (ClickHouseClient client = ClickHouseClient.newInstance()) {
ClickHouseRequest<?> req = newRequest(client, server);
try (ClickHouseResponse resp = req.query("select 1").executeAndWait()) {
Assert.assertEquals(resp.firstRecord().getValue(0).asString(), "1");
}
}
customOptions.put(ClickHouseClientOption.CUSTOM_SOCKET_FACTORY.getKey(), CustomSocketFactory.class.getName());
customOptions.put(ClickHouseClientOption.CUSTOM_SOCKET_FACTORY_OPTIONS.getKey(), "a=1, b = 2, c='3\\,5'");
server = getServer(getProtocol(), customOptions);
try (ClickHouseClient client = ClickHouseClient.newInstance()) {
Assert.assertFalse(CustomSocketFactory.created.get());
ClickHouseRequest<?> req = newRequest(client, server);
Map<String, String> addtionalOptions = new TreeMap<>();
addtionalOptions.put("a", "1");
addtionalOptions.put("b", "2");
addtionalOptions.put("c", "'3,5'");
Assert.assertEquals(req.getConfig().getCustomSocketFactory(), CustomSocketFactory.class.getName());
Assert.assertEquals(req.getConfig().getCustomSocketFactoryOptions(), addtionalOptions);
try (ClickHouseResponse resp = req.query("select 1").executeAndWait()) {
Assert.assertEquals(resp.firstRecord().getValue(0).asString(), "1");
}
Assert.assertTrue(CustomSocketFactory.created.get());
} finally {
CustomSocketFactory.created.set(false);
}
}
}