用线程实现的计时器

#ifndef THREADTIMER_H
#define THREADTIMER_H

template <class TYPE> class CThreadTimer : CObject
{
public:
	typedef void (TYPE::*FN_TIMEOUT)();
public:
	DWORD GetInterval() const { return m_dwInterval; }

	void SetInterval(DWORD dwInterval)
	{
		m_dwInterval = dwInterval;
		m_evtObjects[0].SetEvent();
	}

	BOOL GetEnabled() const { return m_bEnabled; }

	void SetEnabled(BOOL bEnabled)
	{
		m_bEnabled = bEnabled;		
		m_evtObjects[0].SetEvent();
	}

	CThreadTimer(TYPE* pTimeout, FN_TIMEOUT fnTimeout)
		: m_dwInterval(1000), m_bEnabled(FALSE)
	{
		m_pTimeout = pTimeout;
		m_fnTimeout = fnTimeout;
		m_hThread = CreateThread(NULL, 0, ThreadProc, this, 0, NULL);
	}

	~CThreadTimer()
	{
		if (m_hThread)
		{
			m_evtObjects[1].SetEvent();
			if (WaitForSingleObject(m_hThread, 5000) != WAIT_TIMEOUT)
			{
				TerminateThread(m_hThread, WAIT_ABANDONED);
			}
			CloseHandle(m_hThread);
			m_hThread = NULL;
		}
	}
private:
	DWORD ThreadProc()
	{
		DWORD dwInterval = 0;
		CSyncObject* ppObjects[2];
		ppObjects[0] = &m_evtObjects[0];
		ppObjects[1] = &m_evtObjects[1];
		DWORD dwCount = sizeof(m_evtObjects) / sizeof(m_evtObjects[0]);
		for ( ; ; )
		{
			dwInterval = m_bEnabled ? m_dwInterval : INFINITE;

			CMultiLock lock(ppObjects, dwCount);
			DWORD dwWait = lock.Lock(dwInterval, FALSE);
			if (dwWait == WAIT_OBJECT_0 + 1) break;
			if (dwWait == WAIT_TIMEOUT)
			{
				if (m_pTimeout && m_fnTimeout) (m_pTimeout->*m_fnTimeout)();
			}
		}
		return 0;
	}

	static DWORD ThreadProc(LPVOID lpParam)
	{
		return ((CThreadTimer*) lpParam)->ThreadProc();
	}
private:
	TYPE* m_pTimeout;

	FN_TIMEOUT m_fnTimeout;

	DWORD m_dwInterval;

	BOOL m_bEnabled;

	HANDLE m_hThread;

	CEvent m_evtObjects[2];
};

#endif // THREADTIMER_H

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值