comparison Core/DicomNetworking/TimeoutDicomConnectionManager.cpp @ 2603:988936118354 jobs

reorganization
author Sebastien Jodogne <s.jodogne@gmail.com>
date Fri, 18 May 2018 17:02:25 +0200
parents
children 25225f0b4f33
comparison
equal deleted inserted replaced
2602:c25f1a52acbc 2603:988936118354
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 General Public License as
9 * published by the Free Software Foundation, either version 3 of the
10 * License, or (at your option) any later version.
11 *
12 * In addition, as a special exception, the copyright holders of this
13 * program give permission to link the code of its release with the
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
15 * that use the same license as the "OpenSSL" library), and distribute
16 * the linked executables. You must obey the GNU General Public License
17 * in all respects for all of the code used other than "OpenSSL". If you
18 * modify file(s) with this exception, you may extend this exception to
19 * your version of the file(s), but you are not obligated to do so. If
20 * you do not wish to do so, delete this exception statement from your
21 * version. If you delete this exception statement from all source files
22 * in the program, then also delete it here.
23 *
24 * This program is distributed in the hope that it will be useful, but
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
27 * General Public License for more details.
28 *
29 * You should have received a copy of the GNU General Public License
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
31 **/
32
33
34 #include "../PrecompiledHeaders.h"
35 #include "TimeoutDicomConnectionManager.h"
36
37 #include "../OrthancException.h"
38
39 namespace Orthanc
40 {
41 static boost::posix_time::ptime GetNow()
42 {
43 return boost::posix_time::microsec_clock::universal_time();
44 }
45
46 class TimeoutDicomConnectionManager::Resource : public IDicomConnectionManager::IResource
47 {
48 private:
49 TimeoutDicomConnectionManager& that_;
50 boost::mutex::scoped_lock lock_;
51
52 public:
53 Resource(TimeoutDicomConnectionManager& that) :
54 that_(that),
55 lock_(that.mutex_)
56 {
57 }
58
59 ~Resource()
60 {
61 that_.Touch();
62 }
63
64 DicomUserConnection& GetConnection()
65 {
66 if (that_.connection_.get() == NULL)
67 {
68 throw OrthancException(ErrorCode_InternalError);
69 }
70
71 return *that_.connection_;
72 }
73 };
74
75
76 void TimeoutDicomConnectionManager::Touch()
77 {
78 lastUse_ = GetNow();
79 }
80
81
82 void TimeoutDicomConnectionManager::CheckTimeoutInternal()
83 {
84 if (connection_.get() != NULL &&
85 (GetNow() - lastUse_) >= timeout_)
86 {
87 connection_.reset(NULL);
88 }
89 }
90
91
92 void TimeoutDicomConnectionManager::SetTimeout(unsigned int timeout)
93 {
94 boost::mutex::scoped_lock lock(mutex_);
95
96 timeout_ = boost::posix_time::milliseconds(timeout);
97 CheckTimeoutInternal();
98 }
99
100
101 unsigned int TimeoutDicomConnectionManager::GetTimeout()
102 {
103 boost::mutex::scoped_lock lock(mutex_);
104 return timeout_.total_milliseconds();
105 }
106
107
108 void TimeoutDicomConnectionManager::Close()
109 {
110 boost::mutex::scoped_lock lock(mutex_);
111 connection_.reset(NULL);
112 }
113
114
115 void TimeoutDicomConnectionManager::CheckTimeout()
116 {
117 boost::mutex::scoped_lock lock(mutex_);
118 CheckTimeoutInternal();
119 }
120
121
122 IDicomConnectionManager::IResource*
123 TimeoutDicomConnectionManager::AcquireConnection(const std::string& localAet,
124 const RemoteModalityParameters& remote)
125 {
126 boost::mutex::scoped_lock lock(mutex_);
127
128 if (connection_.get() == NULL ||
129 !connection_->IsSameAssociation(localAet, remote))
130 {
131 connection_.reset(new DicomUserConnection(localAet, remote));
132 }
133
134 return new Resource(*this);
135 }
136 }