Mercurial > hg > orthanc

comparison OrthancFramework/Sources/DicomNetworking/TimeoutDicomConnectionManager.cpp @ 4044:d25f4c0fa160 framework

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
splitting code into OrthancFramework and OrthancServer
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 10 Jun 2020 20:30:34 +0200
parents Core/DicomNetworking/TimeoutDicomConnectionManager.cpp@ea1d32861cfc
children bf7b9edf6b81
comparison
equal deleted inserted replaced
4043:6c6239aec462 4044:d25f4c0fa160
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-2020 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 "../Logging.h"
38 #include "../OrthancException.h"
39
40 namespace Orthanc
41 {
42 static boost::posix_time::ptime GetNow()
43 {
44 return boost::posix_time::microsec_clock::universal_time();
45 }
46
47
48 TimeoutDicomConnectionManager::Lock::Lock(TimeoutDicomConnectionManager& that,
49 const std::string& localAet,
50 const RemoteModalityParameters& remote) :
51 that_(that),
52 lock_(that_.mutex_)
53 {
54 // Calling "Touch()" will be done by the "~Lock()" destructor
55 that_.OpenInternal(localAet, remote);
56 }
57
58
59 TimeoutDicomConnectionManager::Lock::~Lock()
60 {
61 that_.TouchInternal();
62 }
63
64
65 DicomStoreUserConnection& TimeoutDicomConnectionManager::Lock::GetConnection()
66 {
67 if (that_.connection_.get() == NULL)
68 {
69 // The allocation should have been done by "that_.Open()" in the constructor
70 throw OrthancException(ErrorCode_InternalError);
71 }
72 else
73 {
74 return *that_.connection_;
75 }
76 }
77
78
79 // Mutex must be locked
80 void TimeoutDicomConnectionManager::TouchInternal()
81 {
82 lastUse_ = GetNow();
83 }
84
85
86 // Mutex must be locked
87 void TimeoutDicomConnectionManager::OpenInternal(const std::string& localAet,
88 const RemoteModalityParameters& remote)
89 {
90 DicomAssociationParameters other(localAet, remote);
91
92 if (connection_.get() == NULL ||
93 !connection_->GetParameters().IsEqual(other))
94 {
95 connection_.reset(new DicomStoreUserConnection(other));
96 }
97 }
98
99
100 // Mutex must be locked
101 void TimeoutDicomConnectionManager::CloseInternal()
102 {
103 if (connection_.get() != NULL)
104 {
105 LOG(INFO) << "Closing inactive DICOM association with modality: "
106 << connection_->GetParameters().GetRemoteModality().GetApplicationEntityTitle();
107
108 connection_.reset(NULL);
109 }
110 }
111
112
113 void TimeoutDicomConnectionManager::SetInactivityTimeout(unsigned int milliseconds)
114 {
115 boost::mutex::scoped_lock lock(mutex_);
116 timeout_ = boost::posix_time::milliseconds(milliseconds);
117 CloseInternal();
118 }
119
120
121 unsigned int TimeoutDicomConnectionManager::GetInactivityTimeout()
122 {
123 boost::mutex::scoped_lock lock(mutex_);
124 return static_cast<unsigned int>(timeout_.total_milliseconds());
125 }
126
127
128 void TimeoutDicomConnectionManager::CloseIfInactive()
129 {
130 boost::mutex::scoped_lock lock(mutex_);
131
132 if (connection_.get() != NULL &&
133 (GetNow() - lastUse_) >= timeout_)
134 {
135 CloseInternal();
136 }
137 }
138 }