Mercurial > hg > orthanc-databases
annotate Framework/Plugins/GlobalProperties.cpp @ 132:5c69c2a14c97 OrthancPostgreSQL-3.2
OrthancPostgreSQL-3.2
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Fri, 01 Mar 2019 12:51:37 +0100 |
| parents | 714c5d2bee76 |
| children | 4cd7e45b671e |
| rev | line source |
|---|---|
| 0 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
| 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
| 4 * Department, University Hospital of Liege, Belgium | |
| 67 | 5 * Copyright (C) 2017-2019 Osimis S.A., Belgium |
| 0 | 6 * |
| 7 * This program is free software: you can redistribute it and/or | |
| 8 * modify it under the terms of the GNU Affero General Public License | |
| 9 * as published by the Free Software Foundation, either version 3 of | |
| 10 * the License, or (at your option) any later version. | |
| 11 * | |
| 12 * This program is distributed in the hope that it will be useful, but | |
| 13 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 15 * Affero General Public License for more details. | |
| 16 * | |
| 17 * You should have received a copy of the GNU Affero General Public License | |
| 18 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 19 **/ | |
| 20 | |
| 21 | |
| 22 #include "GlobalProperties.h" | |
| 23 | |
| 24 #include "../Common/Utf8StringValue.h" | |
| 25 | |
| 26 #include <Core/Logging.h> | |
| 27 #include <Core/OrthancException.h> | |
| 28 | |
| 29 #include <boost/lexical_cast.hpp> | |
| 30 | |
| 31 namespace OrthancDatabases | |
| 32 { | |
| 33 bool LookupGlobalProperty(std::string& target, | |
| 34 IDatabase& db, | |
| 35 ITransaction& transaction, | |
| 36 Orthanc::GlobalProperty property) | |
| 37 { | |
| 38 Query query("SELECT value FROM GlobalProperties WHERE property=${property}", true); | |
| 39 query.SetType("property", ValueType_Integer64); | |
| 40 | |
| 41 std::auto_ptr<IPrecompiledStatement> statement(db.Compile(query)); | |
| 42 | |
| 43 Dictionary args; | |
| 44 args.SetIntegerValue("property", property); | |
| 45 | |
| 46 std::auto_ptr<IResult> result(transaction.Execute(*statement, args)); | |
| 47 | |
| 48 if (result->IsDone()) | |
| 49 { | |
| 50 return false; | |
| 51 } | |
| 52 | |
| 53 result->SetExpectedType(0, ValueType_Utf8String); | |
| 54 | |
| 55 ValueType type = result->GetField(0).GetType(); | |
| 56 | |
| 57 if (type == ValueType_Null) | |
| 58 { | |
| 59 return false; | |
| 60 } | |
| 61 else if (type != ValueType_Utf8String) | |
| 62 { | |
| 63 throw Orthanc::OrthancException(Orthanc::ErrorCode_Database); | |
| 64 } | |
| 65 else | |
| 66 { | |
| 67 target = dynamic_cast<const Utf8StringValue&>(result->GetField(0)).GetContent(); | |
| 68 return true; | |
| 69 } | |
| 70 } | |
| 71 | |
| 72 | |
| 73 bool LookupGlobalProperty(std::string& target /* out */, | |
| 74 DatabaseManager& manager, | |
| 75 Orthanc::GlobalProperty property) | |
| 76 { | |
| 77 DatabaseManager::CachedStatement statement( | |
| 78 STATEMENT_FROM_HERE, manager, | |
| 79 "SELECT value FROM GlobalProperties WHERE property=${property}"); | |
| 80 | |
| 81 statement.SetReadOnly(true); | |
| 82 statement.SetParameterType("property", ValueType_Integer64); | |
| 83 | |
| 84 Dictionary args; | |
| 85 args.SetIntegerValue("property", property); | |
| 86 | |
| 87 statement.Execute(args); | |
| 88 statement.SetResultFieldType(0, ValueType_Utf8String); | |
| 89 | |
| 90 if (statement.IsDone()) | |
| 91 { | |
| 92 return false; | |
| 93 } | |
| 94 | |
| 95 ValueType type = statement.GetResultField(0).GetType(); | |
| 96 | |
| 97 if (type == ValueType_Null) | |
| 98 { | |
| 99 return false; | |
| 100 } | |
| 101 else if (type != ValueType_Utf8String) | |
| 102 { | |
| 103 throw Orthanc::OrthancException(Orthanc::ErrorCode_Database); | |
| 104 } | |
| 105 else | |
| 106 { | |
| 107 target = dynamic_cast<const Utf8StringValue&>(statement.GetResultField(0)).GetContent(); | |
| 108 return true; | |
| 109 } | |
| 110 } | |
| 111 | |
| 112 | |
| 113 void SetGlobalProperty(IDatabase& db, | |
| 114 ITransaction& transaction, | |
| 115 Orthanc::GlobalProperty property, | |
| 116 const std::string& utf8) | |
| 117 { | |
|
132
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
118 // This version of "SetGlobalProperty()" (with an explicit |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
119 // transaction) is called internally by the plugin to set a global |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
120 // property during the initialization of the database. (TODO: |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
121 // Could be replaced by the version with an implicit transaction |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
122 // to avoid code redundancy). |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
123 |
| 0 | 124 if (db.GetDialect() == Dialect_SQLite) |
| 125 { | |
| 126 Query query("INSERT OR REPLACE INTO GlobalProperties VALUES (${property}, ${value})", false); | |
| 127 query.SetType("property", ValueType_Integer64); | |
| 128 query.SetType("value", ValueType_Utf8String); | |
| 129 | |
| 130 std::auto_ptr<IPrecompiledStatement> statement(db.Compile(query)); | |
| 131 | |
| 132 Dictionary args; | |
| 133 args.SetIntegerValue("property", static_cast<int>(property)); | |
| 134 args.SetUtf8Value("value", utf8); | |
| 135 | |
| 136 transaction.ExecuteWithoutResult(*statement, args); | |
| 137 } | |
| 138 else | |
| 139 { | |
| 140 { | |
| 141 Query query("DELETE FROM GlobalProperties WHERE property=${property}", false); | |
| 142 query.SetType("property", ValueType_Integer64); | |
| 143 | |
| 144 std::auto_ptr<IPrecompiledStatement> statement(db.Compile(query)); | |
| 145 | |
| 146 Dictionary args; | |
| 147 args.SetIntegerValue("property", static_cast<int>(property)); | |
| 148 | |
| 149 transaction.ExecuteWithoutResult(*statement, args); | |
| 150 } | |
| 151 | |
| 152 { | |
| 153 Query query("INSERT INTO GlobalProperties VALUES (${property}, ${value})", false); | |
| 154 query.SetType("property", ValueType_Integer64); | |
| 155 query.SetType("value", ValueType_Utf8String); | |
| 156 | |
| 157 std::auto_ptr<IPrecompiledStatement> statement(db.Compile(query)); | |
| 158 | |
| 159 Dictionary args; | |
| 160 args.SetIntegerValue("property", static_cast<int>(property)); | |
| 161 args.SetUtf8Value("value", utf8); | |
| 162 | |
| 163 transaction.ExecuteWithoutResult(*statement, args); | |
| 164 } | |
| 165 } | |
| 166 } | |
| 167 | |
| 168 | |
| 169 void SetGlobalProperty(DatabaseManager& manager, | |
| 170 Orthanc::GlobalProperty property, | |
| 171 const std::string& utf8) | |
| 172 { | |
|
132
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
173 // This version of "SetGlobalProperty()" (without an explicit |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
174 // transaction) is called by Orthanc during its execution. Orthanc |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
175 // manages the transaction at a higher level, but this transaction |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
176 // is always present. |
|
5c69c2a14c97
OrthancPostgreSQL-3.2
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
67
diff
changeset
|
177 |
| 0 | 178 if (manager.GetDialect() == Dialect_SQLite) |
| 179 { | |
| 180 DatabaseManager::CachedStatement statement( | |
| 181 STATEMENT_FROM_HERE, manager, | |
| 182 "INSERT OR REPLACE INTO GlobalProperties VALUES (${property}, ${value})"); | |
| 183 | |
| 184 statement.SetParameterType("property", ValueType_Integer64); | |
| 185 statement.SetParameterType("value", ValueType_Utf8String); | |
| 186 | |
| 187 Dictionary args; | |
| 188 args.SetIntegerValue("property", static_cast<int>(property)); | |
| 189 args.SetUtf8Value("value", utf8); | |
| 190 | |
| 191 statement.Execute(args); | |
| 192 } | |
| 193 else | |
| 194 { | |
| 195 { | |
| 196 DatabaseManager::CachedStatement statement( | |
| 197 STATEMENT_FROM_HERE, manager, | |
| 198 "DELETE FROM GlobalProperties WHERE property=${property}"); | |
| 199 | |
| 200 statement.SetParameterType("property", ValueType_Integer64); | |
| 201 | |
| 202 Dictionary args; | |
| 203 args.SetIntegerValue("property", property); | |
| 204 | |
| 205 statement.Execute(args); | |
| 206 } | |
| 207 | |
| 208 { | |
| 209 DatabaseManager::CachedStatement statement( | |
| 210 STATEMENT_FROM_HERE, manager, | |
| 211 "INSERT INTO GlobalProperties VALUES (${property}, ${value})"); | |
| 212 | |
| 213 statement.SetParameterType("property", ValueType_Integer64); | |
| 214 statement.SetParameterType("value", ValueType_Utf8String); | |
| 215 | |
| 216 Dictionary args; | |
| 217 args.SetIntegerValue("property", static_cast<int>(property)); | |
| 218 args.SetUtf8Value("value", utf8); | |
| 219 | |
| 220 statement.Execute(args); | |
| 221 } | |
| 222 } | |
| 223 } | |
| 224 | |
| 225 | |
| 226 bool LookupGlobalIntegerProperty(int& target, | |
| 227 IDatabase& db, | |
| 228 ITransaction& transaction, | |
| 229 Orthanc::GlobalProperty property) | |
| 230 { | |
| 231 std::string value; | |
| 232 | |
| 233 if (LookupGlobalProperty(value, db, transaction, property)) | |
| 234 { | |
| 235 try | |
| 236 { | |
| 237 target = boost::lexical_cast<int>(value); | |
| 238 return true; | |
| 239 } | |
| 240 catch (boost::bad_lexical_cast&) | |
| 241 { | |
| 242 LOG(ERROR) << "Corrupted PostgreSQL database"; | |
| 243 throw Orthanc::OrthancException(Orthanc::ErrorCode_Database); | |
| 244 } | |
| 245 } | |
| 246 else | |
| 247 { | |
| 248 return false; | |
| 249 } | |
| 250 } | |
| 251 | |
| 252 | |
| 253 void SetGlobalIntegerProperty(IDatabase& db, | |
| 254 ITransaction& transaction, | |
| 255 Orthanc::GlobalProperty property, | |
| 256 int value) | |
| 257 { | |
| 258 SetGlobalProperty(db, transaction, property, boost::lexical_cast<std::string>(value)); | |
| 259 } | |
| 260 } |