126 lines
3.8 KiB
C++
126 lines
3.8 KiB
C++
/*!
|
|
* \copy
|
|
* Copyright (c) 2009-2015, Cisco Systems
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in
|
|
* the documentation and/or other materials provided with the
|
|
* distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
* COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
|
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*
|
|
* \file WelsThreadPool.h
|
|
*
|
|
* \brief Interfaces introduced in thread pool
|
|
*
|
|
* \date 5/09/2012 Created
|
|
*
|
|
*************************************************************************************
|
|
*/
|
|
|
|
|
|
#ifndef _WELS_THREAD_POOL_H_
|
|
#define _WELS_THREAD_POOL_H_
|
|
|
|
#include <stdio.h>
|
|
#include "WelsTask.h"
|
|
#include "WelsTaskThread.h"
|
|
#include "WelsCircleQueue.h"
|
|
#include "WelsList.h"
|
|
|
|
namespace WelsCommon {
|
|
|
|
|
|
class CWelsThreadPool : public CWelsThread, public IWelsTaskThreadSink {
|
|
public:
|
|
enum {
|
|
DEFAULT_THREAD_NUM = 4,
|
|
};
|
|
|
|
static WELS_THREAD_ERROR_CODE SetThreadNum (int32_t iMaxThreadNum);
|
|
|
|
static CWelsThreadPool& AddReference();
|
|
void RemoveInstance();
|
|
|
|
static bool IsReferenced();
|
|
|
|
//IWelsTaskThreadSink
|
|
virtual WELS_THREAD_ERROR_CODE OnTaskStart (CWelsTaskThread* pThread, IWelsTask* pTask);
|
|
virtual WELS_THREAD_ERROR_CODE OnTaskStop (CWelsTaskThread* pThread, IWelsTask* pTask);
|
|
|
|
// CWelsThread
|
|
virtual void ExecuteTask();
|
|
|
|
WELS_THREAD_ERROR_CODE QueueTask (IWelsTask* pTask);
|
|
int32_t GetThreadNum() const {
|
|
return m_iMaxThreadNum;
|
|
}
|
|
|
|
|
|
protected:
|
|
WELS_THREAD_ERROR_CODE Init();
|
|
WELS_THREAD_ERROR_CODE Uninit();
|
|
|
|
WELS_THREAD_ERROR_CODE CreateIdleThread();
|
|
void DestroyThread (CWelsTaskThread* pThread);
|
|
WELS_THREAD_ERROR_CODE AddThreadToIdleQueue (CWelsTaskThread* pThread);
|
|
WELS_THREAD_ERROR_CODE AddThreadToBusyList (CWelsTaskThread* pThread);
|
|
WELS_THREAD_ERROR_CODE RemoveThreadFromBusyList (CWelsTaskThread* pThread);
|
|
void AddTaskToWaitedList (IWelsTask* pTask);
|
|
CWelsTaskThread* GetIdleThread();
|
|
IWelsTask* GetWaitedTask();
|
|
int32_t GetIdleThreadNum();
|
|
int32_t GetBusyThreadNum();
|
|
int32_t GetWaitedTaskNum();
|
|
void ClearWaitedTasks();
|
|
|
|
private:
|
|
CWelsThreadPool();
|
|
virtual ~CWelsThreadPool();
|
|
|
|
WELS_THREAD_ERROR_CODE StopAllRunning();
|
|
|
|
static int32_t m_iRefCount;
|
|
static CWelsLock m_cInitLock;
|
|
static int32_t m_iMaxThreadNum;
|
|
|
|
CWelsCircleQueue<IWelsTask>* m_cWaitedTasks;
|
|
CWelsCircleQueue<CWelsTaskThread>* m_cIdleThreads;
|
|
CWelsList<CWelsTaskThread>* m_cBusyThreads;
|
|
|
|
CWelsLock m_cLockPool;
|
|
CWelsLock m_cLockWaitedTasks;
|
|
CWelsLock m_cLockIdleTasks;
|
|
CWelsLock m_cLockBusyTasks;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN (CWelsThreadPool);
|
|
};
|
|
|
|
}
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|