ClickHouse#24 add some error codes

Author: artpaul

clickhouse/error_codes.h

@@ -0,0 +1,29 @@
+#pragma once
+
+namespace clickhouse {
+
+enum ErrorCodes {
+    CHECKSUM_DOESNT_MATCH = 40,
+    CANNOT_PARSE_DATETIME = 41,
+    UNKNOWN_FUNCTION = 46,
+    UNKNOWN_IDENTIFIER = 47,
+    TABLE_ALREADY_EXISTS = 57,
+    UNKNOWN_TABLE = 60,
+    SYNTAX_ERROR = 62,
+    UNKNOWN_DATABASE = 81,
+    DATABASE_ALREADY_EXISTS = 82,
+    UNKNOWN_PACKET_FROM_CLIENT = 99,
+    UNEXPECTED_PACKET_FROM_CLIENT = 101,
+    RECEIVED_DATA_FOR_WRONG_QUERY_ID = 103,
+    ENGINE_REQUIRED = 119,
+    READONLY = 164,
+    UNKNOWN_USER = 192,
+    WRONG_PASSWORD = 193,
+    REQUIRED_PASSWORD = 194,
+    IP_ADDRESS_NOT_ALLOWED = 195,
+    LIMIT_EXCEEDED = 290,
+    UNKNOWN_DATABASE_ENGINE = 336,
+    UNKNOWN_EXCEPTION = 1002,
+};
+
+}

clickhouse/exceptions.h

@@ -14,6 +14,10 @@ class ServerException : public std::runtime_error {
     {
     }
 
+    int GetCode() const {
+        return exception_->code;
+    }
+
     const Exception& GetException() const {
         return *exception_;
     }

tests/simple/main.cpp

@@ -1,4 +1,5 @@
 #include <clickhouse/client.h>
+#include <clickhouse/error_codes.h>
 #include <clickhouse/types/type_parser.h>
 
 #include <iostream>
@@ -206,12 +207,31 @@ inline void NumbersExample(Client& client) {
     );
 }
 
+inline void ExecptionExample(Client& client) {
+    /// Create a table.
+    client.Execute("CREATE TABLE IF NOT EXISTS test.exceptions (id UInt64, name String) ENGINE = Memory");
+    /// Expect failing on table creation.
+    try {
+        client.Execute("CREATE TABLE test.exceptions (id UInt64, name String) ENGINE = Memory");
+    } catch (const ServerException& e) {
+        if (e.GetCode() == ErrorCodes::TABLE_ALREADY_EXISTS) {
+            // OK
+        } else {
+            throw;
+        }
+    }
+
+    /// Delete table.
+    client.Execute("DROP TABLE test.exceptions");
+}
+
 static void RunTests(Client& client) {
     ArrayExample(client);
     DateExample(client);
     GenericExample(client);
     NullableExample(client);
     NumbersExample(client);
+    ExecptionExample(client);
 }
 
 int main() {