Mercurial > hg > orthanc-databases

comparison Framework/Plugins/IndexConnectionsPool.cpp @ 569:f18e46d7dbf8 attach-custom-data

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
merged find-refactoring -> attach-custom-data
author Alain Mazy <am@orthanc.team>
date Tue, 24 Sep 2024 15:04:21 +0200
parents 54d518dcd74a
children
comparison
equal deleted inserted replaced
368:82f73188b58d 569:f18e46d7dbf8
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-2023 Osimis S.A., Belgium
6 * Copyright (C) 2024-2024 Orthanc Team SRL, Belgium
7 * Copyright (C) 2021-2024 Sebastien Jodogne, ICTEAM UCLouvain, Belgium
8 *
9 * This program is free software: you can redistribute it and/or
10 * modify it under the terms of the GNU Affero General Public License
11 * as published by the Free Software Foundation, either version 3 of
12 * the License, or (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful, but
15 * WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Affero General Public License for more details.
18 *
19 * You should have received a copy of the GNU Affero General Public License
20 * along with this program. If not, see <http://www.gnu.org/licenses/>.
21 **/
22
23
24 #include "IndexConnectionsPool.h"
25
26 namespace OrthancDatabases
27 {
28 class IndexConnectionsPool::ManagerReference : public Orthanc::IDynamicObject
29 {
30 private:
31 DatabaseManager* manager_;
32
33 public:
34 explicit ManagerReference(DatabaseManager& manager) :
35 manager_(&manager)
36 {
37 }
38
39 DatabaseManager& GetManager()
40 {
41 assert(manager_ != NULL);
42 return *manager_;
43 }
44 };
45
46
47 IndexConnectionsPool::IndexConnectionsPool(IndexBackend* backend,
48 size_t countConnections) :
49 backend_(backend),
50 countConnections_(countConnections)
51 {
52 if (countConnections == 0)
53 {
54 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange,
55 "There must be a non-zero number of connections to the database");
56 }
57 else if (backend == NULL)
58 {
59 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer);
60 }
61 else
62 {
63 context_ = backend_->GetContext();
64 }
65 }
66
67
68 IndexConnectionsPool::~IndexConnectionsPool()
69 {
70 for (std::list<DatabaseManager*>::iterator
71 it = connections_.begin(); it != connections_.end(); ++it)
72 {
73 assert(*it != NULL);
74 delete *it;
75 }
76 }
77
78
79 void IndexConnectionsPool::OpenConnections(bool hasIdentifierTags,
80 const std::list<IdentifierTag>& identifierTags)
81 {
82 boost::unique_lock<boost::shared_mutex> lock(connectionsMutex_);
83
84 if (connections_.size() == 0)
85 {
86 assert(backend_.get() != NULL);
87
88 {
89 std::unique_ptr<DatabaseManager> manager(new DatabaseManager(backend_->CreateDatabaseFactory()));
90 manager->GetDatabase(); // Make sure to open the database connection
91
92 backend_->ConfigureDatabase(*manager, hasIdentifierTags, identifierTags);
93 connections_.push_back(manager.release());
94 }
95
96 for (size_t i = 1; i < countConnections_; i++)
97 {
98 connections_.push_back(new DatabaseManager(backend_->CreateDatabaseFactory()));
99 connections_.back()->GetDatabase(); // Make sure to open the database connection
100 }
101
102 for (std::list<DatabaseManager*>::iterator
103 it = connections_.begin(); it != connections_.end(); ++it)
104 {
105 assert(*it != NULL);
106 availableConnections_.Enqueue(new ManagerReference(**it));
107 }
108 }
109 else
110 {
111 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls);
112 }
113 }
114
115
116 void IndexConnectionsPool::CloseConnections()
117 {
118 boost::unique_lock<boost::shared_mutex> lock(connectionsMutex_);
119
120 if (connections_.size() != countConnections_)
121 {
122 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls);
123 }
124 else if (availableConnections_.GetSize() != countConnections_)
125 {
126 throw Orthanc::OrthancException(Orthanc::ErrorCode_Database, "Some connections are still in use, bug in the Orthanc core");
127 }
128 else
129 {
130 for (std::list<DatabaseManager*>::iterator
131 it = connections_.begin(); it != connections_.end(); ++it)
132 {
133 assert(*it != NULL);
134 (*it)->Close();
135 }
136 }
137 }
138
139
140 IndexConnectionsPool::Accessor::Accessor(IndexConnectionsPool& pool) :
141 lock_(pool.connectionsMutex_),
142 pool_(pool),
143 manager_(NULL)
144 {
145 for (;;)
146 {
147 std::unique_ptr<Orthanc::IDynamicObject> manager(pool.availableConnections_.Dequeue(100));
148 if (manager.get() != NULL)
149 {
150 manager_ = &dynamic_cast<ManagerReference&>(*manager).GetManager();
151 return;
152 }
153 }
154 }
155
156
157 IndexConnectionsPool::Accessor::~Accessor()
158 {
159 assert(manager_ != NULL);
160 pool_.availableConnections_.Enqueue(new ManagerReference(*manager_));
161 }
162
163
164 IndexBackend& IndexConnectionsPool::Accessor::GetBackend() const
165 {
166 return *pool_.backend_;
167 }
168
169
170 DatabaseManager& IndexConnectionsPool::Accessor::GetManager() const
171 {
172 assert(manager_ != NULL);
173 return *manager_;
174 }
175 }