Mercurial > hg > orthanc
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 } |