Mercurial > hg > orthanc-databases
annotate Framework/PostgreSQL/PostgreSQLStatement.cpp @ 214:ab96698c73a3
removed useless information about read-only in ITransaction and IPrecompiledStatement
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Thu, 25 Mar 2021 13:56:26 +0100 |
parents | 3236894320d6 |
children | d1b124d116c1 |
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 | |
193
3236894320d6
upgrade to year 2021
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
186
diff
changeset
|
5 * Copyright (C) 2017-2021 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 | |
107
5765cc5fd268
reverted fix for OS X that breaks other targets
Sebastien Jodogne <s.jodogne@orthanc-labs.com>
parents:
105
diff
changeset
|
22 #include "PostgreSQLIncludes.h" // Must be the first |
105 | 23 #include "PostgreSQLStatement.h" |
0 | 24 |
25 #include "../Common/BinaryStringValue.h" | |
26 #include "../Common/FileValue.h" | |
27 #include "../Common/Integer64Value.h" | |
28 #include "../Common/NullValue.h" | |
29 #include "../Common/ResultBase.h" | |
30 #include "../Common/Utf8StringValue.h" | |
31 #include "PostgreSQLResult.h" | |
32 | |
157
275e14f57f1e
replacing deprecated std::auto_ptr by std::unique_ptr
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
152
diff
changeset
|
33 #include <Compatibility.h> // For std::unique_ptr<> |
152 | 34 #include <Logging.h> |
35 #include <OrthancException.h> | |
36 #include <Toolbox.h> | |
37 #include <Endianness.h> | |
0 | 38 |
39 #include <cassert> | |
40 | |
41 | |
42 namespace OrthancDatabases | |
43 { | |
44 class PostgreSQLStatement::Inputs : public boost::noncopyable | |
45 { | |
46 private: | |
47 std::vector<char*> values_; | |
48 std::vector<int> sizes_; | |
49 | |
50 static char* Allocate(const void* source, int size) | |
51 { | |
52 if (size == 0) | |
53 { | |
54 return NULL; | |
55 } | |
56 else | |
57 { | |
58 char* ptr = reinterpret_cast<char*>(malloc(size)); | |
59 | |
60 if (source != NULL) | |
61 { | |
62 memcpy(ptr, source, size); | |
63 } | |
64 | |
65 return ptr; | |
66 } | |
67 } | |
68 | |
69 void Resize(size_t size) | |
70 { | |
71 // Shrinking of the vector | |
72 for (size_t i = size; i < values_.size(); i++) | |
73 { | |
74 if (values_[i] != NULL) | |
75 free(values_[i]); | |
76 } | |
77 | |
78 values_.resize(size, NULL); | |
79 sizes_.resize(size, 0); | |
80 } | |
81 | |
82 void EnlargeForIndex(size_t index) | |
83 { | |
84 if (index >= values_.size()) | |
85 { | |
86 // The vector is too small | |
87 Resize(index + 1); | |
88 } | |
89 } | |
90 | |
91 public: | |
92 Inputs() | |
93 { | |
94 } | |
95 | |
96 ~Inputs() | |
97 { | |
98 Resize(0); | |
99 } | |
100 | |
101 void SetItem(size_t pos, const void* source, int size) | |
102 { | |
103 EnlargeForIndex(pos); | |
104 | |
105 if (sizes_[pos] == size) | |
106 { | |
107 if (source && size != 0) | |
108 { | |
109 memcpy(values_[pos], source, size); | |
110 } | |
111 } | |
112 else | |
113 { | |
114 if (values_[pos] != NULL) | |
115 { | |
116 free(values_[pos]); | |
117 } | |
118 | |
119 values_[pos] = Allocate(source, size); | |
120 sizes_[pos] = size; | |
121 } | |
122 } | |
123 | |
124 void SetItem(size_t pos, int size) | |
125 { | |
126 SetItem(pos, NULL, size); | |
127 } | |
128 | |
129 void* GetItem(size_t pos) const | |
130 { | |
131 if (pos >= values_.size()) | |
132 { | |
133 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
134 } | |
135 | |
136 return values_[pos]; | |
137 } | |
138 | |
139 const std::vector<char*>& GetValues() const | |
140 { | |
141 return values_; | |
142 } | |
143 | |
144 const std::vector<int>& GetSizes() const | |
145 { | |
146 return sizes_; | |
147 } | |
148 }; | |
149 | |
150 | |
151 void PostgreSQLStatement::Prepare() | |
152 { | |
153 if (id_.size() > 0) | |
154 { | |
155 // Already prepared | |
156 return; | |
157 } | |
158 | |
159 for (size_t i = 0; i < oids_.size(); i++) | |
160 { | |
161 if (oids_[i] == 0) | |
162 { | |
163 // The type of an input parameter was not set | |
164 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls); | |
165 } | |
166 } | |
167 | |
168 id_ = Orthanc::Toolbox::GenerateUuid(); | |
169 | |
170 const unsigned int* tmp = oids_.size() ? &oids_[0] : NULL; | |
171 | |
172 PGresult* result = PQprepare(reinterpret_cast<PGconn*>(database_.pg_), | |
173 id_.c_str(), sql_.c_str(), oids_.size(), tmp); | |
174 | |
175 if (result == NULL) | |
176 { | |
177 id_.clear(); | |
178 database_.ThrowException(true); | |
179 } | |
180 | |
181 bool ok = (PQresultStatus(result) == PGRES_COMMAND_OK); | |
182 if (ok) | |
183 { | |
184 PQclear(result); | |
185 } | |
186 else | |
187 { | |
188 std::string message = PQresultErrorMessage(result); | |
189 PQclear(result); | |
190 id_.clear(); | |
191 LOG(ERROR) << "PostgreSQL error: " << message; | |
192 database_.ThrowException(false); | |
193 } | |
194 } | |
195 | |
196 | |
197 void PostgreSQLStatement::Unprepare() | |
198 { | |
199 if (id_.size() > 0) | |
200 { | |
201 // "Although there is no libpq function for deleting a | |
202 // prepared statement, the SQL DEALLOCATE statement can be | |
203 // used for that purpose." | |
130
2d2b268799a2
Explicit deallocation of prepared statements
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
107
diff
changeset
|
204 database_.Execute("DEALLOCATE \"" + id_ + "\""); |
0 | 205 } |
206 | |
207 id_.clear(); | |
208 } | |
209 | |
210 | |
211 void PostgreSQLStatement::DeclareInputInternal(unsigned int param, | |
212 unsigned int /*Oid*/ type) | |
213 { | |
214 Unprepare(); | |
215 | |
216 if (oids_.size() <= param) | |
217 { | |
218 oids_.resize(param + 1, 0); | |
219 binary_.resize(param + 1); | |
220 } | |
221 | |
222 oids_[param] = type; | |
223 binary_[param] = (type == TEXTOID || type == BYTEAOID || type == OIDOID) ? 0 : 1; | |
224 } | |
225 | |
226 | |
227 void PostgreSQLStatement::DeclareInputInteger(unsigned int param) | |
228 { | |
229 DeclareInputInternal(param, INT4OID); | |
230 } | |
231 | |
232 | |
233 void PostgreSQLStatement::DeclareInputInteger64(unsigned int param) | |
234 { | |
235 DeclareInputInternal(param, INT8OID); | |
236 } | |
237 | |
238 | |
239 void PostgreSQLStatement::DeclareInputString(unsigned int param) | |
240 { | |
241 DeclareInputInternal(param, TEXTOID); | |
242 } | |
243 | |
244 | |
245 void PostgreSQLStatement::DeclareInputBinary(unsigned int param) | |
246 { | |
247 DeclareInputInternal(param, BYTEAOID); | |
248 } | |
249 | |
250 | |
251 void PostgreSQLStatement::DeclareInputLargeObject(unsigned int param) | |
252 { | |
253 DeclareInputInternal(param, OIDOID); | |
254 } | |
255 | |
256 | |
257 void* /* PGresult* */ PostgreSQLStatement::Execute() | |
258 { | |
259 Prepare(); | |
260 | |
261 PGresult* result; | |
262 | |
263 if (oids_.size() == 0) | |
264 { | |
265 // No parameter | |
266 result = PQexecPrepared(reinterpret_cast<PGconn*>(database_.pg_), | |
267 id_.c_str(), 0, NULL, NULL, NULL, 1); | |
268 } | |
269 else | |
270 { | |
271 // At least 1 parameter | |
272 result = PQexecPrepared(reinterpret_cast<PGconn*>(database_.pg_), | |
273 id_.c_str(), | |
274 oids_.size(), | |
275 &inputs_->GetValues()[0], | |
276 &inputs_->GetSizes()[0], | |
277 &binary_[0], | |
278 1); | |
279 } | |
280 | |
281 if (result == NULL) | |
282 { | |
283 database_.ThrowException(true); | |
284 } | |
285 | |
286 return result; | |
287 } | |
288 | |
289 | |
290 PostgreSQLStatement::PostgreSQLStatement(PostgreSQLDatabase& database, | |
214
ab96698c73a3
removed useless information about read-only in ITransaction and IPrecompiledStatement
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
193
diff
changeset
|
291 const std::string& sql) : |
0 | 292 database_(database), |
293 sql_(sql), | |
294 inputs_(new Inputs), | |
295 formatter_(Dialect_PostgreSQL) | |
296 { | |
297 LOG(TRACE) << "PostgreSQL: " << sql; | |
298 } | |
299 | |
300 | |
301 PostgreSQLStatement::PostgreSQLStatement(PostgreSQLDatabase& database, | |
302 const Query& query) : | |
303 database_(database), | |
304 inputs_(new Inputs), | |
305 formatter_(Dialect_PostgreSQL) | |
306 { | |
307 query.Format(sql_, formatter_); | |
308 LOG(TRACE) << "PostgreSQL: " << sql_; | |
309 | |
310 for (size_t i = 0; i < formatter_.GetParametersCount(); i++) | |
311 { | |
312 switch (formatter_.GetParameterType(i)) | |
313 { | |
314 case ValueType_Integer64: | |
315 DeclareInputInteger64(i); | |
316 break; | |
317 | |
318 case ValueType_Utf8String: | |
319 DeclareInputString(i); | |
320 break; | |
321 | |
322 case ValueType_BinaryString: | |
323 DeclareInputBinary(i); | |
324 break; | |
325 | |
326 case ValueType_File: | |
327 DeclareInputLargeObject(i); | |
328 break; | |
329 | |
330 case ValueType_Null: | |
331 default: | |
332 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
333 } | |
334 } | |
335 } | |
336 | |
337 | |
46
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
338 PostgreSQLStatement::~PostgreSQLStatement() |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
339 { |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
340 try |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
341 { |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
342 Unprepare(); |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
343 } |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
344 catch (Orthanc::OrthancException&) |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
345 { |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
346 // Ignore possible exceptions due to connection loss |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
347 } |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
348 } |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
349 |
6a574d810b98
Compatibility with MySQL 8.0
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
23
diff
changeset
|
350 |
0 | 351 void PostgreSQLStatement::Run() |
352 { | |
353 PGresult* result = reinterpret_cast<PGresult*>(Execute()); | |
354 assert(result != NULL); // An exception would have been thrown otherwise | |
355 | |
356 bool ok = (PQresultStatus(result) == PGRES_COMMAND_OK || | |
357 PQresultStatus(result) == PGRES_TUPLES_OK); | |
358 if (ok) | |
359 { | |
360 PQclear(result); | |
361 } | |
362 else | |
363 { | |
364 std::string error = PQresultErrorMessage(result); | |
365 PQclear(result); | |
366 LOG(ERROR) << "PostgreSQL error: " << error; | |
367 database_.ThrowException(false); | |
368 } | |
369 } | |
370 | |
371 | |
372 void PostgreSQLStatement::BindNull(unsigned int param) | |
373 { | |
374 if (param >= oids_.size()) | |
375 { | |
376 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
377 } | |
378 | |
379 inputs_->SetItem(param, 0); | |
380 } | |
381 | |
382 | |
383 void PostgreSQLStatement::BindInteger(unsigned int param, | |
384 int value) | |
385 { | |
386 if (param >= oids_.size()) | |
387 { | |
388 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
389 } | |
390 | |
391 if (oids_[param] != INT4OID) | |
392 { | |
393 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadParameterType); | |
394 } | |
395 | |
396 assert(sizeof(int32_t) == 4); | |
397 int32_t v = htobe32(static_cast<int32_t>(value)); | |
398 inputs_->SetItem(param, &v, sizeof(int32_t)); | |
399 } | |
400 | |
401 | |
402 void PostgreSQLStatement::BindInteger64(unsigned int param, | |
403 int64_t value) | |
404 { | |
405 if (param >= oids_.size()) | |
406 { | |
407 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
408 } | |
409 | |
410 if (oids_[param] != INT8OID) | |
411 { | |
412 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadParameterType); | |
413 } | |
414 | |
415 assert(sizeof(int64_t) == 8); | |
416 int64_t v = htobe64(value); | |
417 inputs_->SetItem(param, &v, sizeof(int64_t)); | |
418 } | |
419 | |
420 | |
421 void PostgreSQLStatement::BindString(unsigned int param, | |
422 const std::string& value) | |
423 { | |
424 if (param >= oids_.size()) | |
425 { | |
426 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
427 } | |
428 | |
429 if (oids_[param] != TEXTOID && oids_[param] != BYTEAOID) | |
430 { | |
431 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadParameterType); | |
432 } | |
433 | |
434 if (value.size() == 0) | |
435 { | |
436 inputs_->SetItem(param, "", 1 /* end-of-string character */); | |
437 } | |
438 else | |
439 { | |
440 inputs_->SetItem(param, value.c_str(), | |
441 value.size() + 1); // "+1" for end-of-string character | |
442 } | |
443 } | |
444 | |
445 | |
446 void PostgreSQLStatement::BindLargeObject(unsigned int param, | |
447 const PostgreSQLLargeObject& value) | |
448 { | |
449 if (param >= oids_.size()) | |
450 { | |
451 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
452 } | |
453 | |
454 if (oids_[param] != OIDOID) | |
455 { | |
456 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadParameterType); | |
457 } | |
458 | |
459 inputs_->SetItem(param, value.GetOid().c_str(), | |
460 value.GetOid().size() + 1); // "+1" for end-of-string character | |
461 } | |
462 | |
463 | |
464 class PostgreSQLStatement::ResultWrapper : public ResultBase | |
465 { | |
466 private: | |
157
275e14f57f1e
replacing deprecated std::auto_ptr by std::unique_ptr
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
152
diff
changeset
|
467 std::unique_ptr<PostgreSQLResult> result_; |
0 | 468 |
469 protected: | |
470 virtual IValue* FetchField(size_t index) | |
471 { | |
472 return result_->GetValue(index); | |
473 } | |
474 | |
475 public: | |
186 | 476 explicit ResultWrapper(PostgreSQLStatement& statement) : |
0 | 477 result_(new PostgreSQLResult(statement)) |
478 { | |
479 SetFieldsCount(result_->GetColumnsCount()); | |
480 FetchFields(); | |
481 } | |
482 | |
483 virtual void Next() | |
484 { | |
485 result_->Next(); | |
486 FetchFields(); | |
487 } | |
488 | |
489 virtual bool IsDone() const | |
490 { | |
491 return result_->IsDone(); | |
492 } | |
493 }; | |
494 | |
495 | |
23
b2ff1cd2907a
handling of implicit transactions in DatabaseManager
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
496 IResult* PostgreSQLStatement::Execute(ITransaction& transaction, |
0 | 497 const Dictionary& parameters) |
498 { | |
499 for (size_t i = 0; i < formatter_.GetParametersCount(); i++) | |
500 { | |
501 const std::string& name = formatter_.GetParameterName(i); | |
502 | |
503 switch (formatter_.GetParameterType(i)) | |
504 { | |
505 case ValueType_Integer64: | |
506 BindInteger64(i, dynamic_cast<const Integer64Value&>(parameters.GetValue(name)).GetValue()); | |
507 break; | |
508 | |
509 case ValueType_Null: | |
510 BindNull(i); | |
511 break; | |
512 | |
513 case ValueType_Utf8String: | |
514 BindString(i, dynamic_cast<const Utf8StringValue&> | |
515 (parameters.GetValue(name)).GetContent()); | |
516 break; | |
517 | |
518 case ValueType_BinaryString: | |
519 BindString(i, dynamic_cast<const BinaryStringValue&> | |
520 (parameters.GetValue(name)).GetContent()); | |
521 break; | |
522 | |
523 case ValueType_File: | |
524 { | |
525 const FileValue& blob = | |
526 dynamic_cast<const FileValue&>(parameters.GetValue(name)); | |
527 | |
528 PostgreSQLLargeObject largeObject(database_, blob.GetContent()); | |
529 BindLargeObject(i, largeObject); | |
530 break; | |
531 } | |
532 | |
533 default: | |
534 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
535 } | |
536 } | |
537 | |
538 return new ResultWrapper(*this); | |
539 } | |
540 | |
541 | |
23
b2ff1cd2907a
handling of implicit transactions in DatabaseManager
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
542 void PostgreSQLStatement::ExecuteWithoutResult(ITransaction& transaction, |
0 | 543 const Dictionary& parameters) |
544 { | |
157
275e14f57f1e
replacing deprecated std::auto_ptr by std::unique_ptr
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
152
diff
changeset
|
545 std::unique_ptr<IResult> dummy(Execute(transaction, parameters)); |
0 | 546 } |
547 } |