Adding variant tests

Author: Paultagoras

clickhouse-http-client/src/test/perf/com/clickhouse/client/ClientBenchmark.java

@@ -1,7 +1,6 @@
 package com.clickhouse.client;
 
 import com.clickhouse.client.http.ApacheHttpConnectionImplTest;
-import com.clickhouse.client.http.ClickHouseHttpClientTest;
 import org.openjdk.jmh.annotations.*;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;

client-v2/src/test/perf/com/clickhouse/client/ClientBenchmark.java

@@ -1,13 +1,19 @@
 package com.clickhouse.client;
 
+import com.clickhouse.client.api.Client;
+import com.clickhouse.client.api.enums.Protocol;
+import com.clickhouse.client.api.query.GenericRecord;
 import com.clickhouse.client.insert.InsertTests;
 import com.clickhouse.client.query.QueryTests;
 import org.openjdk.jmh.annotations.*;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.testng.Assert;
 
 import java.util.concurrent.TimeUnit;
 
+import static com.clickhouse.client.ClickHouseServerForTest.isCloud;
+
 @BenchmarkMode(Mode.SampleTime)
 @OutputTimeUnit(TimeUnit.MILLISECONDS)
 @State(Scope.Benchmark)
@@ -19,19 +25,46 @@ public class ClientBenchmark {
     private static final int MEDIUM_SIZE = 1000000;
     private static final int LARGE_SIZE = 10000000;
 
+    private Client newClient;
+    private Client oldClient;
+
+    public static void main(String[] args) throws Exception {
+        org.openjdk.jmh.Main.main(args);
+    }
+
     @Setup
     public void setup() {
         BaseIntegrationTest.setupClickHouseContainer();
+
+        ClickHouseNode node = ClickHouseServerForTest.getClickHouseNode(ClickHouseProtocol.HTTP, isCloud(), ClickHouseNode.builder().build());
+        oldClient = new Client.Builder()
+                .addEndpoint(Protocol.HTTP, node.getHost(), node.getPort(), false)
+                .setUsername("default")
+                .setPassword("")
+                .compressClientRequest(false)
+                .compressServerResponse(false)
+                .useHttpCompression(false)
+                .useNewImplementation(false)
+                .build();
+        newClient = new Client.Builder()
+                .addEndpoint(Protocol.HTTP, node.getHost(), node.getPort(), false)
+                .setUsername("default")
+                .setPassword("")
+                .compressClientRequest(false)
+                .compressServerResponse(false)
+                .useHttpCompression(false)
+                .useNewImplementation(true)
+                .build();
     }
 
     @TearDown
     public void tearDown() {
+        oldClient.close();
+        newClient.close();
         BaseIntegrationTest.teardownClickHouseContainer();
     }
 
-    public static void main(String[] args) throws Exception {
-        org.openjdk.jmh.Main.main(args);
-    }
+
 
     // Add benchmark methods here
     @Benchmark
@@ -68,6 +101,54 @@ public void queryBenchmarkLarge() throws Exception {
         queryTests.tearDown();
     }
 
+    @Threads(4)
+    @Benchmark
+    public void queryBenchmarkSmallParallelOldImpl() throws Exception {
+        LOGGER.info("(V2) Query parallel benchmark");
+        GenericRecord record = oldClient.queryAll("SELECT number FROM system.numbers LIMIT " + SMALL_SIZE).stream().findFirst().get();
+        Assert.assertNotNull(record);
+    }
+
+    @Threads(4)
+    @Benchmark
+    public void queryBenchmarkMediumParallelOldImpl() throws Exception {
+        LOGGER.info("(V2) Query parallel benchmark");
+        GenericRecord record = oldClient.queryAll("SELECT number FROM system.numbers LIMIT " + MEDIUM_SIZE).stream().findFirst().get();
+        Assert.assertNotNull(record);
+    }
+
+    @Threads(4)
+    @Benchmark
+    public void queryBenchmarkLargeParallelOldImpl() throws Exception {
+        LOGGER.info("(V2) Query parallel benchmark");
+        GenericRecord record = oldClient.queryAll("SELECT number FROM system.numbers LIMIT " + LARGE_SIZE).stream().findFirst().get();
+        Assert.assertNotNull(record);
+    }
+
+    @Threads(4)
+    @Benchmark
+    public void queryBenchmarkSmallParallelNewImpl() throws Exception {
+        LOGGER.info("(V2) Query parallel benchmark");
+        GenericRecord record = newClient.queryAll("SELECT number FROM system.numbers LIMIT " + SMALL_SIZE).stream().findFirst().get();
+        Assert.assertNotNull(record);
+    }
+
+    @Threads(4)
+    @Benchmark
+    public void queryBenchmarkMediumParallelNewImpl() throws Exception {
+        LOGGER.info("(V2) Query parallel benchmark");
+        GenericRecord record = newClient.queryAll("SELECT number FROM system.numbers LIMIT " + MEDIUM_SIZE).stream().findFirst().get();
+        Assert.assertNotNull(record);
+    }
+
+    @Threads(4)
+    @Benchmark
+    public void queryBenchmarkLargeParallelNewImpl() throws Exception {
+        LOGGER.info("(V2) Query parallel benchmark");
+        GenericRecord record = newClient.queryAll("SELECT number FROM system.numbers LIMIT " + LARGE_SIZE).stream().findFirst().get();
+        Assert.assertNotNull(record);
+    }
+
     @Benchmark
     public void insertRawDataBenchmarkSmall() throws Exception {
         LOGGER.info("(V2) Insert raw data benchmark");