Mercurial > hg > orthanc-databases

annotate PostgreSQL/Plugins/StoragePlugin.cpp @ 12:41543239072d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
transactions for storage area
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 09 Jul 2018 18:34:56 +0200
parents 17bce6a07b2b
children 9774802fd05f
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
1 /**
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
2 * Orthanc - A Lightweight, RESTful DICOM Store
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
4 * Department, University Hospital of Liege, Belgium
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
5 * Copyright (C) 2017-2018 Osimis S.A., Belgium
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
6 *
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
7 * This program is free software: you can redistribute it and/or
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
8 * modify it under the terms of the GNU Affero General Public License
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
9 * as published by the Free Software Foundation, either version 3 of
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
10 * the License, or (at your option) any later version.
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
11 *
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
12 * This program is distributed in the hope that it will be useful, but
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
15 * Affero General Public License for more details.
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
16 *
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
17 * You should have received a copy of the GNU Affero General Public License
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
18 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
19 **/
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
20
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
21
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
22 #include "../../Framework/Plugins/StorageBackend.h"
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
23
12
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
24 #include "../../Framework/Common/FileValue.h"
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
25 #include "../../Framework/PostgreSQL/PostgreSQLDatabase.h"
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
26 #include "../../Framework/PostgreSQL/PostgreSQLLargeObject.h"
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
27 #include "../../Framework/PostgreSQL/PostgreSQLTransaction.h"
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
28
2
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
29 #include <Plugins/Samples/Common/OrthancPluginCppWrapper.h>
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
30 #include <Core/Logging.h>
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
31
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
32
12
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
33 namespace OrthancDatabases
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
34 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
35 class PostgreSQLStorageArea : public StorageBackend
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
36 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
37 private:
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
38 class Factory : public IDatabaseFactory
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
39 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
40 private:
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
41 PostgreSQLStorageArea& that_;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
42
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
43 public:
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
44 Factory(PostgreSQLStorageArea& that) :
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
45 that_(that)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
46 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
47 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
48
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
49 virtual Dialect GetDialect() const
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
50 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
51 return Dialect_PostgreSQL;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
52 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
53
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
54 virtual IDatabase* Open()
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
55 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
56 return that_.OpenInternal();
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
57 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
58 };
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
59
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
60 OrthancPluginContext* context_;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
61 PostgreSQLParameters parameters_;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
62 bool clearAll_;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
63
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
64 IDatabase* OpenInternal()
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
65 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
66 std::auto_ptr<PostgreSQLDatabase> db(new PostgreSQLDatabase(parameters_));
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
67
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
68 db->Open();
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
69
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
70 if (parameters_.HasLock())
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
71 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
72 db->AdvisoryLock(43 /* some arbitrary constant */);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
73 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
74
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
75 if (clearAll_)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
76 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
77 db->ClearAll();
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
78 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
79
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
80 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
81 PostgreSQLTransaction t(*db);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
82
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
83 if (!db->DoesTableExist("StorageArea"))
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
84 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
85 db->Execute("CREATE TABLE IF NOT EXISTS StorageArea("
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
86 "uuid VARCHAR NOT NULL PRIMARY KEY,"
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
87 "content OID NOT NULL,"
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
88 "type INTEGER NOT NULL)");
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
89
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
90 // Automatically remove the large objects associated with the table
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
91 db->Execute("CREATE OR REPLACE RULE StorageAreaDelete AS ON DELETE "
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
92 "TO StorageArea DO SELECT lo_unlink(old.content);");
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
93 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
94
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
95 t.Commit();
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
96 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
97
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
98 return db.release();
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
99 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
100
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
101 public:
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
102 PostgreSQLStorageArea(const PostgreSQLParameters& parameters) :
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
103 StorageBackend(new Factory(*this)),
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
104 parameters_(parameters),
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
105 clearAll_(false)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
106 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
107 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
108
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
109 void SetClearAll(bool clear)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
110 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
111 clearAll_ = clear;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
112 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
113
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
114
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
115 virtual void Create(DatabaseManager::Transaction& transaction,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
116 const std::string& uuid,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
117 const void* content,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
118 size_t size,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
119 OrthancPluginContentType type)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
120 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
121 std::auto_ptr<FileValue> file(new FileValue(content, size));
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
122
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
123 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
124 DatabaseManager::CachedStatement statement(
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
125 STATEMENT_FROM_HERE, GetManager(),
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
126 "INSERT INTO StorageArea VALUES (${uuid}, ${content}, ${type})");
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
127
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
128 statement.SetParameterType("uuid", ValueType_Utf8String);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
129 statement.SetParameterType("content", ValueType_File);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
130 statement.SetParameterType("type", ValueType_Integer64);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
131
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
132 Dictionary args;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
133 args.SetUtf8Value("uuid", uuid);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
134 args.SetValue("content", file.release());
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
135 args.SetIntegerValue("type", type);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
136
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
137 statement.Execute(args);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
138 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
139 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
140
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
141
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
142 virtual void Read(void*& content,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
143 size_t& size,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
144 DatabaseManager::Transaction& transaction,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
145 const std::string& uuid,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
146 OrthancPluginContentType type)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
147 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
148 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
149
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
150
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
151 virtual void Remove(DatabaseManager::Transaction& transaction,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
152 const std::string& uuid,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
153 OrthancPluginContentType type)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
154 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
155 DatabaseManager::CachedStatement statement(
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
156 STATEMENT_FROM_HERE, GetManager(),
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
157 "DELETE FROM StorageArea WHERE uuid=${uuid} AND type=${type}");
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
158
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
159 statement.SetParameterType("uuid", ValueType_Utf8String);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
160 statement.SetParameterType("type", ValueType_Integer64);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
161
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
162 Dictionary args;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
163 args.SetUtf8Value("uuid", uuid);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
164 args.SetIntegerValue("type", type);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
165
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
166 statement.Execute(args);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
167 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
168 };
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
169 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
170
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
171
2
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
172 static bool DisplayPerformanceWarning()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
173 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
174 (void) DisplayPerformanceWarning; // Disable warning about unused function
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
175 LOG(WARNING) << "Performance warning in PostgreSQL storage area: "
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
176 << "Non-release build, runtime debug assertions are turned on";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
177 return true;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
178 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
179
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
180
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
181 extern "C"
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
182 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
183 ORTHANC_PLUGINS_API int32_t OrthancPluginInitialize(OrthancPluginContext* context)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
184 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
185 Orthanc::Logging::Initialize(context);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
186
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
187 assert(DisplayPerformanceWarning());
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
188
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
189 /* Check the version of the Orthanc core */
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
190 if (OrthancPluginCheckVersion(context) == 0)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
191 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
192 char info[1024];
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
193 sprintf(info, "Your version of Orthanc (%s) must be above %d.%d.%d to run this plugin",
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
194 context->orthancVersion,
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
195 ORTHANC_PLUGINS_MINIMAL_MAJOR_NUMBER,
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
196 ORTHANC_PLUGINS_MINIMAL_MINOR_NUMBER,
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
197 ORTHANC_PLUGINS_MINIMAL_REVISION_NUMBER);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
198 OrthancPluginLogError(context, info);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
199 return -1;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
200 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
201
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
202 OrthancPluginSetDescription(context, "Stores the Orthanc storage area into a PostgreSQL database.");
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
203
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
204 OrthancPlugins::OrthancConfiguration configuration(context);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
205
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
206 if (!configuration.IsSection("PostgreSQL"))
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
207 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
208 LOG(WARNING) << "No available configuration for the PostgreSQL storage area plugin";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
209 return 0;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
210 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
211
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
212 OrthancPlugins::OrthancConfiguration postgresql;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
213 configuration.GetSection(postgresql, "PostgreSQL");
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
214
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
215 bool enable;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
216 if (!postgresql.LookupBooleanValue(enable, "EnableStorage") ||
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
217 !enable)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
218 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
219 LOG(WARNING) << "The PostgreSQL storage area is currently disabled, set \"EnableStorage\" "
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
220 << "to \"true\" in the \"PostgreSQL\" section of the configuration file of Orthanc";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
221 return 0;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
222 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
223
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
224 try
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
225 {
12
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
226 OrthancDatabases::PostgreSQLParameters parameters(postgresql);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
227 OrthancDatabases::StorageBackend::Register
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
228 (context, new OrthancDatabases::PostgreSQLStorageArea(parameters));
2
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
229 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
230 catch (Orthanc::OrthancException& e)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
231 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
232 LOG(ERROR) << e.What();
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
233 return -1;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
234 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
235 catch (...)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
236 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
237 LOG(ERROR) << "Native exception while initializing the plugin";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
238 return -1;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
239 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
240
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
241 return 0;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
242 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
243
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
244
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
245 ORTHANC_PLUGINS_API void OrthancPluginFinalize()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
246 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
247 LOG(WARNING) << "PostgreSQL storage area is finalizing";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
248 OrthancDatabases::StorageBackend::Finalize();
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
249 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
250
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
251
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
252 ORTHANC_PLUGINS_API const char* OrthancPluginGetName()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
253 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
254 return "postgresql-storage";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
255 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
256
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
257
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
258 ORTHANC_PLUGINS_API const char* OrthancPluginGetVersion()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
259 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
260 return ORTHANC_PLUGIN_VERSION;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
261 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
262 }