377 lines
7.7 KiB
C++
377 lines
7.7 KiB
C++
#include "StdAfx.h"
|
|
#include "TimerEngine.h"
|
|
#include "TraceService.h"
|
|
|
|
//枚举
|
|
enum
|
|
{
|
|
TIMER_SPACE = 25, //时间间隔
|
|
TIMER_LEAD = TIMER_SPACE*20 //提前时间
|
|
};
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
//构造函数
|
|
CTimerThread::CTimerThread(void)
|
|
{
|
|
m_pTimerEngine = NULL;
|
|
}
|
|
|
|
//析构函数
|
|
CTimerThread::~CTimerThread(void)
|
|
{
|
|
}
|
|
|
|
//配置函数
|
|
bool CTimerThread::InitThread(CTimerEngine * pTimerEngine)
|
|
{
|
|
if (pTimerEngine == NULL) return false;
|
|
|
|
//设置变量
|
|
m_pTimerEngine = pTimerEngine;
|
|
|
|
return true;
|
|
}
|
|
|
|
//运行函数
|
|
bool CTimerThread::OnEventThreadRun()
|
|
{
|
|
ASSERT(m_pTimerEngine != NULL);
|
|
Sleep(TIMER_SPACE);
|
|
m_pTimerEngine->OnTimerThreadSink();
|
|
return true;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
//构造函数
|
|
CTimerEngine::CTimerEngine(void)
|
|
{
|
|
m_bService = false;
|
|
//m_pIQueueServiceSink = NULL;
|
|
}
|
|
|
|
//析构函数
|
|
CTimerEngine::~CTimerEngine(void)
|
|
{
|
|
//停止服务
|
|
ConcludeService();
|
|
|
|
//清理内存
|
|
tagTimerItem * pTimerItem = NULL;
|
|
for (INT_PTR i = m_TimerItemFree.GetCount()-1; i>=0; i--)
|
|
{
|
|
pTimerItem = m_TimerItemFree[i];
|
|
ASSERT(pTimerItem != NULL);
|
|
SafeDelete(pTimerItem);
|
|
}
|
|
for (INT_PTR i = m_TimerItemActive.GetCount()-1; i>=0; i--)
|
|
{
|
|
pTimerItem = m_TimerItemActive[i];
|
|
ASSERT(pTimerItem != NULL);
|
|
SafeDelete(pTimerItem);
|
|
}
|
|
m_TimerItemFree.RemoveAll();
|
|
m_TimerItemActive.RemoveAll();
|
|
|
|
return;
|
|
}
|
|
|
|
//接口查询
|
|
void * CTimerEngine::QueryInterface(const IID & Guid, DWORD dwQueryVer)
|
|
{
|
|
QUERYINTERFACE(ITimerEngine, Guid, dwQueryVer);
|
|
QUERYINTERFACE_IUNKNOWNEX(ITimerEngine, Guid, dwQueryVer);
|
|
return NULL;
|
|
}
|
|
|
|
//设置定时器
|
|
bool CTimerEngine::SetTimer(DWORD dwTimerID, DWORD dwElapse, DWORD dwRepeat, WPARAM dwBindParameter)
|
|
{
|
|
//效验参数
|
|
ASSERT(dwRepeat > 0L);
|
|
if (dwRepeat == 0) return false;
|
|
|
|
//锁定资源
|
|
CWHDataLocker lock(m_CriticalSection);
|
|
|
|
//查找定时器
|
|
bool bTimerExist = false;
|
|
tagTimerItem * pTimerItem = NULL;
|
|
for (INT_PTR i = m_TimerItemActive.GetCount()-1; i>=0; i-- )
|
|
{
|
|
pTimerItem = m_TimerItemActive[i];
|
|
ASSERT(pTimerItem != NULL);
|
|
if (pTimerItem->wTimerID == dwTimerID)
|
|
{
|
|
bTimerExist = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
//创建定时器
|
|
if ( !bTimerExist )
|
|
{
|
|
INT_PTR nFreeCount = m_TimerItemFree.GetCount();
|
|
if (nFreeCount > 0)
|
|
{
|
|
pTimerItem = m_TimerItemFree[nFreeCount-1];
|
|
ASSERT(pTimerItem != NULL);
|
|
m_TimerItemFree.RemoveAt(nFreeCount - 1);
|
|
}
|
|
else
|
|
{
|
|
try
|
|
{
|
|
pTimerItem = new tagTimerItem;
|
|
ASSERT(pTimerItem != NULL);
|
|
if (pTimerItem == NULL) return false;
|
|
}
|
|
catch (...)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
//设置参数
|
|
ASSERT(pTimerItem != NULL);
|
|
pTimerItem->wTimerID = dwTimerID;
|
|
pTimerItem->wBindParam = dwBindParameter;
|
|
pTimerItem->dwElapse = dwElapse-(dwElapse%TIMER_SPACE);
|
|
pTimerItem->dwRepeatTimes = dwRepeat;
|
|
pTimerItem->dwDestTime = GetTickCount();
|
|
//提前20个粒度进行通知 - TIMER_SPACE * 20
|
|
if ( 1 == pTimerItem->dwRepeatTimes )
|
|
{
|
|
if (pTimerItem->dwElapse>TIMER_SPACE+TIMER_LEAD)
|
|
{
|
|
pTimerItem->dwDestTime += pTimerItem->dwElapse-TIMER_LEAD;
|
|
}
|
|
else
|
|
{
|
|
pTimerItem->dwDestTime += TIMER_SPACE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pTimerItem->dwDestTime += pTimerItem->dwElapse;
|
|
}
|
|
|
|
//激活定时器
|
|
if ( !bTimerExist )
|
|
{
|
|
m_TimerItemActive.Add(pTimerItem);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//删除定时器
|
|
bool CTimerEngine::KillTimer(DWORD dwTimerID)
|
|
{
|
|
//锁定资源
|
|
CWHDataLocker lock(m_CriticalSection);//
|
|
|
|
//查找定时器
|
|
tagTimerItem * pTimerItem = NULL;
|
|
for ( INT_PTR i = m_TimerItemActive.GetCount()-1; i>=0; i-- )
|
|
{
|
|
pTimerItem = m_TimerItemActive[i];
|
|
ASSERT(pTimerItem != NULL);
|
|
if (pTimerItem->wTimerID == dwTimerID)
|
|
{
|
|
m_TimerItemActive.RemoveAt(i);
|
|
m_TimerItemFree.Add(pTimerItem);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
//删除定时器
|
|
bool CTimerEngine::KillAllTimer()
|
|
{
|
|
//锁定资源
|
|
CWHDataLocker lock(m_CriticalSection);//
|
|
|
|
//删除定时器
|
|
m_TimerItemFree.Append(m_TimerItemActive);
|
|
m_TimerItemActive.RemoveAll();
|
|
|
|
return true;
|
|
}
|
|
|
|
//开始服务
|
|
bool CTimerEngine::StartService()
|
|
{
|
|
//效验状态
|
|
if (m_bService)
|
|
{
|
|
CTraceService::TraceString(TEXT("定时器引擎重复启动，启动操作忽略"), TraceLevel_Warning);
|
|
return true;
|
|
}
|
|
|
|
//设置变量
|
|
if ( !m_TimerThread.InitThread(this) )
|
|
{
|
|
CTraceService::TraceString(TEXT("定时器引擎线程服务初始化失败"), TraceLevel_Exception);
|
|
return false;
|
|
}
|
|
|
|
//启动服务
|
|
if ( !m_TimerThread.StartThread() )
|
|
{
|
|
CTraceService::TraceString(TEXT("定时器引擎线程服务启动失败"), TraceLevel_Exception);
|
|
return false;
|
|
}
|
|
|
|
SetThreadPriority(m_TimerThread.GetThreadHandle(), REALTIME_PRIORITY_CLASS);
|
|
|
|
//设置变量
|
|
m_bService = true;
|
|
|
|
return true;
|
|
}
|
|
|
|
//停止服务
|
|
bool CTimerEngine::ConcludeService()
|
|
{
|
|
//设置变量
|
|
m_bService = false;
|
|
|
|
//停止线程
|
|
m_TimerThread.ConcludeThread(INFINITE);
|
|
|
|
//设置变量
|
|
m_TimerItemFree.Append(m_TimerItemActive);
|
|
m_TimerItemActive.RemoveAll();
|
|
|
|
m_QueueServiceEvent.SetQueueServiceSink(NULL);
|
|
|
|
return true;
|
|
}
|
|
|
|
//设置接口
|
|
bool CTimerEngine::SetTimerEngineEvent(IUnknownEx * pIUnknownEx)
|
|
{
|
|
//效验参数
|
|
ASSERT(pIUnknownEx != NULL);
|
|
ASSERT( !m_bService );
|
|
if (m_bService) return false;
|
|
if (pIUnknownEx == NULL) return false;
|
|
|
|
////设置接口
|
|
//ASSERT(pIUnknownEx != NULL);
|
|
//m_pIQueueServiceSink = QUERY_OBJECT_PTR_INTERFACE(pIUnknownEx, IQueueServiceSink);
|
|
//ASSERT(m_pIQueueServiceSink != NULL);
|
|
//return (m_pIQueueServiceSink != NULL);
|
|
|
|
//设置接口
|
|
if ( !m_QueueServiceEvent.SetQueueServiceSink(pIUnknownEx) )
|
|
{
|
|
CTraceService::TraceString(TEXT("定时器引擎与触发服务绑定失败"), TraceLevel_Exception);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//定时器通知
|
|
void CTimerEngine::OnTimerThreadSink()
|
|
{
|
|
//缓冲锁定
|
|
CWHDataLocker lock(m_CriticalSection);//
|
|
|
|
//查询定时器
|
|
tagTimerItem * pTimerItem = NULL;
|
|
DWORD dwTickCount=GetTickCount();
|
|
bool bKillTimer=false;
|
|
for (INT_PTR i = m_TimerItemActive.GetCount() - 1; i >= 0; i--)
|
|
{
|
|
//效验参数
|
|
pTimerItem = m_TimerItemActive[i];
|
|
ASSERT(pTimerItem != NULL);
|
|
if (pTimerItem == NULL) return;
|
|
ASSERT(pTimerItem->dwDestTime != 0);
|
|
|
|
//定时器处理
|
|
bKillTimer = false;
|
|
if (dwTickCount>=pTimerItem->dwDestTime)
|
|
{
|
|
//try
|
|
//{
|
|
// BYTE cbBuffer[MAX_ASYNCHRONISM_DATA] = {0}; //接收缓冲
|
|
// //投递消息
|
|
// NTY_TimerEvent * pTimerEvent = (NTY_TimerEvent *)cbBuffer;
|
|
// pTimerEvent->dwTimerID = pTimerItem->wTimerID;
|
|
// pTimerEvent->dwBindParameter = pTimerItem->wBindParam;
|
|
|
|
// m_pIQueueServiceSink->OnQueueServiceSink(EVENT_TIMER, cbBuffer, sizeof(NTY_TimerEvent));
|
|
//}
|
|
//catch (...) { }
|
|
|
|
m_QueueServiceEvent.PostTimerEvent(pTimerItem->wTimerID, pTimerItem->wBindParam);
|
|
|
|
//设置次数
|
|
if (pTimerItem->dwRepeatTimes != TIMES_INFINITY)
|
|
{
|
|
ASSERT(pTimerItem->dwRepeatTimes > 0);
|
|
pTimerItem->dwRepeatTimes--;
|
|
if (pTimerItem->dwRepeatTimes == 0L)
|
|
{
|
|
bKillTimer = true;
|
|
m_TimerItemActive.RemoveAt(i);
|
|
m_TimerItemFree.Add(pTimerItem);
|
|
}
|
|
}
|
|
|
|
//设置时间
|
|
if ( !bKillTimer )//提前20个粒度进行通知 - TIMER_SPACE * 20
|
|
{
|
|
if ( 1 == pTimerItem->dwRepeatTimes )
|
|
{
|
|
if (pTimerItem->dwElapse>TIMER_SPACE+TIMER_LEAD)
|
|
{
|
|
pTimerItem->dwDestTime += pTimerItem->dwElapse-TIMER_LEAD;
|
|
}
|
|
else
|
|
{
|
|
pTimerItem->dwDestTime += TIMER_SPACE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pTimerItem->dwDestTime += pTimerItem->dwElapse;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
//建立对象函数
|
|
extern "C" __declspec(dllexport) void * CreateTimerEngine(const GUID & Guid, DWORD dwInterfaceVer)
|
|
{
|
|
//建立对象
|
|
CTimerEngine * pTimerEngine = NULL;
|
|
try
|
|
{
|
|
pTimerEngine = new CTimerEngine();
|
|
if (pTimerEngine == NULL) throw TEXT("创建失败");
|
|
void * pObject = pTimerEngine->QueryInterface(Guid, dwInterfaceVer);
|
|
if (pObject == NULL) throw TEXT("接口查询失败");
|
|
return pObject;
|
|
}
|
|
catch (...) {}
|
|
|
|
//清理对象
|
|
SafeDelete(pTimerEngine);
|
|
return NULL;
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|