Mercurial > hg > orthanc-databases
annotate Framework/Common/DatabaseManager.h @ 17:54ea251aed70
unit test
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 10 Jul 2018 10:33:02 +0200 |
parents | 9774802fd05f |
children | c7c54993a92e |
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 | |
5 * Copyright (C) 2017-2018 Osimis S.A., Belgium | |
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 #pragma once | |
23 | |
24 #include "IDatabaseFactory.h" | |
25 #include "StatementLocation.h" | |
26 | |
27 #include <Core/Enumerations.h> | |
28 | |
29 #include <boost/thread/recursive_mutex.hpp> | |
30 #include <memory> | |
31 | |
32 namespace OrthancDatabases | |
33 { | |
34 class DatabaseManager : public boost::noncopyable | |
35 { | |
36 private: | |
37 typedef std::map<StatementLocation, IPrecompiledStatement*> CachedStatements; | |
38 | |
39 boost::recursive_mutex mutex_; | |
40 std::auto_ptr<IDatabaseFactory> factory_; | |
41 std::auto_ptr<IDatabase> database_; | |
42 std::auto_ptr<ITransaction> transaction_; | |
43 CachedStatements cachedStatements_; | |
44 Dialect dialect_; | |
45 | |
46 IDatabase& GetDatabase(); | |
47 | |
48 void CloseIfUnavailable(Orthanc::ErrorCode e); | |
49 | |
50 IPrecompiledStatement* LookupCachedStatement(const StatementLocation& location) const; | |
51 | |
52 IPrecompiledStatement& CacheStatement(const StatementLocation& location, | |
53 const Query& query); | |
54 | |
55 ITransaction& GetTransaction(); | |
14
9774802fd05f
PostgreSQLStorageArea working
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
12
diff
changeset
|
56 |
0 | 57 public: |
58 DatabaseManager(IDatabaseFactory* factory); // Takes ownership | |
59 | |
60 ~DatabaseManager() | |
61 { | |
62 Close(); | |
63 } | |
64 | |
65 Dialect GetDialect() const | |
66 { | |
67 return dialect_; | |
68 } | |
69 | |
70 void Open() | |
71 { | |
72 GetDatabase(); | |
73 } | |
74 | |
75 void Close(); | |
76 | |
77 void StartTransaction(); | |
78 | |
79 void CommitTransaction(); | |
80 | |
81 void RollbackTransaction(); | |
82 | |
12
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
83 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
84 class Transaction : public boost::noncopyable |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
85 { |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
86 private: |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
87 boost::recursive_mutex::scoped_lock lock_; |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
88 DatabaseManager& manager_; |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
89 IDatabase& database_; |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
90 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
91 public: |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
92 Transaction(DatabaseManager& manager) : |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
93 lock_(manager.mutex_), |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
94 manager_(manager), |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
95 database_(manager.GetDatabase()) |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
96 { |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
97 } |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
98 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
99 void Commit() |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
100 { |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
101 manager_.CommitTransaction(); |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
102 } |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
103 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
104 void Rollback() |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
105 { |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
106 manager_.RollbackTransaction(); |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
107 } |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
108 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
109 DatabaseManager& GetManager() |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
110 { |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
111 return manager_; |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
112 } |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
113 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
114 IDatabase& GetDatabase() |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
115 { |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
116 return database_; |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
117 } |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
118 }; |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
119 |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
120 |
0 | 121 class CachedStatement : public boost::noncopyable |
122 { | |
123 private: | |
124 boost::recursive_mutex::scoped_lock lock_; | |
125 DatabaseManager& manager_; | |
126 StatementLocation location_; | |
127 ITransaction& transaction_; | |
128 IPrecompiledStatement* statement_; | |
129 std::auto_ptr<Query> query_; | |
130 std::auto_ptr<IResult> result_; | |
131 | |
12
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
132 void Setup(const char* sql); |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
133 |
0 | 134 IResult& GetResult() const; |
135 | |
136 public: | |
137 CachedStatement(const StatementLocation& location, | |
138 DatabaseManager& manager, | |
139 const char* sql); | |
140 | |
12
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
141 CachedStatement(const StatementLocation& location, |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
142 Transaction& transaction, |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
143 const char* sql); |
41543239072d
transactions for storage area
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
144 |
0 | 145 void SetReadOnly(bool readOnly); |
146 | |
147 void SetParameterType(const std::string& parameter, | |
148 ValueType type); | |
149 | |
150 void Execute(); | |
151 | |
152 void Execute(const Dictionary& parameters); | |
153 | |
154 bool IsDone() const; | |
155 | |
156 void Next(); | |
157 | |
158 size_t GetResultFieldsCount() const; | |
159 | |
160 void SetResultFieldType(size_t field, | |
161 ValueType type); | |
162 | |
163 const IValue& GetResultField(size_t index) const; | |
164 }; | |
165 }; | |
166 } |