blocxx
|
00001 /******************************************************************************* 00002 * Copyright (C) 2005, Vintela, Inc. All rights reserved. 00003 * Copyright (C) 2006, Novell, Inc. All rights reserved. 00004 * 00005 * Redistribution and use in source and binary forms, with or without 00006 * modification, are permitted provided that the following conditions are met: 00007 * 00008 * * Redistributions of source code must retain the above copyright notice, 00009 * this list of conditions and the following disclaimer. 00010 * * Redistributions in binary form must reproduce the above copyright 00011 * notice, this list of conditions and the following disclaimer in the 00012 * documentation and/or other materials provided with the distribution. 00013 * * Neither the name of 00014 * Vintela, Inc., 00015 * nor Novell, Inc., 00016 * nor the names of its contributors or employees may be used to 00017 * endorse or promote products derived from this software without 00018 * specific prior written permission. 00019 * 00020 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 00021 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00022 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 00023 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 00024 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 00025 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 00026 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 00027 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 00028 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 00029 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00030 * POSSIBILITY OF SUCH DAMAGE. 00031 *******************************************************************************/ 00032 00033 00038 #include "blocxx/BLOCXX_config.h" 00039 #include "blocxx/ThreadCounter.hpp" 00040 #include "blocxx/NonRecursiveMutexLock.hpp" 00041 #include "blocxx/Assertion.hpp" 00042 #include "blocxx/TimeoutException.hpp" 00043 #include "blocxx/Timeout.hpp" 00044 00045 namespace BLOCXX_NAMESPACE 00046 { 00047 00048 ThreadCounter::ThreadCounter(Int32 maxThreads) 00049 : m_maxThreads(maxThreads) 00050 , m_runCount(0) 00051 {} 00052 ThreadCounter::~ThreadCounter() 00053 {} 00054 void 00055 ThreadCounter::incThreadCount(UInt32 sTimeout, UInt32 usTimeout) 00056 { 00057 incThreadCount(Timeout::relative(sTimeout + static_cast<float>(usTimeout) * 1000000.0)); 00058 } 00059 void 00060 ThreadCounter::incThreadCount(const Timeout& timeout) 00061 { 00062 NonRecursiveMutexLock l(m_runCountGuard); 00063 while (m_runCount >= m_maxThreads) 00064 { 00065 if (!m_runCountCondition.timedWait(l, timeout)) 00066 { 00067 BLOCXX_THROW(TimeoutException, "ThreadCounter::incThreadCount timedout"); 00068 } 00069 } 00070 ++m_runCount; 00071 } 00072 void 00073 ThreadCounter::decThreadCount() 00074 { 00075 NonRecursiveMutexLock l(m_runCountGuard); 00076 BLOCXX_ASSERT(m_runCount > 0); 00077 --m_runCount; 00078 m_runCountCondition.notifyAll(); 00079 } 00080 Int32 00081 ThreadCounter::getThreadCount() 00082 { 00083 NonRecursiveMutexLock l(m_runCountGuard); 00084 return m_runCount; 00085 } 00086 void 00087 ThreadCounter::waitForAll(UInt32 sTimeout, UInt32 usTimeout) 00088 { 00089 waitForAll(Timeout::relative(sTimeout + static_cast<float>(usTimeout) * 1000000.0)); 00090 } 00091 00092 void 00093 ThreadCounter::waitForAll(const Timeout& timeout) 00094 { 00095 NonRecursiveMutexLock runCountLock(m_runCountGuard); 00096 while (m_runCount > 0) 00097 { 00098 if (!m_runCountCondition.timedWait(runCountLock, timeout)) 00099 { 00100 BLOCXX_THROW(TimeoutException, "ThreadCounter::waitForAll timedout"); 00101 } 00102 } 00103 } 00104 void 00105 ThreadCounter::setMax(Int32 maxThreads) 00106 { 00107 NonRecursiveMutexLock runCountLock(m_runCountGuard); 00108 m_maxThreads = maxThreads; 00109 m_runCountCondition.notifyAll(); 00110 } 00111 ThreadCountDecrementer::ThreadCountDecrementer(ThreadCounterRef const& x) 00112 : m_counter(x) 00113 {} 00114 ThreadCountDecrementer::~ThreadCountDecrementer() 00115 {} 00116 void 00117 ThreadCountDecrementer::doNotifyThreadDone(Thread *) 00118 { 00119 m_counter->decThreadCount(); 00120 } 00121 00122 } // end namespace BLOCXX_NAMESPACE 00123