From 46c49ea73911327e2967198d08038ef4b947267d Mon Sep 17 00:00:00 2001 From: serebrserg <serebrserg@yandex-team.ru> Date: Tue, 22 Mar 2016 13:07:50 +0300 Subject: [PATCH] METR-20494: add executeQuery methods with and without additionalDBParams --- .../ru/yandex/metrika/clickhouse/CHStatement.java | 3 +++ .../yandex/metrika/clickhouse/CHStatementImpl.java | 12 ++++++++++-- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/src/main/java/ru/yandex/metrika/clickhouse/CHStatement.java b/src/main/java/ru/yandex/metrika/clickhouse/CHStatement.java index b79d1644..bd41cf46 100644 --- a/src/main/java/ru/yandex/metrika/clickhouse/CHStatement.java +++ b/src/main/java/ru/yandex/metrika/clickhouse/CHStatement.java @@ -2,6 +2,7 @@ package ru.yandex.metrika.clickhouse; import ru.yandex.metrika.clickhouse.copypaste.ClickhouseResponse; +import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Statement; import java.util.Map; @@ -12,5 +13,7 @@ import java.util.Map; */ public interface CHStatement extends Statement { ClickhouseResponse executeQueryClickhouseResponse(String sql) throws SQLException; + ClickhouseResponse executeQueryClickhouseResponse(String sql, Map<String, String> additionalDBParams) throws SQLException; ClickhouseResponse executeQueryClickhouseResponse(String sql, Map<String, String> additionalDBParams, boolean ignoreDatabase) throws SQLException; + ResultSet executeQuery(String sql, Map<String, String> additionalDBParams) throws SQLException; } diff --git a/src/main/java/ru/yandex/metrika/clickhouse/CHStatementImpl.java b/src/main/java/ru/yandex/metrika/clickhouse/CHStatementImpl.java index 97616da0..ecd49011 100644 --- a/src/main/java/ru/yandex/metrika/clickhouse/CHStatementImpl.java +++ b/src/main/java/ru/yandex/metrika/clickhouse/CHStatementImpl.java @@ -61,7 +61,11 @@ public class CHStatementImpl implements CHStatement { @Override public ResultSet executeQuery(String sql) throws SQLException { - InputStream is = getInputStream(sql, null, false); + return executeQuery(sql, null); + } + + public ResultSet executeQuery(String sql, Map<String, String> additionalDBParams) throws SQLException { + InputStream is = getInputStream(sql, additionalDBParams, false); try { currentResult = new CHResultSet(properties.isCompress() ? new ClickhouseLZ4Stream(is) : is, properties.getBufferSize(), @@ -76,7 +80,11 @@ public class CHStatementImpl implements CHStatement { } public ClickhouseResponse executeQueryClickhouseResponse(String sql) throws SQLException { - return executeQueryClickhouseResponse(sql, null, false); + return executeQueryClickhouseResponse(sql, null); + } + + public ClickhouseResponse executeQueryClickhouseResponse(String sql, Map<String, String> additionalDBParams) throws SQLException { + return executeQueryClickhouseResponse(sql, additionalDBParams, false); } public ClickhouseResponse executeQueryClickhouseResponse(String sql, Map<String, String> additionalDBParams, boolean ignoreDatabase) throws SQLException { -- GitLab