Mercurial > hg > orthanc-databases

annotate PostgreSQL/Plugins/StoragePlugin.cpp @ 14:9774802fd05f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
PostgreSQLStorageArea working
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 09 Jul 2018 20:28:27 +0200
parents 41543239072d
children 9e419261f1c9
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 {
14
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
148 DatabaseManager::CachedStatement statement(
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
149 STATEMENT_FROM_HERE, GetManager(),
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
150 "SELECT content FROM StorageArea WHERE uuid=$1 AND type=$2");
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
151
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
152 statement.SetParameterType("uuid", ValueType_Utf8String);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
153 statement.SetParameterType("type", ValueType_Integer64);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
154
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
155 Dictionary args;
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
156 args.SetUtf8Value("uuid", uuid);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
157 args.SetIntegerValue("type", type);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
158
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
159 statement.Execute(args);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
160
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
161 if (statement.IsDone())
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
162 {
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
163 throw Orthanc::OrthancException(Orthanc::ErrorCode_UnknownResource);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
164 }
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
165 else if (statement.GetResultFieldsCount() != 1 ||
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
166 statement.GetResultField(0).GetType() != ValueType_File)
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
167 {
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
168 throw Orthanc::OrthancException(Orthanc::ErrorCode_Database);
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
169 }
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
170 else
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
171 {
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
172 const FileValue& value = dynamic_cast<const FileValue&>(statement.GetResultField(0));
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
173 ReadFromString(content, size, value.GetContent());
9774802fd05f PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 12
diff changeset
174 }
12
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
175 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
176
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
177
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
178 virtual void Remove(DatabaseManager::Transaction& transaction,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
179 const std::string& uuid,
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
180 OrthancPluginContentType type)
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
181 {
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
182 DatabaseManager::CachedStatement statement(
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
183 STATEMENT_FROM_HERE, GetManager(),
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
184 "DELETE FROM StorageArea WHERE uuid=${uuid} AND type=${type}");
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
185
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
186 statement.SetParameterType("uuid", ValueType_Utf8String);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
187 statement.SetParameterType("type", ValueType_Integer64);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
188
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
189 Dictionary args;
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
190 args.SetUtf8Value("uuid", uuid);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
191 args.SetIntegerValue("type", type);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
192
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
193 statement.Execute(args);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
194 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
195 };
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
196 }
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
197
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
198
2
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
199 static bool DisplayPerformanceWarning()
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 (void) DisplayPerformanceWarning; // Disable warning about unused function
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
202 LOG(WARNING) << "Performance warning in PostgreSQL storage area: "
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
203 << "Non-release build, runtime debug assertions are turned on";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
204 return true;
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
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 extern "C"
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
209 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
210 ORTHANC_PLUGINS_API int32_t OrthancPluginInitialize(OrthancPluginContext* context)
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 Orthanc::Logging::Initialize(context);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
213
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
214 assert(DisplayPerformanceWarning());
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
215
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
216 /* Check the version of the Orthanc core */
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
217 if (OrthancPluginCheckVersion(context) == 0)
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 char info[1024];
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
220 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
221 context->orthancVersion,
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
222 ORTHANC_PLUGINS_MINIMAL_MAJOR_NUMBER,
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
223 ORTHANC_PLUGINS_MINIMAL_MINOR_NUMBER,
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
224 ORTHANC_PLUGINS_MINIMAL_REVISION_NUMBER);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
225 OrthancPluginLogError(context, info);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
226 return -1;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
227 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
228
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
229 OrthancPluginSetDescription(context, "Stores the Orthanc storage area into a PostgreSQL database.");
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
230
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
231 OrthancPlugins::OrthancConfiguration configuration(context);
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
232
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
233 if (!configuration.IsSection("PostgreSQL"))
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 LOG(WARNING) << "No available configuration for the PostgreSQL storage area plugin";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
236 return 0;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
237 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
238
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
239 OrthancPlugins::OrthancConfiguration postgresql;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
240 configuration.GetSection(postgresql, "PostgreSQL");
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
241
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
242 bool enable;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
243 if (!postgresql.LookupBooleanValue(enable, "EnableStorage") ||
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
244 !enable)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
245 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
246 LOG(WARNING) << "The PostgreSQL storage area is currently disabled, set \"EnableStorage\" "
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
247 << "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
248 return 0;
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 try
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
252 {
12
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
253 OrthancDatabases::PostgreSQLParameters parameters(postgresql);
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
254 OrthancDatabases::StorageBackend::Register
41543239072d transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 2
diff changeset
255 (context, new OrthancDatabases::PostgreSQLStorageArea(parameters));
2
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 catch (Orthanc::OrthancException& e)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
258 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
259 LOG(ERROR) << e.What();
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
260 return -1;
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 catch (...)
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
263 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
264 LOG(ERROR) << "Native exception while initializing the plugin";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
265 return -1;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
266 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
267
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
268 return 0;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
269 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
270
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
271
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
272 ORTHANC_PLUGINS_API void OrthancPluginFinalize()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
273 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
274 LOG(WARNING) << "PostgreSQL storage area is finalizing";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
275 OrthancDatabases::StorageBackend::Finalize();
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
276 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
277
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
278
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
279 ORTHANC_PLUGINS_API const char* OrthancPluginGetName()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
280 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
281 return "postgresql-storage";
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
282 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
283
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
284
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
285 ORTHANC_PLUGINS_API const char* OrthancPluginGetVersion()
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
286 {
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
287 return ORTHANC_PLUGIN_VERSION;
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
288 }
17bce6a07b2b storage plugin skeletons
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
289 }