Mercurial > hg > orthanc
diff OrthancFramework/Sources/MultiThreading/Semaphore.h @ 4068:408ac60c3cf8 framework
integration mainline->framework
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 12 Jun 2020 09:29:03 +0200 |
parents | Core/MultiThreading/Semaphore.h@7176ebf08765 Core/MultiThreading/Semaphore.h@d25f4c0fa160 |
children | d5c09b5f882f |
line wrap: on
line diff
--- a/OrthancFramework/Sources/MultiThreading/Semaphore.h Thu Jun 11 18:57:03 2020 +0200 +++ b/OrthancFramework/Sources/MultiThreading/Semaphore.h Fri Jun 12 09:29:03 2020 +0200 @@ -20,7 +20,7 @@ * you do not wish to do so, delete this exception statement from your * version. If you delete this exception statement from all source files * in the program, then also delete it here. - * + * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU @@ -44,12 +44,12 @@ unsigned int availableResources_; boost::mutex mutex_; boost::condition_variable condition_; - - void Release(); + + void Release(unsigned int resourceCount = 1); - void Acquire(); + void Acquire(unsigned int resourceCount = 1); - bool TryAcquire(); + bool TryAcquire(unsigned int resourceCount = 1); public: explicit Semaphore(unsigned int availableResources); @@ -63,38 +63,42 @@ { private: Semaphore& that_; + unsigned int resourceCount_; public: - explicit Locker(Semaphore& that) : - that_(that) + explicit Locker(Semaphore& that, unsigned int resourceCount = 1) : + that_(that), + resourceCount_(resourceCount) { - that_.Acquire(); + that_.Acquire(resourceCount_); } ~Locker() { - that_.Release(); + that_.Release(resourceCount_); } }; class TryLocker : public boost::noncopyable { private: - Semaphore& that_; - bool isAcquired_; + Semaphore& that_; + unsigned int resourceCount_; + bool isAcquired_; public: - explicit TryLocker(Semaphore& that) : - that_(that) + explicit TryLocker(Semaphore& that, unsigned int resourceCount = 1) : + that_(that), + resourceCount_(resourceCount) { - isAcquired_ = that_.TryAcquire(); + isAcquired_ = that_.TryAcquire(resourceCount_); } ~TryLocker() { if (isAcquired_) { - that_.Release(); + that_.Release(resourceCount_); } }