SqlException class

ClickHouse · Apr 26, 2017 · 8915749 · 8915749
1 parent be39186
commit 8915749
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/driver/attr.cpp b/driver/attr.cpp
@@ -110,7 +110,7 @@ impl_SQLSetConnectAttr(SQLHDBC connection_handle, SQLINTEGER attribute,
                 return SQL_SUCCESS;
 
             default:
-                throw std::runtime_error("Unsupported connection attribute.");
+                throw SqlException("Unsupported connection attribute.", "HY092");
         }
     });
 }

diff --git a/driver/diagnostics.cpp b/driver/diagnostics.cpp
@@ -1,7 +1,5 @@
 #include "diagnostics.h"
 
-#include <stdexcept>
-
 DiagnosticRecord::DiagnosticRecord()
 {
     reset();
@@ -13,6 +11,12 @@ void DiagnosticRecord::fromException()
     {
         throw;
     }
+    catch (const SqlException & e)
+    {
+        message = e.what();
+        native_error_code = 1;
+        sql_state = e.sqlState();
+    }
     catch (const std::exception & e)
     {
         message = e.what();

diff --git a/driver/diagnostics.h b/driver/diagnostics.h
@@ -3,6 +3,26 @@
 #include "log.h"
 #include "platform.h"
 
+#include <stdexcept>
+
+class SqlException : public std::runtime_error
+{
+public:
+    SqlException(const std::string & message_, const std::string & state_ = "HY000")
+        : std::runtime_error(message_)
+        , state(state_)
+    {
+    }
+
+    std::string sqlState() const
+    {
+        return state;
+    }
+
+private:
+    const std::string state;
+};
+
 struct DiagnosticRecord
 {
     SQLINTEGER native_error_code;