Новая система сетевого взаимодействия обложена тестами

This commit is contained in:
2025-02-16 11:45:08 +03:00
parent 82ff0019e1
commit 5e9c2cbb7b
3 changed files with 137 additions and 0 deletions
+12
View File
@@ -1,3 +1,5 @@
using mROA.Implementation;
namespace mROA.Abstract;
public interface IInteractionModule : IInjectableModule
@@ -11,4 +13,14 @@ public interface IInteractionModule : IInjectableModule
public Task<byte[]> ReceiveMessage();
public void PostMessage(byte[] message);
}
}
public interface INextGenerationInteractionModule : IInjectableModule
{
int ConntectionId { get; }
public Stream? BaseStream { get; set; }
Task<NetworkMessage> GetNextMessageReceiving();
void PostMessage(NetworkMessage message);
NetworkMessage[] UnhandledMessages { get; }
void HandleMessage(NetworkMessage msg);
}
+54
View File
@@ -0,0 +1,54 @@
using mROA.Abstract;
using mROA.Implementation;
namespace mROA;
public class NextGenerationInteractionModule : INextGenerationInteractionModule
{
private ISerializationToolkit _serialization;
public int ConntectionId { get; set; }
public Stream BaseStream { get; set; }
private Task<NetworkMessage>? _currentReceiving;
private const int BufferSize = ushort.MaxValue;
private byte[] _buffer = new byte[BufferSize];
private List<NetworkMessage> _unhandledMessages = new(8);
public void Inject<T>(T dependency)
{
if (dependency is ISerializationToolkit toolkit)
_serialization = toolkit;
}
public Task<NetworkMessage> GetNextMessageReceiving()
{
if (_currentReceiving is { IsCompleted: false })
return _currentReceiving;
_currentReceiving = GetNextMessage();
return _currentReceiving;
}
public void PostMessage(NetworkMessage message)
{
var rawMessage = _serialization.Serialize(message);
BaseStream.Write(BitConverter.GetBytes((ushort)rawMessage.Length), 0, sizeof(ushort));
BaseStream.Write(rawMessage, 0, rawMessage.Length);
}
public NetworkMessage[] UnhandledMessages => _unhandledMessages.ToArray();
public void HandleMessage(NetworkMessage msg)
{
_unhandledMessages.Remove(msg);
}
private async Task<NetworkMessage> GetNextMessage()
{
await BaseStream.ReadExactlyAsync(_buffer, 0, 2);
var len = BitConverter.ToUInt16(_buffer, 0);
await BaseStream.ReadExactlyAsync(_buffer, 0, len);
return _serialization.Deserialize<NetworkMessage>(_buffer[..len])!;
}
}