54 lines
1.5 KiB
C#
54 lines
1.5 KiB
C#
namespace mROA.Implementation;
|
|
|
|
public class StreamBasedInteractionModule : IInteractionModule
|
|
{
|
|
private Dictionary<int, Stream> _streams = new();
|
|
private Action<int, byte[]> _handler;
|
|
|
|
public void RegisterClient(Stream stream)
|
|
{
|
|
var id = Random.Shared.Next();
|
|
_streams.Add(id, stream);
|
|
ListenTo((id, stream), _handler);
|
|
}
|
|
|
|
public void SetMessageHandler(Action<int, byte[]> handler)
|
|
{
|
|
_handler = handler;
|
|
}
|
|
|
|
public void SendTo(int clientId, byte[] message)
|
|
{
|
|
if (!_streams.TryGetValue(clientId, out var stream))
|
|
{
|
|
throw new KeyNotFoundException($"Client {clientId} not found");
|
|
}
|
|
|
|
stream.Write(BitConverter.GetBytes((ushort)message.Length), 0, sizeof(ushort));
|
|
stream.Write(message, 0, message.Length);
|
|
|
|
}
|
|
|
|
private async Task ListenTo((int id, Stream stream) client, Action<int, byte[]> action)
|
|
{
|
|
const int bufferSize = ushort.MaxValue;
|
|
try
|
|
{
|
|
byte[] buffer = new byte[bufferSize];
|
|
while (client.stream.CanRead)
|
|
{
|
|
await client.stream.ReadExactlyAsync(buffer, 0, 2);
|
|
var len = BitConverter.ToUInt16(buffer, 0);
|
|
await client.stream.ReadExactlyAsync(buffer, 0, len);
|
|
|
|
action(client.id, buffer[..len]);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Console.WriteLine($"Client handling finished:{client.id}");
|
|
_streams.Remove(client.id);
|
|
}
|
|
|
|
}
|
|
} |