public async Task MessageLoop()
{
try
{
while (true)
{
// Get next packet size
var bufferSize = await context.ReadStream.ReadInt32Async();
// Get next packet data (until complete)
var buffer = new byte[bufferSize];
await context.ReadStream.ReadAllAsync(buffer, 0, bufferSize);
// Deserialize as an object
var binaryReader = new BinarySerializationReader(new MemoryStream(buffer));
object obj = null;
binaryReader.Context.SerializerSelector = SerializerSelector.AssetWithReuse;
binaryReader.SerializeExtended <object>(ref obj, ArchiveMode.Deserialize, null);
// If it's a message, process it separately (StreamId)
if (obj is SocketMessage)
{
var socketMessage = (SocketMessage)obj;
ProcessMessage(socketMessage);
}
// Check if there is a specific handler for this packet type
bool handlerFound;
Tuple <Action <object>, bool> handler;
lock (packetHandlers)
{
handlerFound = packetHandlers.TryGetValue(obj.GetType(), out handler);
// one-time handler
if (handlerFound && handler.Item2)
{
packetHandlers.Remove(obj.GetType());
}
}
if (handlerFound)
{
handler.Item1(obj);
}
}
}
catch (Exception)
{
context.Dispose();
throw;
}
}
/// <summary>
/// Handles ServerStarted messages. It happens when service opened a new "server" connection back to us.
/// </summary>
/// <param name="clientSocket">The client socket context.</param>
/// <returns></returns>
private async Task HandleMessageServerStarted(SimpleSocket clientSocket)
{
var guid = await clientSocket.ReadStream.ReadGuidAsync();
var errorCode = await clientSocket.ReadStream.ReadInt32Async();
var errorMessage = (errorCode != 0) ? await clientSocket.ReadStream.ReadStringAsync() : null;
// Notify any waiter that a server with given GUID is available
TaskCompletionSource<SimpleSocket> serverSocketTCS;
lock (pendingServers)
{
if (!pendingServers.TryGetValue(guid, out serverSocketTCS))
{
Log.Error("Could not find a matching server Guid");
clientSocket.Dispose();
return;
}
pendingServers.Remove(guid);
}
if (errorCode != 0)
serverSocketTCS.TrySetException(new Exception(errorMessage));
else
serverSocketTCS.TrySetResult(clientSocket);
}
