Mercurial > hg > orthanc-databases

comparison Framework/Common/DatabaseManager.h @ 0:7cea966b6829

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial commit
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 04 Jul 2018 08:16:29 +0200
parents
children 41543239072d
comparison
equal deleted inserted replaced
-1:000000000000 0:7cea966b6829
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();
56
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
83 class CachedStatement : public boost::noncopyable
84 {
85 private:
86 boost::recursive_mutex::scoped_lock lock_;
87 DatabaseManager& manager_;
88 StatementLocation location_;
89 IDatabase& database_;
90 ITransaction& transaction_;
91 IPrecompiledStatement* statement_;
92 std::auto_ptr<Query> query_;
93 std::auto_ptr<IResult> result_;
94
95 IResult& GetResult() const;
96
97 public:
98 CachedStatement(const StatementLocation& location,
99 DatabaseManager& manager,
100 const char* sql);
101
102 void SetReadOnly(bool readOnly);
103
104 void SetParameterType(const std::string& parameter,
105 ValueType type);
106
107 void Execute();
108
109 void Execute(const Dictionary& parameters);
110
111 IDatabase& GetDatabase()
112 {
113 return database_;
114 }
115
116 bool IsDone() const;
117
118 void Next();
119
120 size_t GetResultFieldsCount() const;
121
122 void SetResultFieldType(size_t field,
123 ValueType type);
124
125 const IValue& GetResultField(size_t index) const;
126 };
127 };
128 }