Skip to content

Commit 01c782e

Browse files
committed
Expose dataset properties like tableNameFilled, tableNameEmpty. so it will be easier to init externally
1 parent 33df298 commit 01c782e

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

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

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,19 +93,26 @@ public void setLimit(int limit) {
9393
this.limit = limit;
9494
}
9595

96+
public void setTableNameFilled(String tableNameFilled) {
97+
this.tableNameFilled = tableNameFilled;
98+
}
99+
100+
public void setTableNameEmpty(String tableNameEmpty) {
101+
this.tableNameEmpty = tableNameEmpty;
102+
}
103+
96104
}
97105

98106
@Setup(Level.Trial)
99107
public void setup(DataState dataState) {
100108
setupEnvironment();
101109
LOGGER.info("Setup benchmarks using dataset: {}", dataState.datasetSourceName);
102-
if ("simple".equals(dataState.datasetSourceName) && dataState.dataSet == null) {
110+
if (dataState.dataSet == null && "simple".equals(dataState.datasetSourceName)) {
103111
dataState.dataSet = new SimpleDataSet();
104112
} else if (dataState.dataSet == null && dataState.datasetSourceName.startsWith("file://")) {
105113
LOGGER.info("Loading data from file " + dataState.datasetSourceName + " with limit " + dataState.limit);
106114
dataState.dataSet = new FileDataSet(dataState.datasetSourceName.substring("file://".length()), dataState.limit);
107115
}
108-
109116
initializeTables(dataState);
110117
}
111118

0 commit comments

Comments
 (0)