mirror of
https://github.com/tiennm99/zfoo.git
synced 2026-10-26 04:17:49 +00:00
43 lines
1.2 KiB
C++
43 lines
1.2 KiB
C++
#ifndef ZFOO_PROTOCOLMANAGER_H
|
|
#define ZFOO_PROTOCOLMANAGER_H
|
|
|
|
#include "ByteBuffer.h"
|
|
#include "zfoocpp/ComplexObject.h"
|
|
#include "zfoocpp/NormalObject.h"
|
|
#include "zfoocpp/ObjectA.h"
|
|
#include "zfoocpp/ObjectB.h"
|
|
#include "zfoocpp/SimpleObject.h"
|
|
|
|
namespace zfoo {
|
|
|
|
const int16_t MAX_PROTOCOL_NUM = 32767;
|
|
const IProtocolRegistration *protocols[MAX_PROTOCOL_NUM];
|
|
|
|
void initProtocol() {
|
|
protocols[100] = new ComplexObjectRegistration();
|
|
protocols[101] = new NormalObjectRegistration();
|
|
protocols[102] = new ObjectARegistration();
|
|
protocols[103] = new ObjectBRegistration();
|
|
protocols[104] = new SimpleObjectRegistration();
|
|
}
|
|
|
|
inline IProtocolRegistration *getProtocol(int16_t protocolId) {
|
|
return const_cast<IProtocolRegistration *>(protocols[protocolId]);
|
|
}
|
|
|
|
void write(ByteBuffer &buffer, IPacket *packet) {
|
|
auto protocolId = packet->protocolId();
|
|
// 写入协议号
|
|
buffer.writeShort(protocolId);
|
|
// 写入包体
|
|
getProtocol(protocolId)->write(buffer, packet);
|
|
}
|
|
|
|
IPacket *read(ByteBuffer &buffer) {
|
|
auto protocolId = buffer.readShort();
|
|
return getProtocol(protocolId)->read(buffer);
|
|
}
|
|
|
|
}
|
|
#endif
|