72 lines
1.9 KiB
C++
72 lines
1.9 KiB
C++
#ifndef __Controller_H__
|
|
#define __Controller_H__
|
|
|
|
#include "Sys.h"
|
|
#include "List.h"
|
|
#include "Net\ITransport.h"
|
|
#include "Stream.h"
|
|
#include "Timer.h"
|
|
|
|
#include "Message.h"
|
|
|
|
// 处理消息，返回是否成功
|
|
typedef bool (*MessageHandler)(Message& msg, void* param);
|
|
|
|
// 消息控制器。负责发送消息、接收消息、分发消息
|
|
class Controller
|
|
{
|
|
private:
|
|
void Init();
|
|
static uint Dispatch(ITransport* transport, byte* buf, uint len, void* param);
|
|
bool Dispatch(Stream& ms, ITransport* port);
|
|
|
|
protected:
|
|
List<ITransport*> _ports; // 数据传输口数组
|
|
//byte _portCount; // 数据传输口个数
|
|
byte MinSize; // 最小消息大小
|
|
|
|
// 收到消息校验后调用该函数。返回值决定消息是否有效，无效消息不交给处理器处理
|
|
virtual bool Valid(Message& msg, ITransport* port);
|
|
// 接收处理
|
|
virtual bool OnReceive(Message& msg, ITransport* port);
|
|
|
|
public:
|
|
Controller(ITransport* port);
|
|
Controller(ITransport* ports[], int count);
|
|
virtual ~Controller();
|
|
|
|
// 添加传输口
|
|
void AddTransport(ITransport* port);
|
|
|
|
// 创建消息
|
|
virtual Message* Create() const = 0;
|
|
// 发送消息，传输口参数为空时向所有传输口发送消息
|
|
virtual int Send(Message& msg, ITransport* port = NULL);
|
|
// 回复对方的请求消息
|
|
virtual int Reply(Message& msg, ITransport* port = NULL);
|
|
|
|
// 发送核心
|
|
protected:
|
|
|
|
// 处理器部分
|
|
private:
|
|
struct HandlerLookup
|
|
{
|
|
uint Code; // 代码
|
|
MessageHandler Handler;// 处理函数
|
|
void* Param; // 参数
|
|
};
|
|
List<HandlerLookup*> _Handlers;
|
|
//byte _HandlerCount;
|
|
|
|
public:
|
|
// 注册消息处理器。考虑到业务情况，不需要取消注册
|
|
void Register(byte code, MessageHandler handler, void* param = NULL);
|
|
|
|
// 收到消息时触发
|
|
MessageHandler Received;
|
|
void* Param;
|
|
};
|
|
|
|
#endif
|