154 lines
3.0 KiB
C++
154 lines
3.0 KiB
C++
#include "BinaryPair.h"
|
||
|
||
// 初始化消息，各字段为0
|
||
/*BinaryPair::BinaryPair(Buffer& bs)
|
||
{
|
||
Data = bs.GetBuffer();
|
||
Length = bs.Length();
|
||
Position = 0;
|
||
}*/
|
||
|
||
BinaryPair::BinaryPair(Stream& ms)
|
||
{
|
||
_s = &ms;
|
||
_p = ms.Position();
|
||
}
|
||
|
||
Buffer BinaryPair::Get(const String& name) const
|
||
{
|
||
//if(!Data || !Length) return Buffer(nullptr, 0);
|
||
// 暂时不方便支持空名称的名值对，而服务端是支持的
|
||
//if(!name) return Buffer(nullptr, 0);
|
||
|
||
// 从当前位置开始向后找，如果找不到，再从头开始找到当前位置。
|
||
// 这样子安排，如果是顺序读取，将会提升性能
|
||
|
||
//Stream ms(Data, Length);
|
||
auto& ms = *_s;
|
||
uint p = ms.Position();
|
||
for(int i=0; i<2; i++)
|
||
{
|
||
while(ms.Remain() && (i ==0 || ms.Position() < p))
|
||
{
|
||
int len = ms.ReadEncodeInt();
|
||
auto nm = ms.ReadBytes(len);
|
||
int ln2 = ms.ReadEncodeInt();
|
||
auto dt = ms.ReadBytes(ln2);
|
||
|
||
if(name == nm) return Buffer(dt, ln2);
|
||
}
|
||
|
||
// 从头开始再来一次
|
||
if(p == _p) break;
|
||
|
||
ms.SetPosition(_p);
|
||
}
|
||
|
||
return Buffer(nullptr, 0);
|
||
}
|
||
|
||
bool BinaryPair::Set(const String& name, const Buffer& bs)
|
||
{
|
||
//if(Position >= Length) return false;
|
||
|
||
//Stream ms(Data + Position, Length - Position);
|
||
auto& ms = *_s;
|
||
ms.WriteArray(name);
|
||
ms.WriteArray(bs);
|
||
|
||
return true;
|
||
}
|
||
|
||
bool BinaryPair::Get(const String& name, byte& value) const
|
||
{
|
||
auto bs = Get(name);
|
||
if(!bs.Length()) return false;
|
||
|
||
value = bs[0];
|
||
|
||
return true;
|
||
}
|
||
|
||
bool BinaryPair::Get(const String& name, ushort& value) const
|
||
{
|
||
auto bs = Get(name);
|
||
if(!bs.Length()) return false;
|
||
|
||
value = bs.ToUInt16();
|
||
|
||
return true;
|
||
}
|
||
|
||
bool BinaryPair::Get(const String& name, uint& value) const
|
||
{
|
||
auto bs = Get(name);
|
||
if(!bs.Length()) return false;
|
||
|
||
value = bs.ToUInt32();
|
||
|
||
return true;
|
||
}
|
||
|
||
bool BinaryPair::Get(const String& name, UInt64& value) const
|
||
{
|
||
auto bs = Get(name);
|
||
if(!bs.Length()) return false;
|
||
|
||
value = bs.ToUInt64();
|
||
|
||
return true;
|
||
}
|
||
|
||
bool BinaryPair::Get(const String& name, Buffer& value) const
|
||
{
|
||
auto bs = Get(name);
|
||
if(!bs.Length()) return false;
|
||
|
||
value = bs;
|
||
|
||
return true;
|
||
}
|
||
|
||
bool BinaryPair::Get(const String& name, IPEndPoint& value) const
|
||
{
|
||
auto bs = Get(name);
|
||
if(bs.Length() < 6) return false;
|
||
if (bs[0] == 6)return false; // 单片机这边不支持ipv6
|
||
Buffer bs2(bs.GetBuffer() + 1, 6);
|
||
value = bs2;
|
||
|
||
return true;
|
||
}
|
||
|
||
|
||
bool BinaryPair::Set(const String& name, byte value)
|
||
{
|
||
return Set(name, Buffer(&value, 1));
|
||
}
|
||
|
||
bool BinaryPair::Set(const String& name, ushort value)
|
||
{
|
||
return Set(name, Buffer(&value, 2));
|
||
}
|
||
|
||
bool BinaryPair::Set(const String& name, uint value)
|
||
{
|
||
return Set(name, Buffer(&value, 4));
|
||
}
|
||
|
||
bool BinaryPair::Set(const String& name, UInt64 value)
|
||
{
|
||
return Set(name, Buffer(&value, 8));
|
||
}
|
||
|
||
bool BinaryPair::Set(const String& name, const IPEndPoint& value)
|
||
{
|
||
MemoryStream ms(7);
|
||
ms.Write(4);
|
||
ms.Write(value.ToArray());
|
||
ByteArray bs(ms.GetBuffer(), ms.Position());
|
||
return Set(name, bs);
|
||
//return Set(name, value.ToArray());
|
||
}
|
||
|