com.clickhouse.benchmark.ServerState Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of clickhouse-benchmark Show documentation
Show all versions of clickhouse-benchmark Show documentation
Benchmarks for ClickHouse clients
The newest version!
package com.clickhouse.benchmark;
import static java.time.temporal.ChronoUnit.SECONDS;
import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.InterfaceAddress;
import java.net.NetworkInterface;
import java.time.Duration;
import java.util.Enumeration;
import org.openjdk.jmh.annotations.Level;
import org.openjdk.jmh.annotations.Scope;
import org.openjdk.jmh.annotations.Setup;
import org.openjdk.jmh.annotations.State;
import org.openjdk.jmh.annotations.TearDown;
import org.testcontainers.containers.GenericContainer;
import org.testcontainers.containers.wait.strategy.Wait;
import org.testcontainers.images.builder.ImageFromDockerfile;
@State(Scope.Benchmark)
public class ServerState {
static String getLocalIpAddress() {
String localIpAddress = null;
try {
for (Enumeration interfaces = NetworkInterface.getNetworkInterfaces(); interfaces
.hasMoreElements();) {
NetworkInterface i = interfaces.nextElement();
if (i.isUp() && !i.isLoopback() && !i.isPointToPoint() && !i.isVirtual()) {
for (InterfaceAddress addr : i.getInterfaceAddresses()) {
InetAddress inetAddr = addr.getAddress();
if (!(inetAddr instanceof Inet4Address)) {
continue;
}
localIpAddress = inetAddr.getHostAddress();
break;
}
}
if (localIpAddress != null) {
break;
}
}
} catch (Exception e) {
// ignore exception
}
return localIpAddress != null ? localIpAddress : Constants.DEFAULT_HOST;
}
private final String host = System.getProperty("dbHost", Constants.DEFAULT_HOST);
private final String user = System.getProperty("dbUser", Constants.DEFAULT_USER);
private final String passwd = System.getProperty("dbPasswd", Constants.DEFAULT_PASSWD);
private final String db = System.getProperty("dbName", Constants.DEFAULT_DB);
private final String localIpAddress = getLocalIpAddress();
private GenericContainer container = null;
@Setup(Level.Trial)
public void doSetup() throws Exception {
if (System.getProperty("dbHost") != null) {
return;
}
String imageTag = System.getProperty("clickhouseVersion");
if (imageTag == null || (imageTag = imageTag.trim()).isEmpty()) {
imageTag = "";
} else {
imageTag = ":" + imageTag;
}
final String imageNameWithTag = "clickhouse/clickhouse-server" + imageTag;
container = new GenericContainer<>(new ImageFromDockerfile().withDockerfileFromBuilder(builder -> builder
.from(imageNameWithTag)
.run("echo '0.0.0.0 8123 "
+ "9000 9004 "
+ "9005 "
+ "9009 "
+ "9100 ' > /etc/clickhouse-server/config.d/custom.xml")))
.withExposedPorts(Constants.GRPC_PORT, Constants.HTTP_PORT, Constants.MYSQL_PORT,
Constants.NATIVE_PORT)
.waitingFor(Wait.forHttp("/ping").forPort(Constants.HTTP_PORT).forStatusCode(200)
.withStartupTimeout(Duration.of(60, SECONDS)));
container.start();
}
@TearDown(Level.Trial)
public void doTearDown() throws Exception {
if (container != null) {
container.stop();
}
}
public String getHost() {
return container != null ? localIpAddress : host;
}
public int getPort(int defaultPort) {
return container != null ? container.getMappedPort(defaultPort) : defaultPort;
}
public String getUser() {
return container != null ? Constants.DEFAULT_USER : user;
}
public String getPassword() {
return container != null ? Constants.DEFAULT_PASSWD : passwd;
}
public String getDatabase() {
return container != null ? Constants.DEFAULT_DB : db;
}
}