Port of 0.9.9 patch to main

CHANGELOG.md

@@ -734,6 +734,16 @@ of `NULL` was not set and read. (https://github.com/ClickHouse/clickhouse-java/i
   and `getObject(column, Object.class)` and the no-type `getObject(column)` overloads now return a decoded `String`
   instead of the internal holder.
 
+## 0.9.9
+
+### Bug Fixes 
+
+- **[client-v2]** `ServerException` with code `159 Execution Timeout` is retried unconditionally. After the fix this
+error treated as non-retriable. (part of https://github.com/ClickHouse/clickhouse-java/issues/2637)
+- **[jdbc-v2]** Fixes `Statement#setQueryTimeout`. By default, client executes query in calling thread and future timeout 
+has no effect. Fix makes `setQueryTimeout` to set `max_execution_time` server setting in this case to overcome limitation.
+  (https://github.com/ClickHouse/clickhouse-java/issues/2637)
+
 ## 0.9.8
 
 ### Improvements 

client-v2/src/main/java/com/clickhouse/client/api/ServerException.java

@@ -8,6 +8,8 @@
 
     public static final int UNKNOWN_SETTING = 115;
 
+    public static final int EXECUTION_TIMEOUT = 159;
+
     private final int code;
 
     private final int transportProtocolCode;
@@ -56,10 +58,9 @@
     private boolean discoverIsRetryable(int code, String message, int transportProtocolCode) {
         // Let's check if we have a ServerException to reference the error code
         // https://github.com/ClickHouse/ClickHouse/blob/master/src/Common/ErrorCodes.cpp
-        switch (code) { // UNEXPECTED_END_OF_FILE
+        switch (code) {
             case 3: // UNEXPECTED_END_OF_FILE
             case 107: // FILE_DOESNT_EXIST
-            case 159: // TIMEOUT_EXCEEDED
             case 164: // READONLY
             case 202: // TOO_MANY_SIMULTANEOUS_QUERIES
             case 203: // NO_FREE_CONNECTION

client-v2/src/test/java/com/clickhouse/client/ClientTests.java

@@ -709,6 +709,24 @@
         Assert.assertEquals(actualIds, new ArrayList<>(queryIds));
     }
 
+    @Test(groups = {"integration"})
+    public void testExecutionTimeout() throws Exception{
+        final String query = "SELECT count(), sum(sipHash64(number)) " +
+                "FROM numbers(1000000000) " +
+                "SETTINGS max_threads = 1;";
+
+        long startTime = System.currentTimeMillis();
+        int maxExecTime = 4000;
+        try (Client client = newClient().serverSetting("max_execution_time",  String.valueOf(TimeUnit.MILLISECONDS.toSeconds(maxExecTime))).build();
+             QueryResponse response = client.query(query).get()) {
+
+        } catch (ServerException e) {
+            long queryTime = System.currentTimeMillis() - startTime;
+            Assert.assertTrue(Math.abs(queryTime - maxExecTime) < 1000);
+            Assert.assertEquals(e.getCode(), ServerException.EXECUTION_TIMEOUT);
+        }
+    }
+
     @Test(groups = {"integration"})
     public void testHostnameWithUnderscore() throws Exception {
 

client-v2/src/test/java/com/clickhouse/client/api/internal/HttpAPIClientHelperTest.java

@@ -341,8 +341,7 @@ public void testExecuteRequestThrowsConnectExceptionOn503() throws Exception {
 
     @DataProvider(name = "serverExceptionRetryCases")
     public static Object[][] serverExceptionRetryCases() {
-        // Server code 159 (TIMEOUT_EXCEEDED) is retryable; code 60 (TABLE_NOT_FOUND) is not.
-        ServerException retryable = new ServerException(159, "TIMEOUT_EXCEEDED", 500, "q1");
+        ServerException retryable = new ServerException(202, "TOO_MANY_SIMULTANEOUS_QUERIES", 500, "q1");
         ServerException nonRetryable = new ServerException(60, "TABLE_NOT_FOUND", 404, "q2");
         return new Object[][]{
                 // ServerException thrown directly (behaviour that already worked; pinned as contrast).

client-v2/src/test/java/com/clickhouse/client/metrics/MetricsTest.java

@@ -41,7 +41,7 @@ void tearDown() {
         meterRegistry.clear();
         Metrics.globalRegistry.clear();
     }
-    
+
     @Test(groups = {"integration"}, enabled = true)
     public void testRegisterMetrics() throws Exception {
         ClickHouseNode node = getServer(ClickHouseProtocol.HTTP);

jdbc-v2/src/main/java/com/clickhouse/jdbc/ConnectionImpl.java

@@ -76,7 +76,7 @@
 
     private final JsonParserFactory jsonParserFactory;
 
     public ConnectionImpl(String url, Properties info) throws SQLException {
         try {
             this.url = url;//Raw URL
             this.config = new JdbcConfiguration(url, info);
@@ -122,6 +122,20 @@
             this.schema = client.getDefaultDatabase();
             this.defaultQuerySettings = new QuerySettings();
 
+            String defaultQuerySettingsProp = config.getDriverProperty(DriverProperties.DEFAULT_QUERY_SETTINGS.getKey(), null);
+            if (defaultQuerySettingsProp != null) {
+                ClientConfigProperties.toKeyValuePairs(defaultQuerySettingsProp)
+                        .forEach((k, v) -> this.defaultQuerySettings.serverSetting(k, v));
+            }
+            Map<String, String> clientProps = config.getClientProperties();
+            for (Map.Entry<String, String> entry : clientProps.entrySet()) {
+                if (entry.getKey() != null && entry.getValue() != null) {
+                    if (entry.getKey().startsWith(ClientConfigProperties.SERVER_SETTING_PREFIX)) {
+                        this.defaultQuerySettings.setOption(entry.getKey(), entry.getValue());
+                    }
+                }
+            }
+
             this.metadata = new DatabaseMetaDataImpl(this, false, url);
             this.defaultCalendar = Calendar.getInstance();
 

jdbc-v2/src/main/java/com/clickhouse/jdbc/StatementImpl.java

@@ -1,6 +1,7 @@
 package com.clickhouse.jdbc;
 
 import com.clickhouse.client.api.ClientConfigProperties;
+import com.clickhouse.client.api.ServerException;
 import com.clickhouse.client.api.data_formats.ClickHouseFormatReader;
 import com.clickhouse.client.api.data_formats.JSONEachRowFormatReader;
 import com.clickhouse.client.api.internal.ServerSettings;
@@ -15,17 +16,15 @@
 import org.slf4j.LoggerFactory;
 
 import java.net.SocketTimeoutException;
-import java.sql.ResultSet;
-import java.sql.SQLException;
-import java.sql.SQLWarning;
-import java.sql.Statement;
+import java.sql.*;
 import java.util.ArrayDeque;
 import java.util.ArrayList;
 import java.util.Deque;
 import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.ConcurrentLinkedQueue;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
 import java.util.function.Supplier;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -66,6 +65,8 @@
     // settings local to a statement
     protected QuerySettings localSettings;
 
+    protected Integer connectionLvlExecTimeout; // to properly reset
+
 
     public StatementImpl(ConnectionImpl connection) throws SQLException {
         this.connection = connection;
@@ -79,6 +80,8 @@
         this.escapeProcessingEnabled = true;
         this.featureManager = new FeatureManager(connection.getJdbcConfig());
         this.queryIdGenerator = connection.getJdbcConfig().getQueryIdGenerator();
+
+        this.connectionLvlExecTimeout = connection.getDefaultQuerySettings().getMaxExecutionTime();
     }
 
     protected void ensureOpen() throws SQLException {
@@ -360,6 +363,7 @@
             }
             handleSocketTimeoutException(e);
             onResultSetClosed(null);
+            throwOnExecutionTimeout(e, mergedSettings.getQueryId());
             throw ExceptionUtils.toSqlState(e);
         }
     }
@@ -370,6 +374,18 @@
         }
     }
 
+    protected void throwOnExecutionTimeout(Exception e, String queryId) throws SQLTimeoutException {
+        boolean shouldThrow = e instanceof TimeoutException || e.getCause() instanceof TimeoutException;
+        ServerException se = e instanceof ServerException ? (ServerException) e : e.getCause() instanceof ServerException ? (ServerException) e.getCause() : null;
+        if (se != null && se.getCode() == ServerException.EXECUTION_TIMEOUT) {
+            shouldThrow = true;
+        }
+
+        if (shouldThrow) {
+            throw new SQLTimeoutException("Query execution time exceeded limit (queryId=" + queryId + ")", e);
+        }
+    }
+
     @Override
     public int executeUpdate(String sql) throws SQLException {
         ensureOpen();
@@ -396,6 +412,7 @@
             lastQueryId = response.getQueryId();
         } catch (Exception e) {
             handleSocketTimeoutException(e);
+            throwOnExecutionTimeout(e, mergedSettings.getQueryId());
             throw ExceptionUtils.toSqlState(e);
         }
 
@@ -469,6 +486,30 @@
     @Override
     public void setQueryTimeout(int seconds) throws SQLException {
         ensureOpen();
+        if (seconds < 0) {
+            throw new SQLException("Timeout should be >= 0 but " + seconds + " was passed");
+        }
+
+        if (seconds > 0) {
+            boolean isAsyncEnabled;
+            try {
+                isAsyncEnabled = Boolean.parseBoolean(
+                        getConnection().getClient().getConfiguration().getOrDefault(ClientConfigProperties.ASYNC_OPERATIONS.getKey(),
+                        ClientConfigProperties.ASYNC_OPERATIONS.getDefaultValue()));
+            } catch (Exception e) {
+                LOG.error("Failed to read client configuration " + ClientConfigProperties.ASYNC_OPERATIONS.getKey(), e);
+                isAsyncEnabled = false;
+            }
+
+            if (!isAsyncEnabled) {
+                // `max_execution_time` is only option when not async operations enabled
+                getLocalSettings().setMaxExecutionTime(seconds);
+            }
+        } else if (connectionLvlExecTimeout != null) {
+            getLocalSettings().setMaxExecutionTime(connectionLvlExecTimeout);
+        } else {
+            getLocalSettings().resetOption(ClientConfigProperties.serverSetting(ServerSettings.MAX_EXECUTION_TIME));
+        }
         queryTimeout = seconds;
     }
 

jdbc-v2/src/main/java/com/clickhouse/jdbc/metadata/DatabaseMetaDataImpl.java

@@ -794,26 +794,26 @@ public ResultSet getProcedureColumns(String catalog, String schemaPattern, Strin
     static final Map<String, String> ENGINE_TO_TABLE_TYPE;
     static {
         Map<String, String> map = new java.util.HashMap<>();
-        
+
         // Log tables
         map.put("Log", TableType.LOG_TABLE.getTypeName());
         map.put("StripeLog", TableType.LOG_TABLE.getTypeName());
         map.put("TinyLog", TableType.LOG_TABLE.getTypeName());
-        
+
         // Memory tables
         map.put("Buffer", TableType.MEMORY_TABLE.getTypeName());
         map.put("Memory", TableType.MEMORY_TABLE.getTypeName());
         map.put("Set", TableType.MEMORY_TABLE.getTypeName());
-        
+
         // Views
         map.put("View", TableType.VIEW.getTypeName());
         map.put("LiveView", TableType.VIEW.getTypeName());
         map.put("MaterializedView", TableType.MATERIALIZED_VIEW.getTypeName());
         map.put("WindowView", TableType.VIEW.getTypeName());
-        
+
         // Dictionary
         map.put("Dictionary", TableType.DICTIONARY.getTypeName());
-        
+
         // Remote/External tables
         map.put("AzureBlobStorage", TableType.REMOTE_TABLE.getTypeName());
         map.put("AzureQueue", TableType.REMOTE_TABLE.getTypeName());
@@ -899,6 +899,18 @@ public ResultSet getProcedureColumns(String catalog, String schemaPattern, Strin
         // Remote engines (appended 08/31/2026)
         map.put("BigQuery", TableType.REMOTE_TABLE.getTypeName());
 
+        // Paimon (appended 05/27/2026)
+        map.put("Paimon", TableType.REMOTE_TABLE.getTypeName());
+        map.put("PaimonAzure", TableType.REMOTE_TABLE.getTypeName());
+        map.put("PaimonHDFS", TableType.REMOTE_TABLE.getTypeName());
+        map.put("PaimonLocal", TableType.REMOTE_TABLE.getTypeName());
+        map.put("PaimonS3", TableType.REMOTE_TABLE.getTypeName());
+
+        // Remote engines (appended 07/21/2026)
+        map.put("QueryRunner", TableType.REMOTE_TABLE.getTypeName());
+        map.put("Remote", TableType.REMOTE_TABLE.getTypeName());
+        map.put("RemoteSecure", TableType.REMOTE_TABLE.getTypeName());
+
         // Special
         map.put("TimeSeries", TableType.TABLE.getTypeName());
         map.put("Null", TableType.TABLE.getTypeName());