-
-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
ClickHouseContainer.java
69 lines (53 loc) · 1.86 KB
/
ClickHouseContainer.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
package org.testcontainers.containers;
import org.testcontainers.containers.wait.HttpWaitStrategy;
import java.time.Duration;
public class ClickHouseContainer extends JdbcDatabaseContainer {
public static final String NAME = "clickhouse";
public static final String IMAGE = "yandex/clickhouse-server";
public static final String DEFAULT_TAG = "18.10.3";
public static final Integer HTTP_PORT = 8123;
public static final Integer NATIVE_PORT = 9000;
private static final String DRIVER_CLASS_NAME = "ru.yandex.clickhouse.ClickHouseDriver";
private static final String JDBC_URL_PREFIX = "jdbc:" + NAME + "://";
private static final String TEST_QUERY = "SELECT 1";
private String databaseName = "default";
private String username = "default";
private String password = "";
public ClickHouseContainer() {
super(IMAGE + ":" + DEFAULT_TAG);
}
public ClickHouseContainer(String dockerImageName) {
super(dockerImageName);
withExposedPorts(HTTP_PORT, NATIVE_PORT);
waitingFor(
new HttpWaitStrategy()
.forStatusCode(200)
.forResponsePredicate(responseBody -> "Ok.".equals(responseBody))
.withStartupTimeout(Duration.ofMinutes(1))
);
}
@Override
protected Integer getLivenessCheckPort() {
return getMappedPort(HTTP_PORT);
}
@Override
public String getDriverClassName() {
return DRIVER_CLASS_NAME;
}
@Override
public String getJdbcUrl() {
return JDBC_URL_PREFIX + getHost() + ":" + getMappedPort(HTTP_PORT) + "/" + databaseName;
}
@Override
public String getUsername() {
return username;
}
@Override
public String getPassword() {
return password;
}
@Override
public String getTestQueryString() {
return TEST_QUERY;
}
}