private void OnReceived(byte[] buffer, int offset, int count)
{
int remainCount = count;
int readPosition = offset;
while (remainCount > 0)
{
switch (m_ReceiveState)
{
case ReceiveState.ReceivingPackageLength:
if (ReceiveToPackageBuffer(buffer
, ref remainCount
, ref readPosition
, m_ReceiveHeaderBuffer
, PACKAGE_HEADER_SIZE))
{
// SIGN:TLength
TLength packageLength = BitConverter.ToInt32(m_ReceiveHeaderBuffer.GetBuffer()
, m_ReceiveHeaderBuffer.GetOffset());
lock (m_BufferPool)
{
m_ReceiveBodyBuffer = m_BufferPool.AllocBuffer(packageLength - PACKAGE_HEADER_SIZE);
}
m_ReceiveState = ReceiveState.ReceivingPackageBody;
}
break;
case ReceiveState.ReceivingPackageBody:
if (ReceiveToPackageBuffer(buffer
, ref remainCount
, ref readPosition
, m_ReceiveBodyBuffer
, m_ReceiveBodyBuffer.GetSize()))
{
lock (m_ReceiveLock)
{
m_ReceivedPackages.Enqueue(m_ReceiveBodyBuffer);
}
m_ReceiveBodyBuffer = null;
m_ReceiveState = ReceiveState.ReceivingPackageLength;
}
break;
}
}
}
private void ProcessConnect(object sender, SocketAsyncEventArgs e)
{
if (e.SocketError == SocketError.Success)
{
MDebug.Log(LOG_TAG, $"Client({GetName()}) 连接成功");
m_BufferPool = new ArrayPool <byte>(1024 * 512);
lock (m_TcpConnectActionLock)
{
m_TcpConnectData = new TcpConnectData(this, e, ConnectState.Connected);
}
m_IsSending = false;
lock (m_BufferPool)
{
m_SendEventArgs = new SocketAsyncEventArgs();
m_SendEventArgs.Completed += new EventHandler <SocketAsyncEventArgs>(ProcessSend);
m_SendingBuffer = m_BufferPool.AllocBuffer(DEFAULT_BUFFER_SIZE);
m_WaitSendBuffer = m_BufferPool.AllocBuffer(DEFAULT_BUFFER_SIZE);
m_WaitSendBufferSize = 0;
m_SendEventArgs.SetBuffer(m_SendingBuffer.GetBuffer(), m_SendingBuffer.GetOffset(), 0);
m_ReceiveEventArgs = new SocketAsyncEventArgs();
m_ReceiveEventArgs.Completed += new EventHandler <SocketAsyncEventArgs>(ProcessReceive);
m_ReceiveBuffer = m_BufferPool.AllocBuffer(DEFAULT_BUFFER_SIZE);
m_ReceiveEventArgs.SetBuffer(m_ReceiveBuffer.GetBuffer(), m_ReceiveBuffer.GetOffset(), m_ReceiveBuffer.GetSize());
m_ReceiveState = ReceiveState.ReceivingPackageLength;
m_ReceivedPackageCount = 0;
m_ReceiveHeaderBuffer = m_BufferPool.AllocBuffer(PACKAGE_HEADER_SIZE);
}
lock (m_ReceiveLock)
{
m_ReceivedPackages = new Queue <ArrayPool <byte> .Node>();
m_ReceivedPackages2 = new BetterList <ArrayPool <byte> .Node>(4);
m_ReceivedPackages2._SetCount(m_ReceivedPackages2.Capacity);
}
StartReceive();
}
else
{
MDebug.LogError(LOG_TAG
, $"Client({GetName()}) 连接失败: {e.SocketError.ToString()}\n{e.ConnectByNameError.ToString()}");
lock (m_TcpConnectActionLock)
{
m_TcpConnectData = new TcpConnectData(this, e, ConnectState.ConnectFailed);
}
m_Client.Close();
m_Client = null;
}
}
/// <summary>
/// 反序列化
/// <see cref="Serialize"/>
/// </summary>
public void Deserialize(ArrayPool <RpcValue> rpcValueArrayPool
, ObjectPool <RpcValue> rpcValuePool
, byte[] buffer
, ref int offset
, bool enableDeserializeValueType = true)
{
if (enableDeserializeValueType)
{
ValueType = (ValueType)buffer[offset];
offset++;
}
switch (ValueType)
{
case ValueType.Byte:
ByteValue = buffer[offset];
offset++;
break;
case ValueType.Short:
ShortValue = BitConverter.ToInt16(buffer, offset);
offset += sizeof(short);
break;
case ValueType.Int:
IntValue = BitConverter.ToInt32(buffer, offset);
offset += sizeof(int);
break;
case ValueType.Long:
LongValue = BitConverter.ToInt64(buffer, offset);
offset += sizeof(long);
break;
case ValueType.Float:
FloatValue = BitConverter.ToSingle(buffer, offset);
offset += sizeof(float);
break;
case ValueType.Double:
DoubleValue = BitConverter.ToDouble(buffer, offset);
offset += sizeof(double);
break;
case ValueType.FixedValueTypeArray:
{
int elementCount = BitConverter.ToInt32(buffer, offset);
offset += sizeof(int);
ValueType elementValueType = (ValueType)buffer[offset];
offset++;
ArrayValue = rpcValueArrayPool.AllocBuffer(elementCount);
int endPoint = ArrayValue.GetOffset() + ArrayValue.GetSize();
RpcValue[] elements = ArrayValue.GetBuffer();
for (int iElement = ArrayValue.GetOffset(); iElement < endPoint; iElement++)
{
RpcValue iterElement = rpcValuePool.Alloc();
elements[iElement] = iterElement;
iterElement.ValueType = elementValueType;
iterElement.Deserialize(rpcValueArrayPool
, rpcValuePool
, buffer
, ref offset
, false);
}
}
break;
case ValueType.VariableValueTypeArray:
{
int elementCount = BitConverter.ToInt32(buffer, offset);
offset += sizeof(int);
ArrayValue = rpcValueArrayPool.AllocBuffer(elementCount);
int endPoint = ArrayValue.GetOffset() + ArrayValue.GetSize();
RpcValue[] elements = ArrayValue.GetBuffer();
for (int iElement = ArrayValue.GetOffset(); iElement < endPoint; iElement++)
{
RpcValue iterElement = rpcValuePool.Alloc();
elements[iElement] = iterElement;
iterElement.Deserialize(rpcValueArrayPool
, rpcValuePool
, buffer
, ref offset);
}
}
break;
default:
throw new Exception("Not support ValueType: " + ValueType);
}
}
public override void OnUpdate(float deltaTime)
{
lock (m_TcpConnectActionLock)
{
if (m_TcpConnectData != null)
{
try
{
switch (m_TcpConnectData.ConnectState)
{
case ConnectState.Connected:
OnConnected?.Invoke(m_TcpConnectData);
break;
case ConnectState.ConnectFailed:
OnConnectFailed?.Invoke(m_TcpConnectData);
break;
case ConnectState.Disconnected:
OnDisconnected?.Invoke(m_TcpConnectData);
break;
default:
MDebug.Assert(false, LOG_TAG, "Not handle ConnectState: " + m_TcpConnectData.ConnectState);
break;
}
}
catch (Exception e)
{
MDebug.LogError(LOG_TAG
, $"Inove connect:{m_TcpConnectData.ConnectState} Exception:\n{e.ToString()}");
}
m_TcpConnectData = null;
}
}
if (!IsConnected() ||
m_ReceivedPackages2 == null)
{
return;
}
int packageCount;
lock (m_ReceiveLock)
{
packageCount = m_ReceivedPackages.Count;
if (m_ReceivedPackages2.Capacity < packageCount)
{
m_ReceivedPackages2.Capacity = m_ReceivedPackages2.Capacity * 2;
m_ReceivedPackages2._SetCount(m_ReceivedPackages2.Capacity);
}
for (int iPackage = 0; iPackage < packageCount; iPackage++)
{
m_ReceivedPackages2[iPackage] = m_ReceivedPackages.Dequeue();
}
#if GF_DEBUG
MDebug.Assert(m_ReceivedPackages.Count == 0, LOG_TAG, "m_ReceivedPackages.Count == 0");
#endif
}
if (packageCount > 0)
{
if (OnReceivedPackage != null)
{
for (int iPackage = 0; iPackage < packageCount; iPackage++)
{
ArrayPool <byte> .Node iterPackage = m_ReceivedPackages2[iPackage];
try
{
OnReceivedPackage.Invoke(iterPackage);
}
catch (Exception e)
{
MDebug.LogError(LOG_TAG
, $"Invoke OnReceivedPackage length: {iterPackage.GetSize()}. Exception:\n{e.ToString()}");
}
}
}
lock (m_BufferPool)
{
for (int iPackage = 0; iPackage < packageCount; iPackage++)
{
m_BufferPool.ReleaseBuffer(m_ReceivedPackages2[iPackage]);
m_ReceivedPackages2[iPackage] = null;
}
}
}
}
/// <summary>
/// 序列化
/// </summary>
/// <param name="buffer">目标Buffer</param>
/// <param name="offset">写入位置</param>
/// <param name="enableSerializeValueType">
/// 是否序列化<see cref="ValueType"/>
/// 需要这个开关是因为序列化<see cref="ValueType.FixedValueTypeArray"/>时，不需要序列化数组中每个元素的<see cref="ValueType"/>
/// </param>
public void Serialize(byte[] buffer
, ref int offset
, bool enableSerializeValueType = true)
{
if (enableSerializeValueType)
{
MBitConverter.WriteTo(buffer, ref offset, (byte)ValueType);
}
switch (ValueType)
{
case ValueType.Byte:
MBitConverter.WriteTo(buffer, ref offset, ByteValue);
break;
case ValueType.Short:
MBitConverter.WriteTo(buffer, ref offset, ShortValue);
break;
case ValueType.Int:
MBitConverter.WriteTo(buffer, ref offset, IntValue);
break;
case ValueType.Long:
MBitConverter.WriteTo(buffer, ref offset, LongValue);
break;
case ValueType.Float:
MBitConverter.WriteTo(buffer, ref offset, FloatValue);
break;
case ValueType.Double:
MBitConverter.WriteTo(buffer, ref offset, DoubleValue);
break;
case ValueType.FixedValueTypeArray:
{
int elementCount = ArrayValue.GetSize();
MBitConverter.WriteTo(buffer, ref offset, elementCount);
MDebug.Assert(elementCount > 0
, "Rpc"
, "elementCount > 0");
int startPoint = ArrayValue.GetOffset();
int endPoint = startPoint + elementCount;
RpcValue[] elements = ArrayValue.GetBuffer();
ValueType elementValueType = elements[startPoint].ValueType;
MBitConverter.WriteTo(buffer, ref offset, (byte)elementValueType);
for (int iElement = startPoint; iElement < endPoint; iElement++)
{
RpcValue iterElement = elements[iElement];
#if GF_DEBUG
MDebug.Assert(iterElement.ValueType == elementValueType
, "Rpc"
, "iterElement.ValueType == elementValueType");
#endif
iterElement.Serialize(buffer, ref offset, false);
}
}
break;
case ValueType.VariableValueTypeArray:
{
int elementCount = ArrayValue.GetSize();
MBitConverter.WriteTo(buffer, ref offset, elementCount);
MDebug.Assert(elementCount > 0
, "Rpc"
, "elementCount > 0");
int startPoint = ArrayValue.GetOffset();
int endPoint = startPoint + elementCount;
RpcValue[] elements = ArrayValue.GetBuffer();
for (int iElement = startPoint; iElement < endPoint; iElement++)
{
RpcValue iterElement = elements[iElement];
iterElement.Serialize(buffer, ref offset);
}
}
break;
default:
throw new Exception("Not support ValueType: " + ValueType);
}
}
