/// <summary>
/// Receive a message from the WebSocket server.
/// </summary>
/// <param name="buffer">An ArraySegment of byte that was returned</param>
/// <returns></returns>
public PPError Receive(ArraySegment <byte> buffer)
{
ThrowIfNotConnected();
// |receiveVar| must be valid until |callback| is invoked.
// Just use a member variable.
return((PPError)PPBWebSocket.ReceiveMessage(this, out receiveVar, new CompletionCallback <ArraySegment <byte> >(OnReceiveData, buffer)));
}
private async Task <WebSocketReceiveResult> ReceiveAsyncCore(ArraySegment <byte> buffer, MessageLoop receiveLoop = null)
{
var tcs = new TaskCompletionSource <WebSocketReceiveResult>();
EventHandler <WebSocketReceiveResult> handler = (s, e) => { tcs.TrySetResult(e); };
try
{
ReceiveData += handler;
if (MessageLoop == null)
{
//Console.WriteLine("Receive no message loop");
var receiveResult = Receive(buffer);
if (receiveResult != PPError.Ok && !tcs.Task.IsCompleted)
{
tcs.TrySetResult(new WebSocketReceiveResult(0, WebSocketMessageType.Close, true));
}
}
else
{
Action <PPError> action = new Action <PPError>((e) =>
{
var rcvMsgResult = (PPError)PPBWebSocket.ReceiveMessage(this, out receiveVar, new BlockUntilComplete());
WebSocketReceiveResult receiveResult = null;
var recVar = (Var)receiveVar;
if (State == WebSocketState.Open)
{
if (recVar.IsArrayBuffer)
{
var arrayBuffer = new VarArrayBuffer(receiveVar);
var size = (uint)Math.Min(buffer.Count, arrayBuffer.ByteLength);
int offs = 0;
var data = arrayBuffer.Map();
for (offs = 0; offs < size; offs++)
{
buffer.Array[offs] = data[offs];
}
arrayBuffer.Unmap();
receiveResult = new WebSocketReceiveResult(size, WebSocketMessageType.Binary, true);
}
else
{
var msg = Encoding.UTF8.GetBytes(recVar.AsString());
var size = (uint)Math.Min(buffer.Count, msg.Length);
int offs = 0;
for (offs = 0; offs < size; offs++)
{
buffer.Array[offs] = msg[offs];
}
receiveResult = new WebSocketReceiveResult(size, WebSocketMessageType.Text, true);
}
}
else
{
receiveResult = new WebSocketReceiveResult(0, WebSocketMessageType.Close, true, CloseStatus, CloseStatusDescription);
}
tcs.TrySetResult(receiveResult);
}
);
InvokeHelper(action, receiveLoop);
}
return(await tcs.Task);
}
catch (Exception exc)
{
Console.WriteLine(exc.Message);
tcs.SetException(exc);
return(new WebSocketReceiveResult(0, WebSocketMessageType.Close, true));
}
finally
{
ReceiveData -= handler;
}
}