public static void Handle(byte[] payloadBytes, ConnectionInfo connectionInfo)
{
var payloadString = Encoding.UTF8.GetString(payloadBytes);
if (payloadString.StartsWith("subscribe#"))
{
connectionInfo.Topic = payloadString.Split("subscribe#").LastOrDefault();
ConnectionStorage.Add(connectionInfo);
}
else
{
PayLoad payload = JsonConvert.DeserializeObject <PayLoad>(payloadString);
PayloadStorage.Add(payload);
}
}
private void ReceiveCallBack(IAsyncResult asyncResult)
{
ConnectionInfo connection = asyncResult.AsyncState as ConnectionInfo;
try
{
Socket senderSocket = connection.Socket;
SocketError response;
int buffSize = senderSocket.EndReceive(asyncResult, out response);
if (response == SocketError.Success)
{
byte[] payload = new byte[buffSize];
Array.Copy(connection.Data, payload, payload.Length);
PayloadHandler.Handle(payload, connection);
}
}catch (Exception e)
{
Console.WriteLine($"Can't receive data: {e.Message}");
}
finally
{
try
{
connection.Socket.BeginReceive(connection.Data, 0, connection.Data.Length, SocketFlags.None, ReceiveCallBack, connection);
}
catch (Exception e)
{
Console.WriteLine($"{e.Message}");
var address = connection.Socket.RemoteEndPoint.ToString();
ConnectionStorage.Remove(address);
connection.Socket.Close();
}
}
}
public void DoSendMessageWork()
{
while (true)
{
while (!PayloadStorage.IsEmpty())
{
var payload = PayloadStorage.GetNext();
if (payload != null)
{
var connections = ConnectionStorage.GetConnectionInfosByTopic(payload.Topic);
foreach (var connection in connections)
{
var payloadString = JsonConvert.SerializeObject(payload);
byte[] data = Encoding.UTF8.GetBytes(payloadString);
connection.Socket.Send(data);
}
}
}
Thread.Sleep(TIME_TO_SLEEP);
}
}
