threadpool.h

SMAL是short message abstract library的缩写,是由风起水流软件工作室（www.zealware.com）开发的一个支持短信网关系统开发的C++底层抽象接口库

/**
 *	threadpool.h
 *	
 *	Short Message Abstractive Library.(SMAL)
 *
 *	Copyright 2003-2006	北京风起水流软件工作室
 *	
 *	http://www.zealware.com
 *	
 *	[e-mail]
 *
 */

#ifndef _SMAL_THREADPOOL_H
#define _SMAL_THREADPOOL_H

#include <smal.h>
#include <common/util.h>

//##ModelId=46B5D43F003A
struct IJobDesc
{
};

//##ModelId=46B5D43F0043
struct IWorker
{
	//##ModelId=46B5D43F004D
	virtual void ProcessJob(IJobDesc* pJob)=0;
};


//##ModelId=46B5D43F0056
struct ThreadInfo
{
	//##ModelId=46B5D43F0061
	DWORD m_pThreadId;
	//##ModelId=46B5D43F0062
	HANDLE m_hThread;
	//##ModelId=46B5D43F0063
	bool	m_bBusyWorking;
	//##ModelId=46B5D43F006A
	ThreadInfo() { m_hThread=0; m_bBusyWorking=false; }
	//##ModelId=46B5D43F006B
	ThreadInfo(HANDLE handle, bool bBusy) { m_hThread=handle; m_bBusyWorking=bBusy; }
	//##ModelId=46B5D43F006E
	ThreadInfo(const ThreadInfo& info) { m_hThread=info.m_hThread; m_bBusyWorking=info.m_bBusyWorking; }
};

//##ModelId=46B5D43F0074
typedef std::map<DWORD, ThreadInfo> ThreadMap;

//##ModelId=46B5D43F0092
class CThreadPool  
{
	//##ModelId=46B5D43F00E3
	friend static unsigned int CThreadPool::ManagerProc(void* p); 
	//##ModelId=46B5D43F00EC
	friend static unsigned int CThreadPool::WorkerProc(void* p);
//protected:
//	//##ModelId=46B5D43F0100
//	enum ThreadPoolStatus { BUSY, IDLE, NORMAL };
public:
	//interface to the outside
 	//##ModelId=46B5D43F0100
 	enum ThreadPoolStatus { BUSY, IDLE, NORMAL };
	//##ModelId=46B5D43F0093
	void Start(unsigned short nStatic, unsigned short nmax);
	//##ModelId=46B5D43F009E
	void Stop(bool bHash=false);
	//##ModelId=46B5D43F00A0
	void ProcessJob(IJobDesc* pJob, IWorker* pWorker) const;

	//constructor and destructor
	//##ModelId=46B5D43F00A6
	CThreadPool();
	//##ModelId=46B5D43F00A7
	virtual ~CThreadPool();

  //##ModelId=46B5D43F00C6
  ThreadPoolStatus GetThreadPoolStatus();
protected:
	//interfaces public:
	//##ModelId=46B5D43F00A9
	HANDLE GetMgrIoPort() const { return m_hMgrIoPort; }
	//##ModelId=46B5D43F00AB
	UINT GetMgrWaitTime() const { return 1000*60; } //1 minute
	//##ModelId=46B5D43F00B1
	HANDLE GetWorkerIoPort() const { return m_hWorkerIoPort; }

private:
	//##ModelId=46B5D43F00B3
	static unsigned int ManagerProc(void* p);
	//##ModelId=46B5D43F00B6
	static unsigned int WorkerProc(void* p);
protected:
	//manager thread
	//##ModelId=46B5D43F00BB
	HANDLE m_hMgrThread;
	//##ModelId=46B5D43F00BC
	HANDLE m_hMgrIoPort;
protected:
	//configuration parameters
	//##ModelId=46B5D43F00BD
	mutable UINT m_nNumberOfStaticThreads;
	//##ModelId=46B5D43F00BE
	mutable UINT m_nNumberOfTotalThreads;

protected:
	//helper functions
	//##ModelId=46B5D43F00C4
	void AddThreads();
	//##ModelId=46B5D43F00C5
	void RemoveThreads();
	//##ModelId=46B5D43F00C7
	void ChangeStatus(DWORD threadId, bool status);
	//##ModelId=46B5D43F00D0
	void RemoveThread(DWORD threadId)
	{
		//CSingleLock lock(&m_arrayCs);
		//lock.Lock();
		m_arrayCs.Enter();
		//m_threadMap.RemoveKey(threadId);
		m_threadMap.erase(threadId);
		//lock.Unlock();
		m_arrayCs.Leave();
	}

protected:
	//all the work threads
	//CMap<DWORD, DWORD&, ThreadInfo, ThreadInfo&> m_threadMap;
	//##ModelId=46B5D43F00D3
	ThreadMap m_threadMap;
	//##ModelId=46B5D43F00D9
	ThreadMap::iterator myIterator;
	//CCriticalSection m_arrayCs;
	//##ModelId=46B5D43F00DE
	MyCriticalSection m_arrayCs;
	//##ModelId=46B5D43F00E2
	HANDLE m_hWorkerIoPort;
};


#endif