11package com .clickhouse .benchmark .clients ;
22
33import com .clickhouse .benchmark .BenchmarkRunner ;
4- import com .clickhouse .benchmark .TestEnvironment ;
54import com .clickhouse .benchmark .data .DataSet ;
65import com .clickhouse .benchmark .data .FileDataSet ;
76import com .clickhouse .benchmark .data .SimpleDataSet ;
@@ -109,8 +108,6 @@ public void setTableNameEmpty(String tableNameEmpty) {
109108
110109 }
111110
112-
113-
114111 @ Setup (Level .Trial )
115112 public void setup (DataState dataState ) {
116113 setupEnvironment ();
@@ -123,28 +120,34 @@ public void setup(DataState dataState) {
123120 }
124121 initializeTables (dataState );
125122 }
123+
126124 @ TearDown (Level .Trial )
127125 public void tearDown () {
128126 cleanupEnvironment ();
129127 }
130128
131129
132- private static void initializeTables (DataState dataState ) {
133- initializeTable (dataState .tableNameFilled , dataState .dataSet .getCreateTableString (dataState .tableNameFilled )
134- , dataState .dataSet .getInputStream (dataState .dataSet .getFormat ()), dataState .dataSet .getFormat ());
135- initializeTable (dataState .tableNameEmpty , dataState .dataSet .getCreateTableString (dataState .tableNameEmpty ), null , null );
130+ public static void initializeTables (DataState dataState ) {
131+ LOGGER .info ("Initializing tables: {}, {}" , dataState .tableNameFilled , dataState .tableNameEmpty );
132+ LOGGER .debug ("Create {}: {}" , dataState .tableNameFilled , dataState .dataSet .getCreateTableString (dataState .tableNameFilled ));
133+ LOGGER .debug ("Create {}: {}" , dataState .tableNameEmpty , dataState .dataSet .getCreateTableString (dataState .tableNameEmpty ));
134+ runAndSyncQuery (dataState .dataSet .getCreateTableString (dataState .tableNameEmpty ), dataState .tableNameEmpty );
135+ runAndSyncQuery (dataState .dataSet .getCreateTableString (dataState .tableNameFilled ), dataState .tableNameFilled );
136+ //Truncate tables if they existed
137+ truncateTable (dataState .tableNameEmpty );
138+ truncateTable (dataState .tableNameFilled );
139+
140+ ClickHouseFormat format = dataState .dataSet .getFormat ();
141+ LOGGER .debug ("Inserting data into table: {}, format: {}" , dataState .tableNameFilled , format );
142+ insertData (dataState .tableNameFilled , dataState .dataSet .getInputStream (format ), format );//For query testing
136143 loadClickHouseRecords (dataState );//For insert testing
137144 }
138145
139146
140147
141- public static void initializeTable (String tableName , String createTableString , InputStream dataStream , ClickHouseFormat format ) {
142- LOGGER .info ("Initializing table: {}" , tableName );
143- runAndSyncQuery (createTableString , tableName );
144- truncateTable (tableName );
145-
146- if (dataStream != null ) {
147- insertData (tableName , dataStream , format );
148+ public static void isNotNull (Object obj , boolean doWeCare ) {
149+ if (obj == null && doWeCare ) {
150+ throw new RuntimeException ("Object is null" );
148151 }
149152 }
150153
0 commit comments