diff Framework/PostgreSQL/PostgreSQLParameters.cpp @ 0:7cea966b6829

initial commit
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 04 Jul 2018 08:16:29 +0200
parents
children 3686ba3f9cdb
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Framework/PostgreSQL/PostgreSQLParameters.cpp	Wed Jul 04 08:16:29 2018 +0200
@@ -0,0 +1,193 @@
+/**
+ * Orthanc - A Lightweight, RESTful DICOM Store
+ * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
+ * Department, University Hospital of Liege, Belgium
+ * Copyright (C) 2017-2018 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 "PostgreSQLParameters.h"
+
+#include <Core/Logging.h>
+#include <Core/OrthancException.h>
+
+#include <boost/lexical_cast.hpp>
+
+
+namespace OrthancDatabases
+{
+  void PostgreSQLParameters::Reset()
+  {
+    host_ = "localhost";
+    port_ = 5432;
+    username_ = "";
+    password_ = "";
+    database_.clear();
+    uri_.clear();
+    lock_ = true;
+  }
+
+
+  PostgreSQLParameters::PostgreSQLParameters()
+  {
+    Reset();
+  }
+
+
+  PostgreSQLParameters::PostgreSQLParameters(const OrthancPlugins::OrthancConfiguration& configuration)
+  {
+    Reset();
+
+    std::string s;
+
+    if (configuration.LookupStringValue(s, "ConnectionUri"))
+    {
+      SetConnectionUri(s);
+    }
+    else
+    {
+      if (configuration.LookupStringValue(s, "Host"))
+      {
+        SetHost(s);
+      }
+
+      unsigned int port;
+      if (configuration.LookupUnsignedIntegerValue(port, "Port"))
+      {
+        SetPortNumber(port);
+      }
+
+      if (configuration.LookupStringValue(s, "Database"))
+      {
+        SetDatabase(s);
+      }
+
+      if (configuration.LookupStringValue(s, "Username"))
+      {
+        SetUsername(s);
+      }
+
+      if (configuration.LookupStringValue(s, "Password"))
+      {
+        SetPassword(s);
+      }
+    }
+
+    lock_ = configuration.GetBooleanValue("Lock", true);  // Use locking by default
+  }
+
+
+  void PostgreSQLParameters::SetConnectionUri(const std::string& uri)
+  {
+    uri_ = uri;
+  }
+
+
+  std::string PostgreSQLParameters::GetConnectionUri() const
+  {
+    if (uri_.empty())
+    {
+      std::string actualUri = "postgresql://";
+
+      if (!username_.empty())
+      {
+        actualUri += username_;
+
+        if (!password_.empty())
+        {
+          actualUri += ":" + password_;
+        }
+
+        actualUri += "@" + host_;
+      }
+      else
+      {
+        actualUri += host_;
+      }
+      
+      if (port_ > 0)
+      {
+        actualUri += ":" + boost::lexical_cast<std::string>(port_);
+      }
+
+      actualUri += "/" + database_;
+
+      return actualUri;
+    }
+    else
+    {
+      return uri_;
+    }
+  }
+
+
+  void PostgreSQLParameters::SetHost(const std::string& host)
+  {
+    uri_.clear();
+    host_ = host;
+  }
+
+  void PostgreSQLParameters::SetPortNumber(unsigned int port)
+  {
+    if (port <= 0 ||
+        port >= 65535)
+    {
+      throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
+    }
+
+    uri_.clear();
+    port_ = port;
+  }
+
+  void PostgreSQLParameters::SetUsername(const std::string& username)
+  {
+    uri_.clear();
+    username_ = username;
+  }
+
+  void PostgreSQLParameters::SetPassword(const std::string& password)
+  {
+    uri_.clear();
+    password_ = password;
+  }
+
+  void PostgreSQLParameters::SetDatabase(const std::string& database)
+  {
+    uri_.clear();
+    database_ = database;
+  }
+
+  void PostgreSQLParameters::Format(std::string& target) const
+  {
+    if (uri_.empty())
+    {
+      target = std::string("sslmode=disable") +  // TODO WHY SSL DOES NOT WORK? ("SSL error: wrong version number")
+        " user=" + username_ + 
+        " password=" + password_ + 
+        " host=" + host_ + 
+        " port=" + boost::lexical_cast<std::string>(port_);
+
+      if (database_.size() > 0)
+      {
+        target += " dbname=" + database_;
+      }
+    }
+    else
+    {
+      target = uri_;
+    }
+  }
+}