root/BITKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

1

Browse Source
This commit is contained in:
CortexCore
2024-06-11 15:39:09 +08:00
parent 3e8bd54a08
commit 0e1bf20595
3 changed files with 86 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
Src/Core/Kcp/KcpNetClient.cs
View File

@@ -170,8 +170,16 @@ namespace BITKit.Net
var id = reader.ReadInt32();
try
{
var value = BITBinary.Read(reader);
_p2p.TryAdd(id,value);
if (reader.ReadBoolean())
{
var value = BITBinary.Read(reader);
_p2p.TryAdd(id,value);
}
else
{
var message = reader.ReadString();
_p2p.TryAdd(id,new Exception(message));
}
}
catch (Exception e)
{
@@ -189,21 +197,30 @@ namespace BITKit.Net
{
throw new NotImplementedException($"未找到对应的方法:{commandObj.GetType().FullName}");
}
var value = await func.As<Func<object, UniTask<object>>>().Invoke(commandObj);
using var _ms = new MemoryStream();
using var _writer = new BinaryWriter(_ms);
_writer.Write((byte)NetCommandType.ReturnToServer);
_writer.Write(requestId);
BITBinary.Write(_writer, value);
var _bytes = _ms.ToArray();
commandQueue.Enqueue(_bytes);
try
{
var value = await func.As<Func<object, UniTask<object>>>().Invoke(commandObj);
_writer.Write(true);
BITBinary.Write(_writer, value);
var _bytes = _ms.ToArray();
commandQueue.Enqueue(_bytes);
}
catch (Exception e)
{
BIT4Log.LogException(e);
_writer.Write(false);
_writer.Write(e.Message);
}
}
catch(Exception e)
{
BIT4Log.LogException(e);
}
break;
default:
BIT4Log.Log<KcpClient>($"未知消息类型:{type},字节:{(byte)type}");
@@ -280,6 +297,10 @@ namespace BITKit.Net
if (_p2p.TryRemove(id, out var value))
{
if (value is Exception e)
{
throw e;
}
return (T)value;
}
await Task.Delay(100);

59
Src/Core/Kcp/KcpNetServer.cs
View File

@@ -189,36 +189,59 @@ namespace BITKit.Net
server.Send(Id,new byte[]{(byte)NetCommandType.Ping},channel);
break;
case NetCommandType.GetFromServer:
{
var requestId = reader.ReadInt32();
var commandObj = BITBinary.Read(reader);
using var _ms = new MemoryStream();
using var _writer = new BinaryWriter(_ms);
_writer.Write((byte)NetCommandType.ReturnToClient);
_writer.Write(requestId);
try
{
var requestId = reader.ReadInt32();
var commandObj = BITBinary.Read(reader);
if (_rpc.TryGetValue(commandObj.GetType().FullName, out var func) is false)
if (_rpc.TryGetValue(commandObj.GetType()!.FullName!, out var func) is false)
{
throw new NotImplementedException($"未找到对应的方法:{commandObj.GetType().FullName}");
_writer.Write(false);
_writer.Write("未找到对应的Rpc方法");
}
else
{
try
{
var value = await func.As<Func<object, UniTask<object>>>().Invoke(commandObj);
_writer.Write(true);
BITBinary.Write(_writer, value);
}
catch (Exception e)
{
BIT4Log.LogException(e);
_writer.Write(false);
_writer.Write(e.Message);
}
}
var value =await func.As<Func<object,UniTask<object>>>().Invoke(commandObj);
using var _ms = new MemoryStream();
using var _writer = new BinaryWriter(_ms);
_writer.Write((byte)NetCommandType.ReturnToClient);
_writer.Write(requestId);
BITBinary.Write(_writer,value);
var _bytes = _ms.ToArray();
server.Send(Id,_bytes,KcpChannel.Reliable);
server.Send(Id, _bytes, KcpChannel.Reliable);
}
catch (Exception e)
{
BIT4Log.LogException(e);
Send(Id,NetCommandType.ReturnToClient,-1,0);
}
}
break;
case NetCommandType.ReturnToServer:
{
var id = reader.ReadInt32();
var value = BITBinary.Read(reader);
_p2p.TryAdd(id, value);
if (reader.ReadBoolean())
{
var value = BITBinary.Read(reader);
_p2p.TryAdd(id, value);
}
else
{
var message = reader.ReadString();
_p2p.TryAdd(id, new Exception(message));
}
}
break;
default:
@@ -274,6 +297,10 @@ namespace BITKit.Net
if (_p2p.TryRemove(index, out var value))
{
if (value is Exception e)
{
throw e;
}
return (T)value;
}
await Task.Delay(100);