middleware_threemetting/StableCore/MsgCenter.cpp

#include "StdAfx.h"
#include "MsgCenter.h"
#include "IMsgObserver.h"


SINGLETON_IMPLEMENT(CMsgCenter)

CMsgCenter::CMsgCenter(void) : m_Stop(false), m_pThreadObj(NULL)
{
}


CMsgCenter::~CMsgCenter(void)
{
	m_Stop = true;

	if(m_pThreadObj != NULL)
	{
		DWORD ExitCode;
		GetExitCodeThread(m_pThreadObj->m_hThread, &ExitCode);
		if(ExitCode == STILL_ACTIVE)
		{
			m_ThreadSleepFlag.SetEvent();
			WaitForSingleObject(m_pThreadObj->m_hThread, INFINITE);
		}
		m_pThreadObj = NULL;
	}	
}

/*****************************************************************
**【函数名称】	__wait
**【函数功能】	置线程于等待信号状态
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::__wait( void )
{
	m_ThreadSleepFlag.Lock();
}

/*****************************************************************
**【函数名称】	__dispatchMsg
**【函数功能】	分派消息
**【参数】		
**【返回值】
****************************************************************/
int CMsgCenter::__dispatchMsg( void )
{
	CSingleLock Locker(&m_RWLock, TRUE);	// 互斥加锁
	
	// 获取事件队列信息
	if(m_MsgList.IsEmpty()) 
		return 0;
	MSG_INNER* pMsg = m_MsgList.RemoveHead();
	
	Locker.Unlock();

	// 根据实际类型分发消息
	__sendMsgToObserver(pMsg->MsgType, pMsg->Content);
	__releaseMsg(pMsg->MsgType, pMsg->Content);
	
	// 删除节点，释放内存
	delete pMsg;

	// 返回
	return m_MsgList.GetCount();
}

/*****************************************************************
**【函数名称】	__sendMsgToObserver
**【函数功能】	消息发送给订阅者
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::__sendMsgToObserver( UINT MsgType, PARAM lpContent )
{
	ObsvrMap::iterator itr;
	ObsvrMap::iterator itrEnd;

	itr = m_ObsvrMap.find(MsgType);
	if(itr == m_ObsvrMap.end())
		return;

	itrEnd = m_ObsvrMap.upper_bound(MsgType);
	for(; itr != itrEnd; ++itr)
	{
		itr->second->onMessage(MsgType, lpContent);
	}
}

/*****************************************************************
**【函数名称】	__releaseMsg
**【函数功能】	释放消息内容
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::__releaseMsg( UINT MsgType, PARAM lpContent )
{
	if(lpContent == NULL)
		return;

	switch(MsgType)
	{
	case SYS_MSG_RES_NEW:
	case SYS_MSG_RES_DEL:	
	case SYS_MSG_RES_STATE:
		delete (DEV_RES_ID*)lpContent;
		break;
	}		
}

/*****************************************************************
**【函数名称】	__dipatchThreadFunc
**【函数功能】	消息分发线程函数
**【参数】		
**【返回值】
****************************************************************/
UINT CMsgCenter::__dipatchThreadFunc( LPVOID pParam )
{
	CMsgCenter* pSelf = (CMsgCenter*)pParam;

	while(!pSelf->m_Stop)
	{
		if(pSelf->__dispatchMsg() == 0)
			pSelf->__wait();
	} // end while

	return 0;
}

/*****************************************************************
**【函数名称】	regist
**【函数功能】	订阅消息
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::regist( UINT MsgType, IMsgObserver* pObserver )
{
	m_ObsvrMap.insert(pair<UINT, IMsgObserver*>(MsgType, pObserver));
}

/*****************************************************************
**【函数名称】	unregist
**【函数功能】	取消订阅
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::unregist( UINT MsgType, IMsgObserver* pObserver )
{
	ObsvrMap::iterator itr;
	ObsvrMap::iterator itrEnd;

	itr = m_ObsvrMap.find(MsgType);
	if(itr == m_ObsvrMap.end())
		return;

	itrEnd = m_ObsvrMap.upper_bound(MsgType);
	while(itr != itrEnd)
	{
		if(itr->second == pObserver)
		{
			m_ObsvrMap.erase(itr);
			break;
		}
		itr++;
	}
}

/*****************************************************************
**【函数名称】	pushMsg
**【函数功能】	压入消息
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::pushMsg( UINT MsgType, PARAM lpContent )
{
	if(m_Stop)
		return;

	// 生成消息实体
	MSG_INNER* pMsg = new MSG_INNER;
	memset(pMsg, 0, sizeof(pMsg));
	pMsg->MsgType = MsgType;			// 消息类型

	// 填充事件内容
	switch(MsgType)
	{
	case SYS_MSG_RES_NEW:
	case SYS_MSG_RES_DEL:	
	case SYS_MSG_RES_STATE:
		{
			DEV_RES_ID* pResId = new DEV_RES_ID(*(DEV_RES_ID*)lpContent);
			pMsg->Content = (PARAM)pResId;
		}
		break;

	default:
		pMsg->Content = lpContent;
	} // end switch

	m_RWLock.Lock();	// 互斥加锁

	// 将消息插入缓冲队列
	m_MsgList.AddTail(pMsg);

	m_RWLock.Unlock();

	// 需要唤醒分发线程
	m_ThreadSleepFlag.SetEvent();
}

/*****************************************************************
**【函数名称】	startInThreadMode
**【函数功能】	以线程方式启动消息循环
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::startInThreadMode( void )
{
	m_pThreadObj = AfxBeginThread(__dipatchThreadFunc, this);
}

/*****************************************************************
**【函数名称】	startMsgPump
**【函数功能】	直接启动消息泵
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::startMsgPump( void )
{
	while(!m_Stop)
	{
		if(__dispatchMsg() == 0)
			__wait();
	}
}

/*****************************************************************
**【函数名称】	stopMsgPump
**【函数功能】	停止消息泵
**【参数】		
**【返回值】
****************************************************************/
void CMsgCenter::stopMsgPump( void )
{
	m_Stop = true;
	m_ThreadSleepFlag.SetEvent();
}