Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.clickhouse.jdbc.JdbcParseHandler Maven / Gradle / Ivy
package com.clickhouse.jdbc;
import java.util.List;
import java.util.Map;
import java.util.Set;
import com.clickhouse.client.config.ClickHouseDefaults;
import com.clickhouse.data.ClickHouseChecker;
import com.clickhouse.data.ClickHouseFormat;
import com.clickhouse.data.ClickHouseUtils;
import com.clickhouse.jdbc.parser.ClickHouseSqlStatement;
import com.clickhouse.jdbc.parser.ParseHandler;
import com.clickhouse.jdbc.parser.StatementType;
public class JdbcParseHandler extends ParseHandler {
private static final String SETTING_MUTATIONS_SYNC = "mutations_sync";
private static final JdbcParseHandler[] INSTANCES;
static {
INSTANCES = new JdbcParseHandler[3 * 2];
for (int i = 0, len = INSTANCES.length; i < len; i++) {
INSTANCES[i] = new JdbcParseHandler((i & 0b100) == 0b100, (i & 0b10) == 0b10, (i & 1) == 1);
}
};
public static final ParseHandler INSTANCE = INSTANCES[0];
public static JdbcParseHandler getInstance(boolean allowLightWeightDelete, boolean allowLightWeightUpdate,
boolean allowLocaleFile) {
return INSTANCES[(allowLightWeightDelete ? 0b100 : 0) | (allowLightWeightUpdate ? 0b10 : 0)
| (allowLocaleFile ? 1 : 0)];
}
private final boolean allowLocalFile;
private final boolean allowLightWeightDelete;
private final boolean allowLightWeightUpdate;
private void addMutationSetting(String sql, StringBuilder builder, Map positions,
Map settings, int index) {
boolean hasSetting = settings != null && !settings.isEmpty();
String setting = hasSetting ? settings.get(SETTING_MUTATIONS_SYNC) : null;
if (setting == null) {
String keyword = "SETTINGS";
Integer settingsIndex = positions.get(keyword);
if (settingsIndex == null) {
builder.append(sql.substring(index)).append(" SETTINGS mutations_sync=1");
if (hasSetting) {
builder.append(',');
}
} else {
builder.append(sql.substring(index, settingsIndex)).append("SETTINGS mutations_sync=1,")
.append(sql.substring(settingsIndex + keyword.length()));
}
} else {
builder.append(sql.substring(index));
}
}
private ClickHouseSqlStatement handleDelete(String sql, StatementType stmtType, String cluster, String database,
String table, String input, String compressAlgorithm, String compressLevel, String format, String file,
List parameters, Map positions, Map settings,
Set tempTables) {
StringBuilder builder = new StringBuilder();
int index = positions.get("DELETE");
if (index > 0) {
builder.append(sql.substring(0, index));
}
index = positions.get("FROM");
Integer whereIdx = positions.get("WHERE");
if (whereIdx != null) {
builder.append("ALTER TABLE ");
if (!ClickHouseChecker.isNullOrEmpty(database)) {
builder.append('`').append(database).append('`').append('.');
}
builder.append('`').append(table).append('`').append(" DELETE ");
addMutationSetting(sql, builder, positions, settings, whereIdx);
} else {
builder.append("TRUNCATE TABLE").append(sql.substring(index + 4));
}
return new ClickHouseSqlStatement(builder.toString(), stmtType, cluster, database, table, input,
compressAlgorithm, compressLevel, format, file, parameters, null, settings, null);
}
private ClickHouseSqlStatement handleUpdate(String sql, StatementType stmtType, String cluster, String database,
String table, String input, String compressAlgorithm, String compressLevel, String format, String file,
List parameters, Map positions, Map settings,
Set tempTables) {
StringBuilder builder = new StringBuilder();
int index = positions.get("UPDATE");
if (index > 0) {
builder.append(sql.substring(0, index));
}
builder.append("ALTER TABLE ");
index = positions.get("SET");
if (!ClickHouseChecker.isNullOrEmpty(database)) {
builder.append('`').append(database).append('`').append('.');
}
builder.append('`').append(table).append('`').append(" UPDATE"); // .append(sql.substring(index + 3));
addMutationSetting(sql, builder, positions, settings, index + 3);
return new ClickHouseSqlStatement(builder.toString(), stmtType, cluster, database, table, input,
compressAlgorithm, compressLevel, format, file, parameters, null, settings, null);
}
private ClickHouseSqlStatement handleInFileForInsertQuery(String sql, StatementType stmtType, String cluster,
String database, String table, String input, String compressAlgorithm, String compressLevel, String format,
String file, List parameters, Map positions, Map settings,
Set tempTables) {
StringBuilder builder = new StringBuilder(sql.length());
builder.append(sql.substring(0, positions.get("FROM")));
Integer index = positions.get("SETTINGS");
if (index == null || index < 0) {
index = positions.get("FORMAT");
}
if (index != null && index > 0) {
builder.append(sql.substring(index));
} else {
ClickHouseFormat f = ClickHouseFormat.fromFileName(ClickHouseUtils.unescape(file));
if (f == null) {
f = (ClickHouseFormat) ClickHouseDefaults.FORMAT.getDefaultValue();
}
format = f.name();
builder.append("FORMAT ").append(format);
}
return new ClickHouseSqlStatement(builder.toString(), stmtType, cluster, database, table, input,
compressAlgorithm, compressLevel, format, file, parameters, null, settings, null);
}
private ClickHouseSqlStatement handleOutFileForSelectQuery(String sql, StatementType stmtType, String cluster,
String database, String table, String input, String compressAlgorithm, String compressLevel, String format,
String file, List parameters, Map positions, Map settings,
Set tempTables) {
StringBuilder builder = new StringBuilder(sql.length());
builder.append(sql.substring(0, positions.get("INTO")));
Integer index = positions.get("FORMAT");
if (index != null && index > 0) {
builder.append(sql.substring(index));
}
return new ClickHouseSqlStatement(builder.toString(), stmtType, cluster, database, table, input,
compressAlgorithm, compressLevel, format, file, parameters, null, settings, null);
}
@Override
public ClickHouseSqlStatement handleStatement(String sql, StatementType stmtType, String cluster, String database,
String table, String input, String compressAlgorithm, String compressLevel, String format, String file,
List parameters, Map positions, Map settings,
Set tempTables) {
boolean hasFile = allowLocalFile && !ClickHouseChecker.isNullOrEmpty(file) && file.charAt(0) == '\'';
ClickHouseSqlStatement s = null;
if (stmtType == StatementType.DELETE) {
s = allowLightWeightDelete ? s
: handleDelete(sql, stmtType, cluster, database, table, input, compressAlgorithm, compressLevel,
format, file, parameters, positions, settings, tempTables);
} else if (stmtType == StatementType.UPDATE) {
s = allowLightWeightUpdate ? s
: handleUpdate(sql, stmtType, cluster, database, table, input, compressAlgorithm, compressLevel,
format, file, parameters, positions, settings, tempTables);
} else if (stmtType == StatementType.INSERT && hasFile) {
s = handleInFileForInsertQuery(sql, stmtType, cluster, database, table, input, compressAlgorithm,
compressLevel, format, file, parameters, positions, settings, tempTables);
} else if (stmtType == StatementType.SELECT && hasFile) {
s = handleOutFileForSelectQuery(sql, stmtType, cluster, database, table, input, compressAlgorithm,
compressLevel, format, file, parameters, positions, settings, tempTables);
}
return s;
}
private JdbcParseHandler(boolean allowLightWeightDelete, boolean allowLightWeightUpdate, boolean allowLocalFile) {
this.allowLightWeightDelete = allowLightWeightDelete;
this.allowLightWeightUpdate = allowLightWeightUpdate;
this.allowLocalFile = allowLocalFile;
}
}
