support host/port arguments for clickhouse

sqlancer · mrigger · Mar 25, 2021 · Mar 9, 2021 · Mar 9, 2021 · Mar 9, 2021
commit 3051257cdeef232ef18edbb6fc2161212626d482
diff --git a/src/sqlancer/clickhouse/ClickHouseProvider.java b/src/sqlancer/clickhouse/ClickHouseProvider.java
@@ -21,6 +21,8 @@
 import sqlancer.common.query.SQLQueryProvider;
 
 public class ClickHouseProvider extends SQLProviderAdapter<ClickHouseGlobalState, ClickHouseOptions> {
+    protected String host;
+    protected String port;
 
     public ClickHouseProvider() {
         super(ClickHouseGlobalState.class, ClickHouseOptions.class);
@@ -102,9 +104,19 @@ public void generateDatabase(ClickHouseGlobalState globalState) throws Exception
 
     @Override
     public SQLConnection createDatabase(ClickHouseGlobalState globalState) throws SQLException {
+	host = globalState.getOptions().getHost();
+	port = globalState.getOptions().getPort();
+	if("sqlancer".equals(host)){
+	    host = "localhost";
+	}
+	if("sqlancer".equals(port)){
+	    port = "8123";
+	}
+
         ClickHouseOptions clickHouseOptions = globalState.getDmbsSpecificOptions();
         globalState.setClickHouseOptions(clickHouseOptions);
-        String url = "jdbc:clickhouse://localhost:8123/default";
+        //String url = "jdbc:clickhouse://localhost:8123/default";
+	String url = "jdbc:clickhouse://" + host + ":" + port + "/defult";
         String databaseName = globalState.getDatabaseName();
         Connection con = DriverManager.getConnection(url, globalState.getOptions().getUserName(),
                 globalState.getOptions().getPassword());