com.clickhouse.client.http.ClickHouseHttpConnection Maven / Gradle / Ivy
package com.clickhouse.client.http;
import java.io.IOException;
import java.io.OutputStream;
import java.io.Serializable;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Optional;
import java.util.Map.Entry;
import com.clickhouse.client.ClickHouseClient;
import com.clickhouse.client.ClickHouseConfig;
import com.clickhouse.client.ClickHouseCredentials;
import com.clickhouse.client.ClickHouseNode;
import com.clickhouse.client.ClickHouseRequest;
import com.clickhouse.client.ClickHouseRequestManager;
import com.clickhouse.client.config.ClickHouseClientOption;
import com.clickhouse.client.http.config.ClickHouseHttpOption;
import com.clickhouse.config.ClickHouseOption;
import com.clickhouse.data.ClickHouseChecker;
import com.clickhouse.data.ClickHouseCompression;
import com.clickhouse.data.ClickHouseExternalTable;
import com.clickhouse.data.ClickHouseInputStream;
import com.clickhouse.data.ClickHouseOutputStream;
import com.clickhouse.data.ClickHouseUtils;
import com.clickhouse.logging.Logger;
import com.clickhouse.logging.LoggerFactory;
public abstract class ClickHouseHttpConnection implements AutoCloseable {
private static final Logger log = LoggerFactory.getLogger(ClickHouseHttpConnection.class);
private static final byte[] HEADER_CONTENT_DISPOSITION = "content-disposition: form-data; name=\""
.getBytes(StandardCharsets.US_ASCII);
private static final byte[] HEADER_OCTET_STREAM = "content-type: application/octet-stream\r\n"
.getBytes(StandardCharsets.US_ASCII);
private static final byte[] HEADER_BINARY_ENCODING = "content-transfer-encoding: binary\r\n\r\n"
.getBytes(StandardCharsets.US_ASCII);
private static final byte[] ERROR_MSG_PREFIX = "ode: ".getBytes(StandardCharsets.US_ASCII);
private static final byte[] DOUBLE_DASH = new byte[] { '-', '-' };
private static final byte[] END_OF_NAME = new byte[] { '"', '\r', '\n' };
private static final byte[] LINE_PREFIX = new byte[] { '\r', '\n', '-', '-' };
private static final byte[] LINE_SUFFIX = new byte[] { '\r', '\n' };
private static final byte[] SUFFIX_QUERY = "query\"\r\n\r\n".getBytes(StandardCharsets.US_ASCII);
private static final byte[] SUFFIX_FORMAT = "_format\"\r\n\r\n".getBytes(StandardCharsets.US_ASCII);
private static final byte[] SUFFIX_STRUCTURE = "_structure\"\r\n\r\n".getBytes(StandardCharsets.US_ASCII);
private static final byte[] SUFFIX_FILENAME = "\"; filename=\"".getBytes(StandardCharsets.US_ASCII);
private static StringBuilder appendQueryParameter(StringBuilder builder, String key, String value) {
return builder.append(urlEncode(key, StandardCharsets.UTF_8)).append('=')
.append(urlEncode(value, StandardCharsets.UTF_8)).append('&');
}
static String urlEncode(String str, Charset charset) {
if (charset == null) {
charset = StandardCharsets.UTF_8;
}
try {
return URLEncoder.encode(str, charset.name());
} catch (UnsupportedEncodingException e) {
// should not happen
throw new IllegalArgumentException(e);
}
}
static String buildQueryParams(ClickHouseRequest request) {
if (request == null) {
return "";
}
ClickHouseConfig config = request.getConfig();
StringBuilder builder = new StringBuilder();
// start with custom query parameters first
Map customParams = ClickHouseOption
.toKeyValuePairs(config.getStrOption(ClickHouseHttpOption.CUSTOM_PARAMS));
for (Entry cp : customParams.entrySet()) {
appendQueryParameter(builder, cp.getKey(), cp.getValue());
}
ClickHouseInputStream chIn = request.getInputStream().orElse(null);
if (chIn != null && chIn.getUnderlyingStream().hasInput()) {
appendQueryParameter(builder, "query", request.getStatements().get(0));
}
if (config.isRequestCompressed() && config.getRequestCompressAlgorithm() == ClickHouseCompression.LZ4) {
// inform server to decompress client request
appendQueryParameter(builder, "decompress", "1");
}
if (config.isResponseCompressed()) {
if (config.getResponseCompressAlgorithm() == ClickHouseCompression.LZ4) {
// request server to compress response
appendQueryParameter(builder, "compress", "1");
} else {
appendQueryParameter(builder, "enable_http_compression", "1");
}
}
Map settings = request.getSettings();
List stmts = request.getStatements(false);
String settingKey = ClickHouseClientOption.MAX_EXECUTION_TIME.getKey();
if (config.getMaxExecutionTime() > 0 && !settings.containsKey(settingKey)) {
appendQueryParameter(builder, settingKey, String.valueOf(config.getMaxExecutionTime()));
}
settingKey = ClickHouseClientOption.MAX_RESULT_ROWS.getKey();
if (config.getMaxResultRows() > 0L && !settings.containsKey(settingKey)) {
appendQueryParameter(builder, settingKey, String.valueOf(config.getMaxResultRows()));
appendQueryParameter(builder, "result_overflow_mode", "break");
}
settingKey = "log_comment";
if (!stmts.isEmpty() && config.getBoolOption(ClickHouseClientOption.LOG_LEADING_COMMENT)
&& !settings.containsKey(settingKey)) {
String comment = ClickHouseUtils.getLeadingComment(stmts.get(0));
if (!comment.isEmpty()) {
appendQueryParameter(builder, settingKey, comment);
}
}
settingKey = "extremes";
if (!settings.containsKey(settingKey)) {
appendQueryParameter(builder, settingKey, "0");
}
Optional optionalValue = request.getSessionId();
if (optionalValue.isPresent()) {
appendQueryParameter(builder, ClickHouseClientOption.SESSION_ID.getKey(), optionalValue.get());
if (config.isSessionCheck()) {
appendQueryParameter(builder, ClickHouseClientOption.SESSION_CHECK.getKey(), "1");
}
if (config.getSessionTimeout() > 0) {
// see default_session_timeout
appendQueryParameter(builder, ClickHouseClientOption.SESSION_TIMEOUT.getKey(),
String.valueOf(config.getSessionTimeout()));
}
}
optionalValue = request.getQueryId();
if (optionalValue.isPresent()) {
appendQueryParameter(builder, "query_id", optionalValue.get());
}
for (Map.Entry entry : settings.entrySet()) {
appendQueryParameter(builder, entry.getKey(), String.valueOf(entry.getValue()));
}
if (builder.length() > 0) {
builder.setLength(builder.length() - 1);
}
return builder.toString();
}
static String buildUrl(String baseUrl, ClickHouseRequest request) {
ClickHouseConfig config = request.getConfig();
StringBuilder builder = new StringBuilder().append(baseUrl);
String context = config.getStrOption(ClickHouseHttpOption.WEB_CONTEXT);
if (!ClickHouseChecker.isNullOrEmpty(context)) {
char prev = '/';
for (int i = 0, len = context.length(); i < len; i++) {
char ch = context.charAt(i);
if (ch == '?' || ch == '#') {
break;
} else if (ch != '/' || ch != prev) {
builder.append(ch);
}
prev = ch;
}
}
String query = buildQueryParams(request);
if (!query.isEmpty()) {
builder.append('?').append(query);
}
return builder.toString();
}
protected static Map createDefaultHeaders(ClickHouseConfig config, ClickHouseNode server,
String userAgent) {
Map map = new LinkedHashMap<>();
boolean hasAuthorizationHeader = false;
// add customer headers
for (Entry header : ClickHouseOption
.toKeyValuePairs(config.getStrOption(ClickHouseHttpOption.CUSTOM_HEADERS)).entrySet()) {
String name = header.getKey().toLowerCase(Locale.ROOT);
String value = header.getValue();
if (value == null) {
continue;
}
if ("authorization".equals(name)) {
hasAuthorizationHeader = true;
}
map.put(name, value);
}
map.put("accept", "*/*");
if (!config.getBoolOption(ClickHouseHttpOption.KEEP_ALIVE)) {
map.put("connection", "Close");
}
map.put("user-agent", !ClickHouseChecker.isNullOrEmpty(userAgent) ? userAgent : config.getClientName());
ClickHouseCredentials credentials = server.getCredentials(config);
if (credentials.useAccessToken()) {
// TODO check if auth-scheme is available and supported
map.put("authorization", credentials.getAccessToken());
} else if (!hasAuthorizationHeader) {
map.put("x-clickhouse-user", credentials.getUserName());
if (config.isSsl() && !ClickHouseChecker.isNullOrEmpty(config.getSslCert())) {
map.put("x-clickhouse-ssl-certificate-auth", "on");
} else if (!ClickHouseChecker.isNullOrEmpty(credentials.getPassword())) {
map.put("x-clickhouse-key", credentials.getPassword());
}
}
String database = server.getDatabase(config);
if (!ClickHouseChecker.isNullOrEmpty(database)) {
map.put("x-clickhouse-database", database);
}
// Also, you can use the ‘default_format’ URL parameter
map.put("x-clickhouse-format", config.getFormat().name());
if (config.isResponseCompressed()) {
map.put("accept-encoding", config.getResponseCompressAlgorithm().encoding());
}
if (config.isRequestCompressed()
&& config.getRequestCompressAlgorithm() != ClickHouseCompression.LZ4) {
map.put("content-encoding", config.getRequestCompressAlgorithm().encoding());
}
return map;
}
protected static String parseErrorFromException(String errorCode, String serverName, IOException e, byte[] bytes) {
log.debug("Failed to read error message[code=%s] from server [%s] due to: %s", errorCode, serverName,
e.getMessage());
int index = ClickHouseUtils.indexOf(bytes, ERROR_MSG_PREFIX);
final String errorMsg;
if (index > 0) {
bytes[--index] = (byte) 'C';
errorMsg = new String(bytes, index, bytes.length - index, StandardCharsets.UTF_8);
} else if (!ClickHouseChecker.isNullOrBlank(errorCode)) {
errorMsg = new StringBuilder().append("Code: ").append(errorCode).append(", server: ").append(serverName)
.append(", ").append(new String(bytes, StandardCharsets.UTF_8)).toString();
} else {
// uncomment to debug
// log.debug("Unparsable error message[code=%s] returned from server[%s]: %s",
// errorCode, serverName, ClickHouseUtils.toJavaByteArrayExpression(bytes));
errorMsg = new String(bytes, StandardCharsets.UTF_8);
}
return errorMsg;
}
protected static void postData(ClickHouseConfig config, byte[] boundary, String sql, ClickHouseInputStream data,
List tables, OutputStream requestStream) throws IOException {
final boolean hasCustomInput = data != null && data.getUnderlyingStream().hasInput();
try (OutputStream rawOut = requestStream;
ClickHouseOutputStream out = hasCustomInput
? ClickHouseOutputStream.of(rawOut, config.getWriteBufferSize())
: (data != null || boundary != null // NOSONAR
? ClickHouseClient.getAsyncRequestOutputStream(config, rawOut, null) // latch::countDown)
: ClickHouseClient.getRequestOutputStream(config, rawOut, null))) {
byte[] sqlBytes = hasCustomInput ? new byte[0] : sql.getBytes(StandardCharsets.UTF_8);
if (boundary != null) {
rawOut.write(LINE_PREFIX);
rawOut.write(boundary);
rawOut.write(LINE_SUFFIX);
rawOut.write(HEADER_CONTENT_DISPOSITION);
rawOut.write(SUFFIX_QUERY);
rawOut.write(sqlBytes);
final int writeBufferSize = config.getWriteBufferSize();
for (ClickHouseExternalTable t : tables) {
byte[] tableName = t.getName().getBytes(StandardCharsets.UTF_8);
for (int i = 0; i < 3; i++) {
rawOut.write(LINE_PREFIX);
rawOut.write(boundary);
rawOut.write(LINE_SUFFIX);
rawOut.write(HEADER_CONTENT_DISPOSITION);
rawOut.write(tableName);
if (i == 0) {
rawOut.write(SUFFIX_FORMAT);
rawOut.write(t.getFormat().name().getBytes(StandardCharsets.US_ASCII));
} else if (i == 1) {
rawOut.write(SUFFIX_STRUCTURE);
rawOut.write(t.getStructure().getBytes(StandardCharsets.UTF_8));
} else {
rawOut.write(SUFFIX_FILENAME);
rawOut.write(tableName);
rawOut.write(END_OF_NAME);
break;
}
}
rawOut.write(HEADER_OCTET_STREAM);
rawOut.write(HEADER_BINARY_ENCODING);
ClickHouseInputStream.pipe(t.getContent(), rawOut, writeBufferSize);
}
rawOut.write(LINE_PREFIX);
rawOut.write(boundary);
rawOut.write(DOUBLE_DASH);
rawOut.write(LINE_SUFFIX);
} else {
out.writeBytes(sqlBytes);
if (data != null && data.available() > 0) {
// append \n
if (sqlBytes.length > 0 && sqlBytes[sqlBytes.length - 1] != (byte) '\n') {
out.write(10);
}
ClickHouseInputStream.pipe(data, out, config.getWriteBufferSize());
}
}
}
}
protected final ClickHouseNode server;
protected final ClickHouseRequestManager rm;
protected final ClickHouseConfig config;
protected final Map defaultHeaders;
protected final String url;
protected ClickHouseHttpConnection(ClickHouseNode server, ClickHouseRequest request) {
if (server == null || request == null) {
throw new IllegalArgumentException("Non-null server and request are required");
}
this.server = server;
this.rm = request.getManager();
ClickHouseConfig c = request.getConfig();
this.config = c;
this.defaultHeaders = Collections.unmodifiableMap(createDefaultHeaders(c, server, getUserAgent()));
this.url = buildUrl(server.getBaseUri(), request);
}
protected void closeQuietly() {
try {
close();
} catch (Exception e) {
// ignore
}
}
protected String getBaseUrl() {
String u = url;
int index = u.indexOf('?');
if (index < 1) {
index = u.length();
}
String baseUrl = u.substring(0, index);
if (u.charAt(index - 1) != '/') {
baseUrl = baseUrl.concat("/");
}
return baseUrl;
}
protected String getDefaultUserAgent() {
return config.getClientName();
}
protected final String getUserAgent() {
final ClickHouseConfig c = config;
String name = c.getClientName();
if (!ClickHouseClientOption.CLIENT_NAME.getDefaultValue().equals(name)) {
return name;
}
String userAgent = getDefaultUserAgent();
name = c.getProductName();
return ClickHouseClientOption.PRODUCT_NAME.getDefaultValue().equals(name) ? userAgent
: new StringBuilder(name).append(userAgent.substring(userAgent.indexOf('/'))).toString();
}
/**
* Creates a merged map.
*
* @param requestHeaders request headers
* @return non-null merged headers
*/
protected Map mergeHeaders(Map requestHeaders) {
if (requestHeaders == null || requestHeaders.isEmpty()) {
return defaultHeaders;
} else if (isReusable()) {
return requestHeaders;
}
Map merged = new LinkedHashMap<>();
merged.putAll(defaultHeaders);
for (Entry header : requestHeaders.entrySet()) {
String name = header.getKey().toLowerCase(Locale.ROOT);
String value = header.getValue();
if (value == null) {
merged.remove(name);
} else {
merged.put(name, value);
}
}
return merged;
}
/**
* Posts query and data to server.
*
* @param query non-blank query
* @param config non-null configuration
* @param data optionally input stream for batch updating
* @param tables optionally external tables for query
* @param output optionally output stream
* @param url optionally url
* @param headers optionally request headers
* @param postCloseAction optionally post action
* @return response
* @throws IOException when error occured posting request and/or server failed
* to respond
*/
protected abstract ClickHouseHttpResponse post(ClickHouseConfig config, String query, ClickHouseInputStream data,
List tables, ClickHouseOutputStream output, String url,
Map headers, Runnable postCloseAction) throws IOException;
/**
* Checks whether the connection is reusable or not. This method will be called
* in
* {@link ClickHouseHttpClient#checkConnection(ClickHouseHttpConnection, ClickHouseNode, ClickHouseNode, ClickHouseRequest)}
* for making a decision of whether to create a new connection. In addition to
* that, if a connection is NOT reusable, it will be closed right after
* corresponding ClickHouseResponse is closed.
*
* @return true if it's reusable; false otherwise
*/
protected boolean isReusable() {
return true;
}
/**
* Sends a request to {@code /ping} for liveness detection.
*
* @param timeout timeout in millisecond
* @return true if server responded {@code Ok.}; false otherwise
*/
public abstract boolean ping(int timeout);
public ClickHouseHttpResponse update(String query) throws IOException {
return post(config, query, null, null, null, null, null, null);
}
public ClickHouseHttpResponse update(String query, Map headers) throws IOException {
return post(config, query, null, null, null, null, headers, null);
}
public ClickHouseHttpResponse update(String query, ClickHouseInputStream data) throws IOException {
return post(config, query, data, null, null, null, null, null);
}
public ClickHouseHttpResponse update(String query, ClickHouseInputStream data, Map headers)
throws IOException {
return post(config, query, data, null, null, null, headers, null);
}
public ClickHouseHttpResponse query(String query) throws IOException {
return post(config, query, null, null, null, null, null, null);
}
public ClickHouseHttpResponse query(String query, Map headers) throws IOException {
return post(config, query, null, null, null, null, headers, null);
}
public ClickHouseHttpResponse query(String query, List tables) throws IOException {
return post(config, query, null, tables, null, null, null, null);
}
public ClickHouseHttpResponse query(String query, List tables, Map headers)
throws IOException {
return post(config, query, null, tables, null, null, headers, null);
}
@Override
public void close() throws IOException {
// ignore
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy