diff Framework/Odbc/OdbcEnvironment.cpp @ 329:b5fb8b77ce4d

initial commit of ODBC framework
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 10 Aug 2021 20:08:53 +0200
parents
children 674bbb9d1c83
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Framework/Odbc/OdbcEnvironment.cpp	Tue Aug 10 20:08:53 2021 +0200
@@ -0,0 +1,97 @@
+/**
+ * Orthanc - A Lightweight, RESTful DICOM Store
+ * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
+ * Department, University Hospital of Liege, Belgium
+ * Copyright (C) 2017-2021 Osimis S.A., Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU Affero General Public License
+ * as published by the Free Software Foundation, either version 3 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Affero General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+
+#include "OdbcEnvironment.h"
+
+#include <Logging.h>
+#include <OrthancException.h>
+
+#include <boost/lexical_cast.hpp>
+#include <sqlext.h>
+
+
+namespace OrthancDatabases
+{
+  OdbcEnvironment::OdbcEnvironment()
+  {
+    LOG(INFO) << "Creating the ODBC environment";
+      
+    /* Allocate an environment handle */
+    if (!SQL_SUCCEEDED(SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &handle_)))
+    {
+      throw Orthanc::OrthancException(Orthanc::ErrorCode_Database,
+                                      "Cannot create ODBC environment");
+    }
+      
+    /* We want ODBC 3 support */
+    if (!SQL_SUCCEEDED(SQLSetEnvAttr(handle_, SQL_ATTR_ODBC_VERSION, (void *) SQL_OV_ODBC3, 0)))
+    {
+      SQLFreeHandle(SQL_HANDLE_ENV, handle_);
+      throw Orthanc::OrthancException(Orthanc::ErrorCode_Database,
+                                      "Your environment doesn't support ODBC 3.x");
+    }
+  }
+
+
+  OdbcEnvironment::~OdbcEnvironment()
+  {
+    LOG(INFO) << "Destructing the ODBC environment";
+      
+    if (!SQL_SUCCEEDED(SQLFreeHandle(SQL_HANDLE_ENV, handle_)))
+    {
+      LOG(ERROR) << "Cannot tear down ODBC environment";
+    }
+  }
+
+
+  std::string OdbcEnvironment::FormatError(SQLHANDLE handle,
+                                           SQLSMALLINT type)
+  {
+    SQLINTEGER   i = 0;
+    SQLINTEGER   native;
+    SQLCHAR      state[SQL_SQLSTATE_SIZE + 1];
+    SQLCHAR      text[256];
+    SQLSMALLINT  len;
+
+    std::string s;
+      
+    for (;;)
+    {
+      SQLRETURN ret = SQLGetDiagRec(type, handle, ++i, state, &native, text, sizeof(text), &len);
+      if (SQL_SUCCEEDED(ret))
+      {
+        if (i >= 2)
+        {
+          s += "\n";
+        }
+          
+        s += (std::string(reinterpret_cast<const char*>(state)) + " : " +
+              boost::lexical_cast<std::string>(i) + "/" +
+              boost::lexical_cast<std::string>(native) + " " +
+              std::string(reinterpret_cast<const char*>(text)));
+      }
+      else
+      {
+        return s;
+      }
+    }
+  }
+}