added option to test without compression

Author: chernser

performance/src/test/com/clickhouse/benchmark/BenchmarkRunner.java

@@ -1,7 +1,6 @@
 package com.clickhouse.benchmark;
 
 
-import com.clickhouse.benchmark.clients.BenchmarkBase;
 import com.clickhouse.benchmark.clients.Components;
 import com.clickhouse.benchmark.clients.InsertClient;
 import com.clickhouse.benchmark.clients.QueryClient;
@@ -31,17 +30,15 @@ public static void main(String[] args) throws Exception {
         Map<String, String> argMap = parseArguments(args);
 
         Options opt = new OptionsBuilder()
-//                .param("datasetSourceName", argMap.getOrDefault("dataset", "simple"))
-//                .include(QueryClient.class.getSimpleName())
-//                .include(InsertClient.class.getSimpleName())
+                .include(QueryClient.class.getSimpleName())
+                .include(InsertClient.class.getSimpleName())
                 .include(Components.class.getSimpleName())
                 .forks(1) // must be a fork. No fork only for debugging
                 .mode(Mode.AverageTime)
                 .timeUnit(TimeUnit.MILLISECONDS)
                 .threads(1)
                 .addProfiler(GCProfiler.class)
                 .addProfiler(MemPoolProfiler.class)
-                .addProfiler("async", "libPath=/Users/mark.zitnik/Downloads/async-profiler-3.0-macos/lib/libasyncProfiler.dylib;output=flamegraph;dir=profile-results;event=alloc")
                 .warmupIterations(0)
                 .warmupTime(TimeValue.seconds(10))
                 .measurementIterations(10)

performance/src/test/com/clickhouse/benchmark/clients/BenchmarkBase.java

@@ -81,6 +81,9 @@ public static class DataState {
         String tableNameFilled;
         @Param({"data_empty"})
         String tableNameEmpty;
+
+        @Param({"true"})
+        boolean useClientCompression = true;
         DataSet dataSet;
 
         public void setDataSet(DataSet dataSet) {

performance/src/test/com/clickhouse/benchmark/clients/Components.java

@@ -4,12 +4,8 @@
 import com.clickhouse.client.api.internal.ClickHouseLZ4OutputStream;
 import com.clickhouse.data.ClickHouseOutputStream;
 import com.clickhouse.data.stream.Lz4OutputStream;
-import net.jpountz.lz4.LZ4Compressor;
 import net.jpountz.lz4.LZ4Factory;
 import org.openjdk.jmh.annotations.Benchmark;
-import org.openjdk.jmh.annotations.Level;
-import org.openjdk.jmh.annotations.Setup;
-import org.openjdk.jmh.annotations.TearDown;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -18,18 +14,6 @@
 public class Components extends BenchmarkBase {
     private static final Logger LOGGER = LoggerFactory.getLogger(Components.class);
 
-    @Setup(Level.Trial)
-    public void setup(DataState dataState) throws Exception {
-        super.setup(dataState, false);
-    }
-
-
-    @TearDown(Level.Invocation)
-    public void tearDownIteration(DataState dataState) {
-        verifyRowsInsertedAndCleanup(dataState.dataSet, false);
-    }
-
-
     @Benchmark
     public void CompressingOutputStreamV1(DataState dataState) {
         DataSet dataSet = dataState.dataSet;

performance/src/test/com/clickhouse/benchmark/clients/InsertClient.java

@@ -42,6 +42,7 @@ public void insertV1(DataState dataState) {
             try (ClickHouseResponse response = clientV1.read(getServer())
                     .write()
                     .option(ClickHouseClientOption.ASYNC, false)
+                    .option(ClickHouseClientOption.COMPRESS, dataState.useClientCompression)
                     .format(format)
                     .query(BenchmarkRunner.getInsertQuery(dataState.tableNameEmpty))
                     .data(out -> {
@@ -63,10 +64,9 @@ public void insertV2(DataState dataState) {
             try (InsertResponse response = clientV2.insert(dataState.tableNameEmpty, out -> {
                 for (byte[] bytes: dataState.dataSet.getBytesList(format)) {
                     out.write(bytes);
-
                 }
                 out.close();
-            }, format, new InsertSettings()).get()) {
+            }, format, new InsertSettings().compressClientRequest(dataState.useClientCompression)).get()) {
                 response.getWrittenRows();
             }
         } catch (Exception e) {