blocxx

ThreadCounter.cpp

Go to the documentation of this file.
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