package ru.yandex.clickhouse.settings;
import ru.yandex.clickhouse.util.apache.StringUtils;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
public class ClickHouseProperties {
// connection settings
private boolean async;
private int bufferSize;
private int apacheBufferSize;
private int socketTimeout;
private int connectionTimeout;
private int dataTransferTimeout;
private int keepAliveTimeout;
private int timeToLiveMillis;
private int defaultMaxPerRoute;
private int maxTotal;
private String host;
private int port;
private boolean usePathAsDb;
private String path;
private boolean ssl;
private String sslRootCertificate;
private String sslMode;
/**
* Maximum number of allowed redirects. Active only when {@link ClickHouseProperties#checkForRedirects}
* is <code>true</code>
*/
private int maxRedirects;
/**
* If set to <code>true</code>, driver will first try to connect to the server using GET request. If the response is 307,
* it will use URI given in the response's Location header instead of the original one.
* <p>
* Those queries will be repeated until response is anything other than 307, or until
* {@link ClickHouseProperties#maxRedirects maxRedirects} is hit.
* <p>
* This is a workaround to issues with properly following HTTP POST redirects.
* Namely, Apache HTTP client's inability to process early responses, and difficulties with resending non-repeatable
* {@link org.apache.http.entity.InputStreamEntity InputStreamEntity}
*/
private boolean checkForRedirects;
//additional
private int maxCompressBufferSize;
private boolean useServerTimeZone;
private String useTimeZone;
private boolean useServerTimeZoneForDates;
private boolean useObjectsInArrays;
// queries settings
private Integer maxParallelReplicas;
private Integer maxPartitionsPerInsertBlock;
private String totalsMode;
private String quotaKey;
private Integer priority;
private String database;
private boolean compress;
private boolean decompress;
private boolean extremes;
private Integer maxThreads;
private Integer maxExecutionTime;
private Integer maxBlockSize;
private Integer maxRowsToGroupBy;
private String profile;
private String user;
private String password;
private String httpAuthorization;
private boolean distributedAggregationMemoryEfficient;
private Long maxBytesBeforeExternalGroupBy;
private Long maxBytesBeforeExternalSort;
private Long maxMemoryUsage;
private Long maxMemoryUsageForUser;
private Long maxMemoryUsageForAllQueries;
private Long preferredBlockSizeBytes;
private Long maxQuerySize;
private Long maxAstElements;
private boolean sessionCheck;
private String sessionId;
private Long sessionTimeout;
private Long insertQuorum;
private Long insertQuorumTimeout;
private Long selectSequentialConsistency;
private Boolean enableOptimizePredicateExpression;
private Long maxInsertBlockSize;
private Boolean insertDeduplicate;
private Boolean insertDistributedSync;
private Boolean anyJoinDistinctRightTableKeys;
public ClickHouseProperties() {
this(new Properties());
}
public ClickHouseProperties(Properties info) {
// need casts for java 6
this.async = (Boolean)getSetting(info, ClickHouseConnectionSettings.ASYNC);
this.bufferSize = (Integer)getSetting(info, ClickHouseConnectionSettings.BUFFER_SIZE);
this.apacheBufferSize = (Integer)getSetting(info, ClickHouseConnectionSettings.APACHE_BUFFER_SIZE);
this.socketTimeout = (Integer)getSetting(info, ClickHouseConnectionSettings.SOCKET_TIMEOUT);
this.connectionTimeout = (Integer)getSetting(info, ClickHouseConnectionSettings.CONNECTION_TIMEOUT);
this.dataTransferTimeout = (Integer)getSetting(info, ClickHouseConnectionSettings.DATA_TRANSFER_TIMEOUT);
this.keepAliveTimeout = (Integer)getSetting(info, ClickHouseConnectionSettings.KEEP_ALIVE_TIMEOUT);
this.timeToLiveMillis = (Integer)getSetting(info, ClickHouseConnectionSettings.TIME_TO_LIVE_MILLIS);
this.defaultMaxPerRoute = (Integer)getSetting(info, ClickHouseConnectionSettings.DEFAULT_MAX_PER_ROUTE);
this.maxTotal = (Integer)getSetting(info, ClickHouseConnectionSettings.MAX_TOTAL);
this.maxCompressBufferSize = (Integer) getSetting(info, ClickHouseConnectionSettings.MAX_COMPRESS_BUFFER_SIZE);
this.ssl = (Boolean) getSetting(info, ClickHouseConnectionSettings.SSL);
this.sslRootCertificate = (String) getSetting(info, ClickHouseConnectionSettings.SSL_ROOT_CERTIFICATE);
this.sslMode = (String) getSetting(info, ClickHouseConnectionSettings.SSL_MODE);
this.usePathAsDb = (Boolean) getSetting(info, ClickHouseConnectionSettings.USE_PATH_AS_DB);
this.path = (String) getSetting(info, ClickHouseConnectionSettings.PATH);
this.maxRedirects = (Integer) getSetting(info, ClickHouseConnectionSettings.MAX_REDIRECTS);
this.checkForRedirects = (Boolean) getSetting(info, ClickHouseConnectionSettings.CHECK_FOR_REDIRECTS);
this.useServerTimeZone = (Boolean)getSetting(info, ClickHouseConnectionSettings.USE_SERVER_TIME_ZONE);
this.useTimeZone = (String)getSetting(info, ClickHouseConnectionSettings.USE_TIME_ZONE);
this.useServerTimeZoneForDates = (Boolean)getSetting(info, ClickHouseConnectionSettings.USE_SERVER_TIME_ZONE_FOR_DATES);
this.useObjectsInArrays = (Boolean)getSetting(info, ClickHouseConnectionSettings.USE_OBJECTS_IN_ARRAYS);
this.maxParallelReplicas = getSetting(info, ClickHouseQueryParam.MAX_PARALLEL_REPLICAS);
this.maxPartitionsPerInsertBlock = getSetting(info, ClickHouseQueryParam.MAX_PARTITIONS_PER_INSERT_BLOCK);
this.totalsMode = getSetting(info, ClickHouseQueryParam.TOTALS_MODE);
this.quotaKey = getSetting(info, ClickHouseQueryParam.QUOTA_KEY);
this.priority = getSetting(info, ClickHouseQueryParam.PRIORITY);
this.database = getSetting(info, ClickHouseQueryParam.DATABASE);
this.compress = (Boolean)getSetting(info, ClickHouseQueryParam.COMPRESS);
this.decompress = (Boolean)getSetting(info, ClickHouseQueryParam.DECOMPRESS);
this.extremes = (Boolean)getSetting(info, ClickHouseQueryParam.EXTREMES);
this.maxThreads = getSetting(info, ClickHouseQueryParam.MAX_THREADS);
this.maxExecutionTime = getSetting(info, ClickHouseQueryParam.MAX_EXECUTION_TIME);
this.maxBlockSize = getSetting(info, ClickHouseQueryParam.MAX_BLOCK_SIZE);
this.maxRowsToGroupBy = getSetting(info, ClickHouseQueryParam.MAX_ROWS_TO_GROUP_BY);
this.profile = getSetting(info, ClickHouseQueryParam.PROFILE);
this.user = getSetting(info, ClickHouseQueryParam.USER);
this.password = getSetting(info, ClickHouseQueryParam.PASSWORD);
this.httpAuthorization = getSetting(info, ClickHouseQueryParam.AUTHORIZATION);
this.distributedAggregationMemoryEfficient = (Boolean)getSetting(info, ClickHouseQueryParam.DISTRIBUTED_AGGREGATION_MEMORY_EFFICIENT);
this.maxBytesBeforeExternalGroupBy = (Long)getSetting(info, ClickHouseQueryParam.MAX_BYTES_BEFORE_EXTERNAL_GROUP_BY);
this.maxBytesBeforeExternalSort = (Long)getSetting(info, ClickHouseQueryParam.MAX_BYTES_BEFORE_EXTERNAL_SORT);
this.maxMemoryUsage = getSetting(info, ClickHouseQueryParam.MAX_MEMORY_USAGE);
this.maxMemoryUsageForUser = getSetting(info, ClickHouseQueryParam.MAX_MEMORY_USAGE_FOR_USER);
this.maxMemoryUsageForAllQueries = getSetting(info, ClickHouseQueryParam.MAX_MEMORY_USAGE_FOR_ALL_QUERIES);
this.preferredBlockSizeBytes = getSetting(info, ClickHouseQueryParam.PREFERRED_BLOCK_SIZE_BYTES);
this.maxQuerySize = getSetting(info, ClickHouseQueryParam.MAX_QUERY_SIZE);
this.maxAstElements = getSetting(info, ClickHouseQueryParam.MAX_AST_ELEMENTS);
this.sessionCheck = (Boolean) getSetting(info, ClickHouseQueryParam.SESSION_CHECK);
this.sessionId = getSetting(info, ClickHouseQueryParam.SESSION_ID);
this.sessionTimeout = getSetting(info, ClickHouseQueryParam.SESSION_TIMEOUT);
this.insertQuorum = (Long)getSetting(info, ClickHouseQueryParam.INSERT_QUORUM);
this.insertQuorumTimeout = (Long)getSetting(info, ClickHouseQueryParam.INSERT_QUORUM_TIMEOUT);
this.selectSequentialConsistency = (Long)getSetting(info, ClickHouseQueryParam.SELECT_SEQUENTIAL_CONSISTENCY);
this.enableOptimizePredicateExpression = getSetting(info, ClickHouseQueryParam.ENABLE_OPTIMIZE_PREDICATE_EXPRESSION);
this.maxInsertBlockSize = getSetting(info, ClickHouseQueryParam.MAX_INSERT_BLOCK_SIZE);
this.insertDeduplicate = getSetting(info, ClickHouseQueryParam.INSERT_DEDUPLICATE);
this.insertDistributedSync = getSetting(info, ClickHouseQueryParam.INSERT_DISTRIBUTED_SYNC);
this.anyJoinDistinctRightTableKeys = getSetting(info, ClickHouseQueryParam.ANY_JOIN_DISTINCT_RIGHT_TABLE_KEYS);
}
public Properties asProperties() {
PropertiesBuilder ret = new PropertiesBuilder();
ret.put(ClickHouseConnectionSettings.ASYNC.getKey(), String.valueOf(async));
ret.put(ClickHouseConnectionSettings.BUFFER_SIZE.getKey(), String.valueOf(bufferSize));
ret.put(ClickHouseConnectionSettings.APACHE_BUFFER_SIZE.getKey(), String.valueOf(apacheBufferSize));
ret.put(ClickHouseConnectionSettings.SOCKET_TIMEOUT.getKey(), String.valueOf(socketTimeout));
ret.put(ClickHouseConnectionSettings.CONNECTION_TIMEOUT.getKey(), String.valueOf(connectionTimeout));
ret.put(ClickHouseConnectionSettings.DATA_TRANSFER_TIMEOUT.getKey(), String.valueOf(dataTransferTimeout));
ret.put(ClickHouseConnectionSettings.KEEP_ALIVE_TIMEOUT.getKey(), String.valueOf(keepAliveTimeout));
ret.put(ClickHouseConnectionSettings.TIME_TO_LIVE_MILLIS.getKey(), String.valueOf(timeToLiveMillis));
ret.put(ClickHouseConnectionSettings.DEFAULT_MAX_PER_ROUTE.getKey(), String.valueOf(defaultMaxPerRoute));
ret.put(ClickHouseConnectionSettings.MAX_TOTAL.getKey(), String.valueOf(maxTotal));
ret.put(ClickHouseConnectionSettings.MAX_COMPRESS_BUFFER_SIZE.getKey(), String.valueOf(maxCompressBufferSize));
ret.put(ClickHouseConnectionSettings.SSL.getKey(), String.valueOf(ssl));
ret.put(ClickHouseConnectionSettings.SSL_ROOT_CERTIFICATE.getKey(), String.valueOf(sslRootCertificate));
ret.put(ClickHouseConnectionSettings.SSL_MODE.getKey(), String.valueOf(sslMode));
ret.put(ClickHouseConnectionSettings.USE_PATH_AS_DB.getKey(), String.valueOf(usePathAsDb));
ret.put(ClickHouseConnectionSettings.PATH.getKey(), String.valueOf(path));
ret.put(ClickHouseConnectionSettings.MAX_REDIRECTS.getKey(), String.valueOf(maxRedirects));
ret.put(ClickHouseConnectionSettings.CHECK_FOR_REDIRECTS.getKey(), String.valueOf(checkForRedirects));
ret.put(ClickHouseConnectionSettings.USE_SERVER_TIME_ZONE.getKey(), String.valueOf(useServerTimeZone));
ret.put(ClickHouseConnectionSettings.USE_TIME_ZONE.getKey(), String.valueOf(useTimeZone));
ret.put(ClickHouseConnectionSettings.USE_SERVER_TIME_ZONE_FOR_DATES.getKey(), String.valueOf(useServerTimeZoneForDates));
ret.put(ClickHouseConnectionSettings.USE_OBJECTS_IN_ARRAYS.getKey(), String.valueOf(useObjectsInArrays));
ret.put(ClickHouseQueryParam.MAX_PARALLEL_REPLICAS.getKey(), maxParallelReplicas);
ret.put(ClickHouseQueryParam.MAX_PARTITIONS_PER_INSERT_BLOCK.getKey(), maxPartitionsPerInsertBlock);
ret.put(ClickHouseQueryParam.TOTALS_MODE.getKey(), totalsMode);
ret.put(ClickHouseQueryParam.QUOTA_KEY.getKey(), quotaKey);
ret.put(ClickHouseQueryParam.PRIORITY.getKey(), priority);
ret.put(ClickHouseQueryParam.DATABASE.getKey(), database);
ret.put(ClickHouseQueryParam.COMPRESS.getKey(), String.valueOf(compress));
ret.put(ClickHouseQueryParam.DECOMPRESS.getKey(), String.valueOf(decompress));
ret.put(ClickHouseQueryParam.EXTREMES.getKey(), String.valueOf(extremes));
ret.put(ClickHouseQueryParam.MAX_THREADS.getKey(), maxThreads);
ret.put(ClickHouseQueryParam.MAX_EXECUTION_TIME.getKey(), maxExecutionTime);
ret.put(ClickHouseQueryParam.MAX_BLOCK_SIZE.getKey(), maxBlockSize);
ret.put(ClickHouseQueryParam.MAX_ROWS_TO_GROUP_BY.getKey(), maxRowsToGroupBy);
ret.put(ClickHouseQueryParam.PROFILE.getKey(), profile);
ret.put(ClickHouseQueryParam.USER.getKey(), user);
ret.put(ClickHouseQueryParam.PASSWORD.getKey(), password);
ret.put(ClickHouseQueryParam.AUTHORIZATION.getKey(), httpAuthorization);
ret.put(ClickHouseQueryParam.DISTRIBUTED_AGGREGATION_MEMORY_EFFICIENT.getKey(), String.valueOf(distributedAggregationMemoryEfficient));
ret.put(ClickHouseQueryParam.MAX_BYTES_BEFORE_EXTERNAL_GROUP_BY.getKey(), maxBytesBeforeExternalGroupBy);
ret.put(ClickHouseQueryParam.MAX_BYTES_BEFORE_EXTERNAL_SORT.getKey(), maxBytesBeforeExternalSort);
ret.put(ClickHouseQueryParam.MAX_MEMORY_USAGE.getKey(), maxMemoryUsage);
ret.put(ClickHouseQueryParam.MAX_MEMORY_USAGE_FOR_USER.getKey(), maxMemoryUsageForUser);
ret.put(ClickHouseQueryParam.MAX_MEMORY_USAGE_FOR_ALL_QUERIES.getKey(), maxMemoryUsageForAllQueries);
ret.put(ClickHouseQueryParam.PREFERRED_BLOCK_SIZE_BYTES.getKey(), preferredBlockSizeBytes);
ret.put(ClickHouseQueryParam.MAX_QUERY_SIZE.getKey(), maxQuerySize);
ret.put(ClickHouseQueryParam.MAX_AST_ELEMENTS.getKey(), maxAstElements);
ret.put(ClickHouseQueryParam.SESSION_CHECK.getKey(), String.valueOf(sessionCheck));
ret.put(ClickHouseQueryParam.SESSION_ID.getKey(), sessionId);
ret.put(ClickHouseQueryParam.SESSION_TIMEOUT.getKey(), sessionTimeout);
ret.put(ClickHouseQueryParam.INSERT_QUORUM.getKey(), insertQuorum);
ret.put(ClickHouseQueryParam.INSERT_QUORUM_TIMEOUT.getKey(), insertQuorumTimeout);
ret.put(ClickHouseQueryParam.SELECT_SEQUENTIAL_CONSISTENCY.getKey(), selectSequentialConsistency);
ret.put(ClickHouseQueryParam.ENABLE_OPTIMIZE_PREDICATE_EXPRESSION.getKey(), enableOptimizePredicateExpression);
ret.put(ClickHouseQueryParam.MAX_INSERT_BLOCK_SIZE.getKey(), maxInsertBlockSize);
ret.put(ClickHouseQueryParam.INSERT_DEDUPLICATE.getKey(), insertDeduplicate);
ret.put(ClickHouseQueryParam.INSERT_DISTRIBUTED_SYNC.getKey(), insertDistributedSync);
ret.put(ClickHouseQueryParam.ANY_JOIN_DISTINCT_RIGHT_TABLE_KEYS.getKey(), anyJoinDistinctRightTableKeys);
return ret.getProperties();
}
public ClickHouseProperties(ClickHouseProperties properties) {
setHost(properties.host);
setPort(properties.port);
setAsync(properties.async);
setBufferSize(properties.bufferSize);
setApacheBufferSize(properties.apacheBufferSize);
setSocketTimeout(properties.socketTimeout);
setConnectionTimeout(properties.connectionTimeout);
setDataTransferTimeout(properties.dataTransferTimeout);
setKeepAliveTimeout(properties.keepAliveTimeout);
setTimeToLiveMillis(properties.timeToLiveMillis);
setDefaultMaxPerRoute(properties.defaultMaxPerRoute);
setMaxTotal(properties.maxTotal);
setMaxCompressBufferSize(properties.maxCompressBufferSize);
setSsl(properties.ssl);
setSslRootCertificate(properties.sslRootCertificate);
setSslMode(properties.sslMode);
setUsePathAsDb(properties.usePathAsDb);
setPath(properties.path);
setMaxRedirects(properties.maxRedirects);
setCheckForRedirects(properties.checkForRedirects);
setUseServerTimeZone(properties.useServerTimeZone);
setUseTimeZone(properties.useTimeZone);
setUseServerTimeZoneForDates(properties.useServerTimeZoneForDates);
setUseObjectsInArrays(properties.useObjectsInArrays);
setMaxParallelReplicas(properties.maxParallelReplicas);
setMaxPartitionsPerInsertBlock(properties.maxPartitionsPerInsertBlock);
setTotalsMode(properties.totalsMode);
setQuotaKey(properties.quotaKey);
setPriority(properties.priority);
setDatabase(properties.database);
setCompress(properties.compress);
setDecompress(properties.decompress);
setExtremes(properties.extremes);
setMaxThreads(properties.maxThreads);
setMaxExecutionTime(properties.maxExecutionTime);
setMaxBlockSize(properties.maxBlockSize);
setMaxRowsToGroupBy(properties.maxRowsToGroupBy);
setProfile(properties.profile);
setUser(properties.user);
setPassword(properties.password);
setHttpAuthorization(properties.httpAuthorization);
setDistributedAggregationMemoryEfficient(properties.distributedAggregationMemoryEfficient);
setMaxBytesBeforeExternalGroupBy(properties.maxBytesBeforeExternalGroupBy);
setMaxBytesBeforeExternalSort(properties.maxBytesBeforeExternalSort);
setMaxMemoryUsage(properties.maxMemoryUsage);
setMaxMemoryUsageForUser(properties.maxMemoryUsageForUser);
setMaxMemoryUsageForAllQueries(properties.maxMemoryUsageForAllQueries);
setSessionCheck(properties.sessionCheck);
setSessionId(properties.sessionId);
setSessionTimeout(properties.sessionTimeout);
setInsertQuorum(properties.insertQuorum);
setInsertQuorumTimeout(properties.insertQuorumTimeout);
setSelectSequentialConsistency(properties.selectSequentialConsistency);
setPreferredBlockSizeBytes(properties.preferredBlockSizeBytes);
setMaxQuerySize(properties.maxQuerySize);
setMaxAstElements(properties.maxAstElements);
setEnableOptimizePredicateExpression(properties.enableOptimizePredicateExpression);
setMaxInsertBlockSize(properties.maxInsertBlockSize);
setInsertDeduplicate(properties.insertDeduplicate);
setInsertDistributedSync(properties.insertDistributedSync);
setAnyJoinDistinctRightTableKeys(properties.anyJoinDistinctRightTableKeys);
}
public Map<ClickHouseQueryParam, String> buildQueryParams(boolean ignoreDatabase){
Map<ClickHouseQueryParam, String> params = new HashMap<ClickHouseQueryParam, String>();
if (maxParallelReplicas != null) params.put(ClickHouseQueryParam.MAX_PARALLEL_REPLICAS, String.valueOf(maxParallelReplicas));
if (maxPartitionsPerInsertBlock != null) params.put(ClickHouseQueryParam.MAX_PARTITIONS_PER_INSERT_BLOCK, String.valueOf(maxPartitionsPerInsertBlock));
if (maxRowsToGroupBy != null) params.put(ClickHouseQueryParam.MAX_ROWS_TO_GROUP_BY, String.valueOf(maxRowsToGroupBy));
if (totalsMode != null) params.put(ClickHouseQueryParam.TOTALS_MODE, totalsMode);
if (quotaKey != null) params.put(ClickHouseQueryParam.QUOTA_KEY, quotaKey);
if (priority != null) params.put(ClickHouseQueryParam.PRIORITY, String.valueOf(priority));
if (!StringUtils.isBlank(database) && !ignoreDatabase) params.put(ClickHouseQueryParam.DATABASE, getDatabase());
if (compress) params.put(ClickHouseQueryParam.COMPRESS, "1");
if (decompress) params.put(ClickHouseQueryParam.DECOMPRESS, "1");
if (extremes) params.put(ClickHouseQueryParam.EXTREMES, "1");
if (StringUtils.isBlank(profile)) {
if (getMaxThreads() != null) {
params.put(ClickHouseQueryParam.MAX_THREADS, String.valueOf(maxThreads));
}
// in seconds there
if (getMaxExecutionTime() != null) {
params.put(ClickHouseQueryParam.MAX_EXECUTION_TIME, String.valueOf((maxExecutionTime)));
}
if (getMaxBlockSize() != null) {
params.put(ClickHouseQueryParam.MAX_BLOCK_SIZE, String.valueOf(getMaxBlockSize()));
}
} else {
params.put(ClickHouseQueryParam.PROFILE, profile);
}
if (user != null) params.put(ClickHouseQueryParam.USER, user);
if (password != null) params.put(ClickHouseQueryParam.PASSWORD, password);
if (distributedAggregationMemoryEfficient) params.put(ClickHouseQueryParam.DISTRIBUTED_AGGREGATION_MEMORY_EFFICIENT, "1");
if (maxBytesBeforeExternalGroupBy != null) params.put(ClickHouseQueryParam.MAX_BYTES_BEFORE_EXTERNAL_GROUP_BY, String.valueOf(maxBytesBeforeExternalGroupBy));
if (maxBytesBeforeExternalSort != null) params.put(ClickHouseQueryParam.MAX_BYTES_BEFORE_EXTERNAL_SORT, String.valueOf(maxBytesBeforeExternalSort));
if (maxMemoryUsage != null) {
params.put(ClickHouseQueryParam.MAX_MEMORY_USAGE, String.valueOf(maxMemoryUsage));
}
if (maxMemoryUsageForUser != null) {
params.put(ClickHouseQueryParam.MAX_MEMORY_USAGE_FOR_USER, String.valueOf(maxMemoryUsageForUser));
}
if (maxMemoryUsageForAllQueries != null) {
params.put(ClickHouseQueryParam.MAX_MEMORY_USAGE_FOR_ALL_QUERIES, String.valueOf(maxMemoryUsageForAllQueries));
}
if (preferredBlockSizeBytes != null) {
params.put(ClickHouseQueryParam.PREFERRED_BLOCK_SIZE_BYTES, String.valueOf(preferredBlockSizeBytes));
}
if (maxQuerySize != null) {
params.put(ClickHouseQueryParam.MAX_QUERY_SIZE, String.valueOf(maxQuerySize));
}
if (maxAstElements != null) {
params.put(ClickHouseQueryParam.MAX_AST_ELEMENTS, String.valueOf(maxAstElements));
}
if (sessionCheck) {
params.put(ClickHouseQueryParam.SESSION_CHECK, "1");
}
if (sessionId != null) {
params.put(ClickHouseQueryParam.SESSION_ID, String.valueOf(sessionId));
}
if (sessionTimeout != null) {
params.put(ClickHouseQueryParam.SESSION_TIMEOUT, String.valueOf(sessionTimeout));
}
addQueryParam(insertQuorum, ClickHouseQueryParam.INSERT_QUORUM, params);
addQueryParam(insertQuorumTimeout, ClickHouseQueryParam.INSERT_QUORUM_TIMEOUT, params);
addQueryParam(selectSequentialConsistency, ClickHouseQueryParam.SELECT_SEQUENTIAL_CONSISTENCY, params);
addQueryParam(maxInsertBlockSize, ClickHouseQueryParam.MAX_INSERT_BLOCK_SIZE, params);
addQueryParam(insertDeduplicate, ClickHouseQueryParam.INSERT_DEDUPLICATE, params);
addQueryParam(insertDistributedSync, ClickHouseQueryParam.INSERT_DISTRIBUTED_SYNC, params);
addQueryParam(anyJoinDistinctRightTableKeys, ClickHouseQueryParam.ANY_JOIN_DISTINCT_RIGHT_TABLE_KEYS, params);
if (enableOptimizePredicateExpression != null) {
params.put(ClickHouseQueryParam.ENABLE_OPTIMIZE_PREDICATE_EXPRESSION, enableOptimizePredicateExpression ? "1" : "0");
}
return params;
}
private void addQueryParam(Object param, ClickHouseQueryParam definition, Map<ClickHouseQueryParam, String> params) {
if (param != null) {
if (definition.getClazz() == Boolean.class || definition.getClazz() == boolean.class) {
params.put(definition, ((Boolean) param) ? "1" : "0");
} else {
params.put(definition, String.valueOf(param));
}
}
}
public ClickHouseProperties withCredentials(String user, String password){
ClickHouseProperties copy = new ClickHouseProperties(this);
copy.setUser(user);
copy.setPassword(password);
return copy;
}
private <T> T getSetting(Properties info, ClickHouseQueryParam param){
return getSetting(info, param.getKey(), param.getDefaultValue(), param.getClazz());
}
private <T> T getSetting(Properties info, ClickHouseConnectionSettings settings){
return getSetting(info, settings.getKey(), settings.getDefaultValue(), settings.getClazz());
}
@SuppressWarnings("unchecked")
private <T> T getSetting(Properties info, String key, Object defaultValue, Class clazz){
String val = info.getProperty(key);
if (val == null)
return (T)defaultValue;
if (clazz == int.class || clazz == Integer.class) {
return (T) clazz.cast(Integer.valueOf(val));
}
if (clazz == long.class || clazz == Long.class) {
return (T) clazz.cast(Long.valueOf(val));
}
if (clazz == boolean.class || clazz == Boolean.class) {
final Boolean boolValue;
if ("1".equals(val) || "0".equals(val)) {
boolValue = "1".equals(val);
} else {
boolValue = Boolean.valueOf(val);
}
return (T) clazz.cast(boolValue);
}
return (T) clazz.cast(val);
}
public String getProfile() {
return profile;
}
public void setProfile(String profile) {
this.profile = profile;
}
public boolean isCompress() {
return compress;
}
public void setCompress(boolean compress) {
this.compress = compress;
}
public boolean isDecompress() {
return decompress;
}
public void setDecompress(boolean decompress) {
this.decompress = decompress;
}
public boolean isAsync() {
return async;
}
public void setAsync(boolean async) {
this.async = async;
}
public Integer getMaxThreads() {
return maxThreads;
}
public void setMaxThreads(Integer maxThreads) {
this.maxThreads = maxThreads;
}
public Integer getMaxBlockSize() {
return maxBlockSize;
}
public void setMaxBlockSize(Integer maxBlockSize) {
this.maxBlockSize = maxBlockSize;
}
public int getBufferSize() {
return bufferSize;
}
public void setBufferSize(int bufferSize) {
this.bufferSize = bufferSize;
}
public int getApacheBufferSize() {
return apacheBufferSize;
}
public void setApacheBufferSize(int apacheBufferSize) {
this.apacheBufferSize = apacheBufferSize;
}
public int getSocketTimeout() {
return socketTimeout;
}
public void setSocketTimeout(int socketTimeout) {
this.socketTimeout = socketTimeout;
}
public int getConnectionTimeout() {
return connectionTimeout;
}
public void setConnectionTimeout(int connectionTimeout) {
this.connectionTimeout = connectionTimeout;
}
public int getDataTransferTimeout() {
return dataTransferTimeout;
}
public void setDataTransferTimeout(int dataTransferTimeout) {
this.dataTransferTimeout = dataTransferTimeout;
}
public int getKeepAliveTimeout() {
return keepAliveTimeout;
}
public void setKeepAliveTimeout(int keepAliveTimeout) {
this.keepAliveTimeout = keepAliveTimeout;
}
public String getUser() {
return user;
}
public void setUser(String user) {
this.user = user;
}
public int getTimeToLiveMillis() {
return timeToLiveMillis;
}
public void setTimeToLiveMillis(int timeToLiveMillis) {
this.timeToLiveMillis = timeToLiveMillis;
}
public int getDefaultMaxPerRoute() {
return defaultMaxPerRoute;
}
public void setDefaultMaxPerRoute(int defaultMaxPerRoute) {
this.defaultMaxPerRoute = defaultMaxPerRoute;
}
public int getMaxTotal() {
return maxTotal;
}
public void setMaxTotal(int maxTotal) {
this.maxTotal = maxTotal;
}
public int getMaxCompressBufferSize() {
return maxCompressBufferSize;
}
public void setMaxCompressBufferSize(int maxCompressBufferSize) {
this.maxCompressBufferSize = maxCompressBufferSize;
}
public boolean getSsl() {
return ssl;
}
public void setSsl(boolean ssl) {
this.ssl = ssl;
}
public String getSslRootCertificate() {
return sslRootCertificate;
}
public void setSslRootCertificate(String sslRootCertificate) {
this.sslRootCertificate = sslRootCertificate;
}
public String getSslMode() {
return sslMode;
}
public void setSslMode(String sslMode) {
this.sslMode = sslMode;
}
public int getMaxRedirects() {
return maxRedirects;
}
public void setMaxRedirects(int maxRedirects) {
this.maxRedirects = maxRedirects;
}
public boolean isCheckForRedirects() {
return checkForRedirects;
}
public void setCheckForRedirects(boolean checkForRedirects) {
this.checkForRedirects = checkForRedirects;
}
public boolean isUseServerTimeZone() {
return useServerTimeZone;
}
public void setUseServerTimeZone(boolean useServerTimeZone) {
this.useServerTimeZone = useServerTimeZone;
}
public String getUseTimeZone() {
return useTimeZone;
}
public void setUseTimeZone(String useTimeZone) {
this.useTimeZone = useTimeZone;
}
public boolean isUseObjectsInArrays() {
return useObjectsInArrays;
}
public void setUseObjectsInArrays(boolean useObjectsInArrays) {
this.useObjectsInArrays = useObjectsInArrays;
}
public boolean isUseServerTimeZoneForDates() {
return useServerTimeZoneForDates;
}
public void setUseServerTimeZoneForDates(boolean useServerTimeZoneForDates) {
this.useServerTimeZoneForDates = useServerTimeZoneForDates;
}
public Integer getMaxParallelReplicas() {
return maxParallelReplicas;
}
public void setMaxParallelReplicas(Integer maxParallelReplicas) {
this.maxParallelReplicas = maxParallelReplicas;
}
public Integer getMaxPartitionsPerInsertBlock() {
return maxPartitionsPerInsertBlock;
}
public void setMaxPartitionsPerInsertBlock(Integer maxPartitionsPerInsertBlock) {
this.maxPartitionsPerInsertBlock = maxPartitionsPerInsertBlock;
}
public String getTotalsMode() {
return totalsMode;
}
public void setTotalsMode(String totalsMode) {
this.totalsMode = totalsMode;
}
public String getQuotaKey() {
return quotaKey;
}
public void setQuotaKey(String quotaKey) {
this.quotaKey = quotaKey;
}
public Integer getPriority() {
return priority;
}
public void setPriority(Integer priority) {
this.priority = priority;
}
public String getDatabase() {
return database;
}
public void setDatabase(String database) {
this.database = database;
}
public boolean isExtremes() {
return extremes;
}
public void setExtremes(boolean extremes) {
this.extremes = extremes;
}
public Integer getMaxExecutionTime() {
return maxExecutionTime;
}
public void setMaxExecutionTime(Integer maxExecutionTime) {
this.maxExecutionTime = maxExecutionTime;
}
public Integer getMaxRowsToGroupBy() {
return maxRowsToGroupBy;
}
public void setMaxRowsToGroupBy(Integer maxRowsToGroupBy) {
this.maxRowsToGroupBy = maxRowsToGroupBy;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
public String getHttpAuthorization() {
return httpAuthorization;
}
public void setHttpAuthorization(String httpAuthorization) {
this.httpAuthorization = httpAuthorization;
}
public String getHost() {
return host;
}
public void setHost(String host) {
this.host = host;
}
public int getPort() {
return port;
}
public void setPort(int port) {
this.port = port;
}
public boolean isUsePathAsDb() {
return usePathAsDb;
}
public void setUsePathAsDb(boolean usePathAsDb) {
this.usePathAsDb = usePathAsDb;
}
public String getPath() {
return path;
}
public void setPath(String path) {
this.path = path;
}
public boolean isDistributedAggregationMemoryEfficient() {
return distributedAggregationMemoryEfficient;
}
public void setDistributedAggregationMemoryEfficient(boolean distributedAggregationMemoryEfficient) {
this.distributedAggregationMemoryEfficient = distributedAggregationMemoryEfficient;
}
public Long getMaxBytesBeforeExternalGroupBy() {
return maxBytesBeforeExternalGroupBy;
}
public void setMaxBytesBeforeExternalGroupBy(Long maxBytesBeforeExternalGroupBy) {
this.maxBytesBeforeExternalGroupBy = maxBytesBeforeExternalGroupBy;
}
public Long getMaxBytesBeforeExternalSort() {
return maxBytesBeforeExternalSort;
}
public void setMaxBytesBeforeExternalSort(Long maxBytesBeforeExternalSort) {
this.maxBytesBeforeExternalSort = maxBytesBeforeExternalSort;
}
public Long getMaxMemoryUsage() {
return maxMemoryUsage;
}
public void setMaxMemoryUsage(Long maxMemoryUsage) {
this.maxMemoryUsage = maxMemoryUsage;
}
public Long getMaxMemoryUsageForUser() {
return maxMemoryUsageForUser;
}
public void setMaxMemoryUsageForUser(Long maxMemoryUsageForUser) {
this.maxMemoryUsageForUser = maxMemoryUsageForUser;
}
public Long getMaxMemoryUsageForAllQueries() {
return maxMemoryUsageForAllQueries;
}
public void setMaxMemoryUsageForAllQueries(Long maxMemoryUsageForAllQueries) {
this.maxMemoryUsageForAllQueries = maxMemoryUsageForAllQueries;
}
public Long getPreferredBlockSizeBytes() {
return preferredBlockSizeBytes;
}
public void setPreferredBlockSizeBytes(Long preferredBlockSizeBytes) {
this.preferredBlockSizeBytes = preferredBlockSizeBytes;
}
public Long getMaxQuerySize() {
return maxQuerySize;
}
public void setMaxQuerySize(Long maxQuerySize) {
this.maxQuerySize = maxQuerySize;
}
public void setMaxAstElements(Long maxAstElements) {
this.maxAstElements = maxAstElements;
}
public Long getMaxAstElements() {
return this.maxAstElements;
}
public boolean isSessionCheck() { return sessionCheck; }
public void setSessionCheck(boolean sessionCheck) { this.sessionCheck = sessionCheck; }
public String getSessionId() { return sessionId; }
public void setSessionId(String sessionId) { this.sessionId = sessionId; }
public Long getSessionTimeout() { return sessionTimeout; }
public void setSessionTimeout(Long sessionTimeout) { this.sessionTimeout = sessionTimeout; }
public Long getInsertQuorum() {
return insertQuorum;
}
public void setInsertQuorum(Long insertQuorum) {
this.insertQuorum = insertQuorum;
}
public Long getInsertQuorumTimeout() {
return insertQuorumTimeout;
}
public void setInsertQuorumTimeout(Long insertQuorumTimeout) {
this.insertQuorumTimeout = insertQuorumTimeout;
}
public Long getSelectSequentialConsistency() {
return selectSequentialConsistency;
}
public void setSelectSequentialConsistency(Long selectSequentialConsistency) {
this.selectSequentialConsistency = selectSequentialConsistency;
}
public Boolean getEnableOptimizePredicateExpression() {
return enableOptimizePredicateExpression;
}
public void setEnableOptimizePredicateExpression(Boolean enableOptimizePredicateExpression) {
this.enableOptimizePredicateExpression = enableOptimizePredicateExpression;
}
public Long getMaxInsertBlockSize() {
return maxInsertBlockSize;
}
public void setMaxInsertBlockSize(Long maxInsertBlockSize) {
this.maxInsertBlockSize = maxInsertBlockSize;
}
public Boolean getInsertDeduplicate() {
return insertDeduplicate;
}
public void setInsertDeduplicate(Boolean insertDeduplicate) {
this.insertDeduplicate = insertDeduplicate;
}
public Boolean getInsertDistributedSync() {
return insertDistributedSync;
}
public void setInsertDistributedSync(Boolean insertDistributedSync) {
this.insertDistributedSync = insertDistributedSync;
}
public void setAnyJoinDistinctRightTableKeys(Boolean anyJoinDistinctRightTableKeys) {
this.anyJoinDistinctRightTableKeys = anyJoinDistinctRightTableKeys;
}
public Boolean getAnyJoinDistinctRightTableKeys() {
return anyJoinDistinctRightTableKeys;
}
private static class PropertiesBuilder {
private final Properties properties;
public PropertiesBuilder() {
properties = new Properties();
}
public void put(String key, int value) {
properties.put(key, value);
}
public void put(String key, Integer value) {
if (value != null) {
properties.put(key, value.toString());
}
}
public void put(String key, Long value) {
if (value != null) {
properties.put(key, value.toString());
}
}
public void put(String key, Boolean value) {
if (value != null) {
properties.put(key, value.toString());
}
}
public void put(String key, String value) {
if (value != null) {
properties.put(key, value);
}
}
public Properties getProperties() {
return properties;
}
}
public ClickHouseProperties merge(ClickHouseProperties second){
Properties properties = this.asProperties();
for (Map.Entry<Object, Object> entry : second.asProperties().entrySet())
properties.put(entry.getKey(), entry.getValue());
return new ClickHouseProperties(properties);
}
public ClickHouseProperties merge(Properties other){
Properties properties = this.asProperties();
for (Map.Entry<Object, Object> entry : other.entrySet())
properties.put(entry.getKey(), entry.getValue());
return new ClickHouseProperties(properties);
}
}
