Adding ClickHouseException as base Excpetion

Author: mzitnik

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

@@ -0,0 +1,18 @@
+package com.clickhouse.client.api;
+
+public class ClickHouseException extends RuntimeException {
+    protected boolean isRetryable  = false;
+
+    public ClickHouseException(String message) {
+        super(message);
+    }
+
+    public ClickHouseException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public ClickHouseException(Throwable cause) {
+        super(cause);
+    }
+    public boolean isRetryable() { return isRetryable; }
+}

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

@@ -12,8 +12,6 @@
 import com.clickhouse.client.api.data_formats.internal.ProcessParser;
 import com.clickhouse.client.api.enums.Protocol;
 import com.clickhouse.client.api.enums.ProxyType;
-import com.clickhouse.client.api.exception.ClientException;
-import com.clickhouse.client.api.exception.ServerException;
 import com.clickhouse.client.api.http.ClickHouseHttpProto;
 import com.clickhouse.client.api.insert.InsertResponse;
 import com.clickhouse.client.api.insert.InsertSettings;

…lient/api/exception/ClientException.java …ickhouse/client/api/ClientException.javaclient-v2/src/main/java/com/clickhouse/client/api/exception/ClientException.java renamed to client-v2/src/main/java/com/clickhouse/client/api/ClientException.java client-v2/src/main/java/com/clickhouse/client/api/exception/ClientException.java renamed to client-v2/src/main/java/com/clickhouse/client/api/ClientException.java

@@ -1,6 +1,6 @@
-package com.clickhouse.client.api.exception;
+package com.clickhouse.client.api;
 
-public class ClientException extends RuntimeException {
+public class ClientException extends ClickHouseException {
 
     public ClientException(String message) {
         super(message);

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

@@ -8,4 +8,5 @@ public enum ClientFaultCause {
     ConnectTimeout,
     ConnectionRequestTimeout,
     SocketTimeout,
+    ServerRetryable,
 }

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

@@ -1,7 +1,5 @@
 package com.clickhouse.client.api;
 
-import com.clickhouse.client.api.exception.ClientException;
-
 /**
  * Represents errors caused by a client misconfiguration.
  */

…ption/ConnectionInitiationException.java …t/api/ConnectionInitiationException.javaclient-v2/src/main/java/com/clickhouse/client/api/exception/ConnectionInitiationException.java renamed to client-v2/src/main/java/com/clickhouse/client/api/ConnectionInitiationException.java client-v2/src/main/java/com/clickhouse/client/api/exception/ConnectionInitiationException.java renamed to client-v2/src/main/java/com/clickhouse/client/api/ConnectionInitiationException.java

@@ -1,12 +1,14 @@
-package com.clickhouse.client.api.exception;
+package com.clickhouse.client.api;
 
-public class ConnectionInitiationException extends ClientException {
+public class ConnectionInitiationException extends ClickHouseException {
 
     public ConnectionInitiationException(String message) {
         super(message);
+        this.isRetryable = true;
     }
 
     public ConnectionInitiationException(String message, Throwable cause) {
         super(message, cause);
+        this.isRetryable = true;
     }
 }

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

@@ -0,0 +1,12 @@
+package com.clickhouse.client.api;
+
+public class DataTransferException extends ClickHouseException {
+
+    public DataTransferException(String message) {
+        super(message);
+    }
+
+    public DataTransferException(String message, Throwable cause) {
+        super(message, cause);
+    }
+}

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

@@ -0,0 +1,75 @@
+package com.clickhouse.client.api;
+
+public class ServerException extends ClickHouseException {
+
+    public static final int CODE_UNKNOWN = 0;
+
+    public static final int TABLE_NOT_FOUND = 60;
+
+    private final int code;
+
+    private final int transportProtocolCode;
+
+    public ServerException(int code, String message) {
+        this(code, message, 500);
+    }
+
+    public ServerException(int code, String message, int transportProtocolCode) {
+        super(message);
+        this.code = code;
+        this.transportProtocolCode = transportProtocolCode;
+        this.isRetryable = discoverIsRetryable(code, message, transportProtocolCode);
+    }
+
+    /**
+     * Returns CH server error code. May return 0 if code is unknown.
+     * @return - error code from server response
+     */
+    public int getCode() {
+        return code;
+    }
+
+    /**
+     * Returns error code of underlying transport protocol. For example, HTTP status.
+     * By default, will return {@code 500 } what is derived from HTTP Server Internal Error.
+     *
+     * @return - transport status code
+     */
+    public int getTransportProtocolCode() {
+        return transportProtocolCode;
+    }
+
+    public boolean isRetryable() {
+        return isRetryable;
+    }
+
+    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
+            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
+            case 209: // SOCKET_TIMEOUT
+            case 210: // NETWORK_ERROR
+            case 241: // MEMORY_LIMIT_EXCEEDED
+            case 242: // TABLE_IS_READ_ONLY
+            case 252: // TOO_MANY_PARTS
+            case 285: // TOO_FEW_LIVE_REPLICAS
+            case 319: // UNKNOWN_STATUS_OF_INSERT
+            case 425: // SYSTEM_ERROR
+            case 999: // KEEPER_EXCEPTION
+                return true;
+            case 0:
+                switch (transportProtocolCode) { // Bad request
+                    case 400: // Bad request
+                    case 500: // Internal server error
+                        return true;
+                };
+        };
+        return false;
+    }
+}

client-v2/src/main/java/com/clickhouse/client/api/command/CommandResponse.java

@@ -1,6 +1,6 @@
 package com.clickhouse.client.api.command;
 
-import com.clickhouse.client.api.exception.ClientException;
+import com.clickhouse.client.api.ClientException;
 import com.clickhouse.client.api.metrics.OperationMetrics;
 import com.clickhouse.client.api.metrics.ServerMetrics;
 import com.clickhouse.client.api.query.QueryResponse;

client-v2/src/main/java/com/clickhouse/client/api/data_formats/NativeFormatReader.java

@@ -1,6 +1,6 @@
 package com.clickhouse.client.api.data_formats;
 
-import com.clickhouse.client.api.exception.ClientException;
+import com.clickhouse.client.api.ClientException;
 import com.clickhouse.client.api.data_formats.internal.AbstractBinaryFormatReader;
 import com.clickhouse.client.api.data_formats.internal.BinaryStreamReader;
 import com.clickhouse.client.api.metadata.TableSchema;