网络编程状态机的实现－宏定义

状态机最简单的实现就是使用大量的switch case或者是if else，对不同的状态分别处理。

这样的代码比较直观，但是项目逐渐变大时就让人不爽了。

 

另外的一种实现方式是定义一个散列表，维护状态和实现。

例如，魔兽世界的开源代码就是使用了这种方式。维护了消息类型和回调函数指针的一个数组。

 

还有一种方式是设计模式中的状态模式。不过，是不是该计算一下，状态不断增加后，将会有多少个State的派生类。

 

最后一种使用宏定义实现的比较直观。

 

看来wtl的源码，我用宏写了一个类似的小例子。

虽然和传统意义上的状态机有些分别，（不是处理一个状态自动进入下一个状态）但是对于网络中接到一个数据包做相应的逻辑处理的程序已经够用了。

 

稍后，将实现能够自动切换状态的状态机。

struct SPacket //网络包 { WORD m_wCmd; //网络命令 DWORD m_dwSize;//包大小 DWORD m_pData; //为了方便，假设数据类型是DWORD类型的。 }; struct SMyData //自定义数据 { }; class CNetStream //网络数据流，例如socket实现 { public: //为了方便测试，加了一个参数。 SPacket* GetOneNetData(int n) { SPacket* pNetData=new SPacket(); pNetData->m_wCmd=n; pNetData->m_pData=n+100; return pNetData; } void ReleaseNetData(SPacket* pPaket) { delete pPaket; pPaket=NULL; } }; #define BEGIN_MSG_MAP(theClass) / public: / BOOL ProcessMessage(UINT uMsg, SPacket* aSystemData, void* aUserData, LRESULT& lResult) / { / uMsg; / aSystemData; / aUserData; / lResult; / / if(uMsg==0)/ return TRUE;/ #define MESSAGE_CMD_RANGE_HANDLER(msgFirst, msgLast, func) / if(uMsg >= msgFirst && uMsg <= msgLast) / { / lResult = func(uMsg, aSystemData, aUserData); / } #define MESSAGE_CMD_HANDLER(id, func) / if(uMsg == id ) / { / lResult = func(aSystemData, aUserData); / } #define END_MSG_MAP() / return TRUE; / } class CState { public: enum StateMsg { Cmd1=10, Cmd2, Cmd3, }; CState() { m_pStream=new CNetStream(); m_bShuntDown=FALSE; } ~CState() { delete m_pStream; m_pStream=NULL; } BEGIN_MSG_MAP(CState) MESSAGE_CMD_HANDLER(Cmd1, HandleCmd1) MESSAGE_CMD_HANDLER(Cmd2, HandleCmd2) END_MSG_MAP() //消息驱动器 void Run() { LRESULT lResult; int nCount=0; SPacket* pPacket=NULL; while(TRUE) { if(m_bShuntDown) { break; } pPacket=m_pStream->GetOneNetData(nCount); ProcessMessage(pPacket->m_wCmd, pPacket,NULL, lResult); m_pStream->ReleaseNetData(pPacket); nCount++; Sleep(500); } } void SendExitCmd() { m_bShuntDown=TRUE; } private: LRESULT HandleCmd1(SPacket* pSystemData, void* aUserData) { cout<<pSystemData->m_pData<<endl;//用户逻辑1，输出数据 return 0; } LRESULT HandleCmd2(SPacket* pSystemData, void* aUserData) { cout<<&(pSystemData->m_pData)<<endl;//用户逻辑2，输出数据地址 SendExitCmd();//发送退出消息 return 0; } private: CNetStream* m_pStream; BOOL m_bShuntDown; }; void main() { cout<<"state begin"<<endl; CState aState; aState.Run(); cout<<"state over"<<endl; }

其中，定义了一些数据类型，需要包含windows.h的头文件。

 

自动切换状态的状态机

#include "StdAfx.h" struct SPacket //网络包 { WORD m_wCmd; //网络命令 DWORD m_dwSize;//包大小 DWORD m_pData; //为了方便，假设数据类型是DWORD类型的。 }; struct SMyData //自定义数据 { }; class CNetStream //网络数据流，例如socket实现 { public: //为了方便测试，加了一个参数。 SPacket* GetOneNetData(int n) { SPacket* pNetData=new SPacket(); pNetData->m_wCmd=n; pNetData->m_pData=n+100; return pNetData; } void ReleaseNetData(SPacket* pPaket) { delete pPaket; pPaket=NULL; } }; #define BEGIN_MSG_MAP(theClass,initState) / public: / StateMsg m_State;/ BOOL ProcessMessage(SPacket* aSystemData, void* aUserData, LRESULT& lResult) / { / aSystemData; / aUserData; / lResult; / m_State=initState; / while(TRUE)/ {/ switch(m_State)/ {/ case CmdExit:/ goto end;/ break; #define MESSAGE_CMD_HANDLER(curr,next, func) / case curr : / { / lResult = func(aSystemData, aUserData); / m_State=next;/ }/ break;/ #define END_MSG_MAP() / }/ }/ end: return TRUE; / } class CState { public: enum StateMsg { CmdExit=0xFFFF, Cmd1=10, Cmd2, Cmd3, }; CState() { m_pStream=new CNetStream(); m_bShuntDown=FALSE; } ~CState() { delete m_pStream; m_pStream=NULL; } BEGIN_MSG_MAP(CState,Cmd1) MESSAGE_CMD_HANDLER(Cmd1,Cmd2, HandleCmd1) MESSAGE_CMD_HANDLER(Cmd2,CmdExit, HandleCmd2) END_MSG_MAP() //消息驱动器 void Run() { LRESULT lResult; SPacket aPacket; aPacket.m_pData=10; ProcessMessage(&aPacket,NULL, lResult); } void SendExitCmd() { m_bShuntDown=TRUE; } private: LRESULT HandleCmd1(SPacket* pSystemData, void* aUserData) { cout<<pSystemData->m_pData<<endl;//用户逻辑1，输出数据 return 0; } LRESULT HandleCmd2(SPacket* pSystemData, void* aUserData) { cout<<&(pSystemData->m_pData)<<endl;//用户逻辑2，输出数据地址 SendExitCmd();//发送退出消息 return 0; } private: CNetStream* m_pStream; BOOL m_bShuntDown; }; void main() { cout<<"state begin"<<endl; CState aState; aState.Run(); cout<<"state over"<<endl; }

 

状态机中还有一种情况时，有些状态依赖于其他的状态，或需要等待某一条件打到后才执行。

可以设计MESSAGE_CMD_HANDLER(Cmd1,Cmd2, HandleCmd1，TRUE)

或MESSAGE_CMD_HANDLER(Cmd1,Cmd2, HandleCmd1，FALSE)

来标识是否立即执行下一状态。

这个功能后续实现。