Add WebAssembly session
This commit is contained in:
parent
a0e45ab61a
commit
91627643e2
|
@ -97,7 +97,7 @@ namespace Ooui.AspNetCore
|
||||||
//
|
//
|
||||||
var token = CancellationToken.None;
|
var token = CancellationToken.None;
|
||||||
var webSocket = await context.WebSockets.AcceptWebSocketAsync ("ooui");
|
var webSocket = await context.WebSockets.AcceptWebSocketAsync ("ooui");
|
||||||
var session = new Ooui.UI.Session (webSocket, activeSession.Element, w, h, token);
|
var session = new Ooui.WebSocketSession (webSocket, activeSession.Element, w, h, token);
|
||||||
await session.RunAsync ().ConfigureAwait (false);
|
await session.RunAsync ().ConfigureAwait (false);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -133,7 +133,7 @@ namespace Xamarin.Forms
|
||||||
{
|
{
|
||||||
if (timer != null)
|
if (timer != null)
|
||||||
return;
|
return;
|
||||||
var interval = TimeSpan.FromSeconds (1.0 / Ooui.UI.Session.MaxFps);
|
var interval = TimeSpan.FromSeconds (1.0 / Ooui.UI.MaxFps);
|
||||||
timer = new Timer ((_ => {
|
timer = new Timer ((_ => {
|
||||||
this.SendSignals ();
|
this.SendSignals ();
|
||||||
}), null, (int)interval.TotalMilliseconds, (int)interval.TotalMilliseconds);
|
}), null, (int)interval.TotalMilliseconds, (int)interval.TotalMilliseconds);
|
||||||
|
|
|
@ -0,0 +1,103 @@
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
|
||||||
|
namespace Ooui
|
||||||
|
{
|
||||||
|
public abstract class Session
|
||||||
|
{
|
||||||
|
protected readonly Element element;
|
||||||
|
protected readonly double initialWidth;
|
||||||
|
protected readonly double initialHeight;
|
||||||
|
|
||||||
|
protected readonly HashSet<string> createdIds;
|
||||||
|
|
||||||
|
protected readonly List<Message> queuedMessages = new List<Message> ();
|
||||||
|
|
||||||
|
public Session (Element element, double initialWidth, double initialHeight)
|
||||||
|
{
|
||||||
|
this.element = element;
|
||||||
|
this.initialWidth = initialWidth;
|
||||||
|
this.initialHeight = initialHeight;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Keep a list of all the elements for which we've transmitted the initial state
|
||||||
|
//
|
||||||
|
createdIds = new HashSet<string> {
|
||||||
|
"window",
|
||||||
|
"document",
|
||||||
|
"document.body",
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
void QueueStateMessagesLocked (EventTarget target)
|
||||||
|
{
|
||||||
|
if (target == null) return;
|
||||||
|
var created = false;
|
||||||
|
foreach (var m in target.StateMessages) {
|
||||||
|
if (m.MessageType == MessageType.Create) {
|
||||||
|
createdIds.Add (m.TargetId);
|
||||||
|
created = true;
|
||||||
|
}
|
||||||
|
if (created) {
|
||||||
|
QueueMessageLocked (m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void QueueMessageLocked (Message message)
|
||||||
|
{
|
||||||
|
//
|
||||||
|
// Make sure all the referenced objects have been created
|
||||||
|
//
|
||||||
|
if (!createdIds.Contains (message.TargetId)) {
|
||||||
|
QueueStateMessagesLocked (element.GetElementById (message.TargetId));
|
||||||
|
}
|
||||||
|
if (message.Value is EventTarget ve) {
|
||||||
|
if (!createdIds.Contains (ve.Id)) {
|
||||||
|
QueueStateMessagesLocked (ve);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (message.Value is Array a) {
|
||||||
|
for (var i = 0; i < a.Length; i++) {
|
||||||
|
// Console.WriteLine ($"A{i} = {a.GetValue(i)}");
|
||||||
|
if (a.GetValue (i) is EventTarget e && !createdIds.Contains (e.Id)) {
|
||||||
|
QueueStateMessagesLocked (e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Add it to the queue
|
||||||
|
//
|
||||||
|
//Console.WriteLine ($"QM {message.MessageType} {message.TargetId} {message.Key} {message.Value}");
|
||||||
|
queuedMessages.Add (message);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected virtual void QueueMessage (Message message)
|
||||||
|
{
|
||||||
|
lock (queuedMessages) {
|
||||||
|
QueueMessageLocked (message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
protected void Error (string message, Exception ex)
|
||||||
|
{
|
||||||
|
#if PCL
|
||||||
|
System.Diagnostics.Debug.WriteLine (string.Format ("{0}: {1}", message, ex));
|
||||||
|
#else
|
||||||
|
Console.ForegroundColor = ConsoleColor.Red;
|
||||||
|
Console.WriteLine ("{0}: {1}", message, ex);
|
||||||
|
Console.ResetColor ();
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
protected void Info (string message)
|
||||||
|
{
|
||||||
|
#if PCL
|
||||||
|
System.Diagnostics.Debug.WriteLine (message);
|
||||||
|
#else
|
||||||
|
Console.WriteLine (message);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
261
Ooui/UI.cs
261
Ooui/UI.cs
|
@ -7,16 +7,13 @@ using System.Threading;
|
||||||
using System.Threading.Tasks;
|
using System.Threading.Tasks;
|
||||||
using System.Net;
|
using System.Net;
|
||||||
|
|
||||||
#if !PCL
|
|
||||||
using System.Net.WebSockets;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
namespace Ooui
|
namespace Ooui
|
||||||
{
|
{
|
||||||
public static class UI
|
public static class UI
|
||||||
{
|
{
|
||||||
#if !PCL
|
public const int MaxFps = 30;
|
||||||
|
|
||||||
|
#if !PCL
|
||||||
static readonly ManualResetEvent started = new ManualResetEvent (false);
|
static readonly ManualResetEvent started = new ManualResetEvent (false);
|
||||||
|
|
||||||
[ThreadStatic]
|
[ThreadStatic]
|
||||||
|
@ -538,8 +535,8 @@ namespace Ooui
|
||||||
//
|
//
|
||||||
// Connect the web socket
|
// Connect the web socket
|
||||||
//
|
//
|
||||||
WebSocketContext webSocketContext = null;
|
System.Net.WebSockets.WebSocketContext webSocketContext = null;
|
||||||
WebSocket webSocket = null;
|
System.Net.WebSockets.WebSocket webSocket = null;
|
||||||
try {
|
try {
|
||||||
webSocketContext = await listenerContext.AcceptWebSocketAsync (subProtocol: "ooui").ConfigureAwait (false);
|
webSocketContext = await listenerContext.AcceptWebSocketAsync (subProtocol: "ooui").ConfigureAwait (false);
|
||||||
webSocket = webSocketContext.WebSocket;
|
webSocket = webSocketContext.WebSocket;
|
||||||
|
@ -577,10 +574,10 @@ namespace Ooui
|
||||||
// Create a new session and let it handle everything from here
|
// Create a new session and let it handle everything from here
|
||||||
//
|
//
|
||||||
try {
|
try {
|
||||||
var session = new Session (webSocket, element, w, h, serverToken);
|
var session = new WebSocketSession (webSocket, element, w, h, serverToken);
|
||||||
await session.RunAsync ().ConfigureAwait (false);
|
await session.RunAsync ().ConfigureAwait (false);
|
||||||
}
|
}
|
||||||
catch (WebSocketException ex) when (ex.WebSocketErrorCode == WebSocketError.ConnectionClosedPrematurely) {
|
catch (System.Net.WebSockets.WebSocketException ex) when (ex.WebSocketErrorCode == System.Net.WebSockets.WebSocketError.ConnectionClosedPrematurely) {
|
||||||
// The remote party closed the WebSocket connection without completing the close handshake.
|
// The remote party closed the WebSocket connection without completing the close handshake.
|
||||||
}
|
}
|
||||||
catch (Exception ex) {
|
catch (Exception ex) {
|
||||||
|
@ -598,218 +595,44 @@ namespace Ooui
|
||||||
Console.ResetColor ();
|
Console.ResetColor ();
|
||||||
}
|
}
|
||||||
|
|
||||||
public class Session
|
|
||||||
{
|
|
||||||
readonly WebSocket webSocket;
|
|
||||||
readonly Element element;
|
|
||||||
readonly Action<Message> handleElementMessageSent;
|
|
||||||
|
|
||||||
readonly CancellationTokenSource sessionCts = new CancellationTokenSource ();
|
|
||||||
readonly CancellationTokenSource linkedCts;
|
|
||||||
readonly CancellationToken token;
|
|
||||||
|
|
||||||
readonly HashSet<string> createdIds;
|
|
||||||
readonly List<Message> queuedMessages = new List<Message> ();
|
|
||||||
|
|
||||||
public const int MaxFps = 30;
|
|
||||||
|
|
||||||
readonly System.Timers.Timer sendThrottle;
|
|
||||||
DateTime lastTransmitTime = DateTime.MinValue;
|
|
||||||
readonly TimeSpan throttleInterval = TimeSpan.FromSeconds (1.0 / MaxFps);
|
|
||||||
readonly double initialWidth;
|
|
||||||
readonly double initialHeight;
|
|
||||||
|
|
||||||
public Session (WebSocket webSocket, Element element, double initialWidth, double initialHeight, CancellationToken serverToken)
|
|
||||||
{
|
|
||||||
this.webSocket = webSocket;
|
|
||||||
this.element = element;
|
|
||||||
this.initialWidth = initialWidth;
|
|
||||||
this.initialHeight = initialHeight;
|
|
||||||
|
|
||||||
//
|
|
||||||
// Create a new session cancellation token that will trigger
|
|
||||||
// automatically if the server shutsdown or the session shutsdown.
|
|
||||||
//
|
|
||||||
linkedCts = CancellationTokenSource.CreateLinkedTokenSource (serverToken, sessionCts.Token);
|
|
||||||
token = linkedCts.Token;
|
|
||||||
|
|
||||||
//
|
|
||||||
// Keep a list of all the elements for which we've transmitted the initial state
|
|
||||||
//
|
|
||||||
createdIds = new HashSet<string> {
|
|
||||||
"window",
|
|
||||||
"document",
|
|
||||||
"document.body",
|
|
||||||
};
|
|
||||||
|
|
||||||
//
|
|
||||||
// Preparse handlers for the element
|
|
||||||
//
|
|
||||||
handleElementMessageSent = QueueMessage;
|
|
||||||
|
|
||||||
//
|
|
||||||
// Create a timer to use as a throttle when sending messages
|
|
||||||
//
|
|
||||||
sendThrottle = new System.Timers.Timer (throttleInterval.TotalMilliseconds);
|
|
||||||
sendThrottle.Elapsed += (s, e) => {
|
|
||||||
// System.Console.WriteLine ("TICK SEND THROTTLE FOR {0}", element);
|
|
||||||
if ((e.SignalTime - lastTransmitTime) >= throttleInterval) {
|
|
||||||
sendThrottle.Enabled = false;
|
|
||||||
lastTransmitTime = e.SignalTime;
|
|
||||||
TransmitQueuedMessages ();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
public async Task RunAsync ()
|
|
||||||
{
|
|
||||||
//
|
|
||||||
// Start watching for changes in the element
|
|
||||||
//
|
|
||||||
element.MessageSent += handleElementMessageSent;
|
|
||||||
|
|
||||||
try {
|
|
||||||
//
|
|
||||||
// Add it to the document body
|
|
||||||
//
|
|
||||||
if (element.WantsFullScreen) {
|
|
||||||
element.Style.Width = initialWidth;
|
|
||||||
element.Style.Height = initialHeight;
|
|
||||||
}
|
|
||||||
QueueMessage (Message.Call ("document.body", "appendChild", element));
|
|
||||||
|
|
||||||
//
|
|
||||||
// Start the Read Loop
|
|
||||||
//
|
|
||||||
var receiveBuffer = new byte[64*1024];
|
|
||||||
|
|
||||||
while (webSocket.State == WebSocketState.Open && !token.IsCancellationRequested) {
|
|
||||||
var receiveResult = await webSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), token).ConfigureAwait (false);
|
|
||||||
|
|
||||||
if (receiveResult.MessageType == WebSocketMessageType.Close) {
|
|
||||||
await webSocket.CloseAsync (WebSocketCloseStatus.NormalClosure, "", token).ConfigureAwait (false);
|
|
||||||
sessionCts.Cancel ();
|
|
||||||
}
|
|
||||||
else if (receiveResult.MessageType == WebSocketMessageType.Binary) {
|
|
||||||
await webSocket.CloseAsync (WebSocketCloseStatus.InvalidMessageType, "Cannot accept binary frame", token).ConfigureAwait (false);
|
|
||||||
sessionCts.Cancel ();
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
var size = receiveResult.Count;
|
|
||||||
while (!receiveResult.EndOfMessage) {
|
|
||||||
if (size >= receiveBuffer.Length) {
|
|
||||||
await webSocket.CloseAsync (WebSocketCloseStatus.MessageTooBig, "Message too big", token).ConfigureAwait (false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
receiveResult = await webSocket.ReceiveAsync (new ArraySegment<byte>(receiveBuffer, size, receiveBuffer.Length - size), token).ConfigureAwait (false);
|
|
||||||
size += receiveResult.Count;
|
|
||||||
}
|
|
||||||
var receivedString = Encoding.UTF8.GetString (receiveBuffer, 0, size);
|
|
||||||
|
|
||||||
try {
|
|
||||||
// Console.WriteLine ("RECEIVED: {0}", receivedString);
|
|
||||||
var message = Newtonsoft.Json.JsonConvert.DeserializeObject<Message> (receivedString);
|
|
||||||
element.Receive (message);
|
|
||||||
}
|
|
||||||
catch (Exception ex) {
|
|
||||||
Error ("Failed to process received message", ex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
finally {
|
|
||||||
element.MessageSent -= handleElementMessageSent;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void QueueStateMessagesLocked (EventTarget target)
|
|
||||||
{
|
|
||||||
if (target == null) return;
|
|
||||||
var created = false;
|
|
||||||
foreach (var m in target.StateMessages) {
|
|
||||||
if (m.MessageType == MessageType.Create) {
|
|
||||||
createdIds.Add (m.TargetId);
|
|
||||||
created = true;
|
|
||||||
}
|
|
||||||
if (created) {
|
|
||||||
QueueMessageLocked (m);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void QueueMessageLocked (Message message)
|
|
||||||
{
|
|
||||||
//
|
|
||||||
// Make sure all the referenced objects have been created
|
|
||||||
//
|
|
||||||
if (!createdIds.Contains (message.TargetId)) {
|
|
||||||
QueueStateMessagesLocked (element.GetElementById (message.TargetId));
|
|
||||||
}
|
|
||||||
if (message.Value is EventTarget ve) {
|
|
||||||
if (!createdIds.Contains (ve.Id)) {
|
|
||||||
QueueStateMessagesLocked (ve);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else if (message.Value is Array a) {
|
|
||||||
for (var i = 0; i < a.Length; i++) {
|
|
||||||
// Console.WriteLine ($"A{i} = {a.GetValue(i)}");
|
|
||||||
if (a.GetValue (i) is EventTarget e && !createdIds.Contains (e.Id)) {
|
|
||||||
QueueStateMessagesLocked (e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//
|
|
||||||
// Add it to the queue
|
|
||||||
//
|
|
||||||
//Console.WriteLine ($"QM {message.MessageType} {message.TargetId} {message.Key} {message.Value}");
|
|
||||||
queuedMessages.Add (message);
|
|
||||||
}
|
|
||||||
|
|
||||||
void QueueMessage (Message message)
|
|
||||||
{
|
|
||||||
lock (queuedMessages) {
|
|
||||||
QueueMessageLocked (message);
|
|
||||||
}
|
|
||||||
sendThrottle.Enabled = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
async void TransmitQueuedMessages ()
|
|
||||||
{
|
|
||||||
try {
|
|
||||||
//
|
|
||||||
// Dequeue as many messages as we can
|
|
||||||
//
|
|
||||||
var messagesToSend = new List<Message> ();
|
|
||||||
System.Runtime.CompilerServices.ConfiguredTaskAwaitable task;
|
|
||||||
lock (queuedMessages) {
|
|
||||||
messagesToSend.AddRange (queuedMessages);
|
|
||||||
queuedMessages.Clear ();
|
|
||||||
|
|
||||||
if (messagesToSend.Count == 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
//
|
|
||||||
// Now actually send this message
|
|
||||||
// Do this while locked to make sure SendAsync is called in the right order
|
|
||||||
//
|
|
||||||
var json = Newtonsoft.Json.JsonConvert.SerializeObject (messagesToSend);
|
|
||||||
var outputBuffer = new ArraySegment<byte> (Encoding.UTF8.GetBytes (json));
|
|
||||||
//Console.WriteLine ("TRANSMIT " + json);
|
|
||||||
task = webSocket.SendAsync (outputBuffer, WebSocketMessageType.Text, true, token).ConfigureAwait (false);
|
|
||||||
}
|
|
||||||
await task;
|
|
||||||
}
|
|
||||||
catch (Exception ex) {
|
|
||||||
Error ("Failed to send queued messages, aborting session", ex);
|
|
||||||
element.MessageSent -= handleElementMessageSent;
|
|
||||||
sessionCts.Cancel ();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
static readonly Dictionary<string, Element> globalElements = new Dictionary<string, Element> ();
|
||||||
|
static readonly Dictionary<string, WebAssemblySession> globalElementSessions = new Dictionary<string, WebAssemblySession> ();
|
||||||
|
|
||||||
|
public static void SetGlobalElement (string globalElementId, Element element)
|
||||||
|
{
|
||||||
|
lock (globalElements) {
|
||||||
|
globalElements[globalElementId] = element;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void StartWebAssemblySession (string sessionId, string globalElementId)
|
||||||
|
{
|
||||||
|
Element element;
|
||||||
|
lock (globalElements) {
|
||||||
|
if (!globalElements.TryGetValue (globalElementId, out element))
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
var g = new WebAssemblySession (sessionId, element, 640, 480);
|
||||||
|
lock (globalElementSessions) {
|
||||||
|
globalElementSessions[sessionId] = g;
|
||||||
|
}
|
||||||
|
g.StartSession ();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void ReceiveWebAssemblySessionMessageJson (string sessionId, string json)
|
||||||
|
{
|
||||||
|
WebAssemblySession g;
|
||||||
|
lock (globalElementSessions) {
|
||||||
|
if (!globalElementSessions.TryGetValue (sessionId, out g))
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
g.ReceiveMessageJson (json);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
static readonly Dictionary<string, Style> styles =
|
static readonly Dictionary<string, Style> styles =
|
||||||
new Dictionary<string, Style> ();
|
new Dictionary<string, Style> ();
|
||||||
static readonly StyleSelectors rules = new StyleSelectors ();
|
static readonly StyleSelectors rules = new StyleSelectors ();
|
||||||
|
|
|
@ -0,0 +1,84 @@
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
|
||||||
|
namespace Ooui
|
||||||
|
{
|
||||||
|
public class WebAssemblySession : Session
|
||||||
|
{
|
||||||
|
readonly string id;
|
||||||
|
readonly Action<Message> handleElementMessageSent;
|
||||||
|
|
||||||
|
public WebAssemblySession (string id, Element element, double initialWidth, double initialHeight)
|
||||||
|
: base (element, initialWidth, initialHeight)
|
||||||
|
{
|
||||||
|
this.id = id;
|
||||||
|
handleElementMessageSent = QueueMessage;
|
||||||
|
}
|
||||||
|
|
||||||
|
void TransmitQueuedMessages ()
|
||||||
|
{
|
||||||
|
//
|
||||||
|
// Dequeue as many messages as we can
|
||||||
|
//
|
||||||
|
var messagesToSend = new List<Message> ();
|
||||||
|
lock (queuedMessages) {
|
||||||
|
messagesToSend.AddRange (queuedMessages);
|
||||||
|
queuedMessages.Clear ();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (messagesToSend.Count == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Now actually send the messages
|
||||||
|
//
|
||||||
|
var json = Newtonsoft.Json.JsonConvert.SerializeObject (messagesToSend);
|
||||||
|
SendMessagesJson (json);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected override void QueueMessage (Message message)
|
||||||
|
{
|
||||||
|
base.QueueMessage (message);
|
||||||
|
TransmitQueuedMessages ();
|
||||||
|
}
|
||||||
|
|
||||||
|
void SendMessagesJson (string json)
|
||||||
|
{
|
||||||
|
Info ("SEND: " + json);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void ReceiveMessageJson (string json)
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
Info ("RECEIVED: " + json);
|
||||||
|
var message = Newtonsoft.Json.JsonConvert.DeserializeObject<Message> (json);
|
||||||
|
element.Receive (message);
|
||||||
|
}
|
||||||
|
catch (Exception ex) {
|
||||||
|
Error ("Failed to process received message", ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void StartSession ()
|
||||||
|
{
|
||||||
|
//
|
||||||
|
// Start watching for changes in the element
|
||||||
|
//
|
||||||
|
element.MessageSent += handleElementMessageSent;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Add it to the document body
|
||||||
|
//
|
||||||
|
if (element.WantsFullScreen) {
|
||||||
|
element.Style.Width = initialWidth;
|
||||||
|
element.Style.Height = initialHeight;
|
||||||
|
}
|
||||||
|
QueueMessage (Message.Call ("document.body", "appendChild", element));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void StopSession ()
|
||||||
|
{
|
||||||
|
element.MessageSent -= handleElementMessageSent;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,205 @@
|
||||||
|
#if !PCL
|
||||||
|
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.IO;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Text;
|
||||||
|
using System.Threading;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using System.Net;
|
||||||
|
using System.Net.WebSockets;
|
||||||
|
|
||||||
|
namespace Ooui
|
||||||
|
{
|
||||||
|
public class WebSocketSession : Session
|
||||||
|
{
|
||||||
|
readonly WebSocket webSocket;
|
||||||
|
readonly Action<Message> handleElementMessageSent;
|
||||||
|
|
||||||
|
readonly CancellationTokenSource sessionCts = new CancellationTokenSource ();
|
||||||
|
readonly CancellationTokenSource linkedCts;
|
||||||
|
readonly CancellationToken token;
|
||||||
|
|
||||||
|
readonly System.Timers.Timer sendThrottle;
|
||||||
|
DateTime lastTransmitTime = DateTime.MinValue;
|
||||||
|
readonly TimeSpan throttleInterval = TimeSpan.FromSeconds (1.0 / UI.MaxFps);
|
||||||
|
|
||||||
|
public WebSocketSession (WebSocket webSocket, Element element, double initialWidth, double initialHeight, CancellationToken serverToken)
|
||||||
|
: base (element, initialWidth, initialHeight)
|
||||||
|
{
|
||||||
|
this.webSocket = webSocket;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Create a new session cancellation token that will trigger
|
||||||
|
// automatically if the server shutsdown or the session shutsdown.
|
||||||
|
//
|
||||||
|
linkedCts = CancellationTokenSource.CreateLinkedTokenSource (serverToken, sessionCts.Token);
|
||||||
|
token = linkedCts.Token;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Preparse handlers for the element
|
||||||
|
//
|
||||||
|
handleElementMessageSent = QueueMessage;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Create a timer to use as a throttle when sending messages
|
||||||
|
//
|
||||||
|
sendThrottle = new System.Timers.Timer (throttleInterval.TotalMilliseconds);
|
||||||
|
sendThrottle.Elapsed += (s, e) => {
|
||||||
|
// System.Console.WriteLine ("TICK SEND THROTTLE FOR {0}", element);
|
||||||
|
if ((e.SignalTime - lastTransmitTime) >= throttleInterval) {
|
||||||
|
sendThrottle.Enabled = false;
|
||||||
|
lastTransmitTime = e.SignalTime;
|
||||||
|
TransmitQueuedMessages ();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task RunAsync ()
|
||||||
|
{
|
||||||
|
//
|
||||||
|
// Start watching for changes in the element
|
||||||
|
//
|
||||||
|
element.MessageSent += handleElementMessageSent;
|
||||||
|
|
||||||
|
try {
|
||||||
|
//
|
||||||
|
// Add it to the document body
|
||||||
|
//
|
||||||
|
if (element.WantsFullScreen) {
|
||||||
|
element.Style.Width = initialWidth;
|
||||||
|
element.Style.Height = initialHeight;
|
||||||
|
}
|
||||||
|
QueueMessage (Message.Call ("document.body", "appendChild", element));
|
||||||
|
|
||||||
|
//
|
||||||
|
// Start the Read Loop
|
||||||
|
//
|
||||||
|
var receiveBuffer = new byte[64 * 1024];
|
||||||
|
|
||||||
|
while (webSocket.State == WebSocketState.Open && !token.IsCancellationRequested) {
|
||||||
|
var receiveResult = await webSocket.ReceiveAsync (new ArraySegment<byte> (receiveBuffer), token).ConfigureAwait (false);
|
||||||
|
|
||||||
|
if (receiveResult.MessageType == WebSocketMessageType.Close) {
|
||||||
|
await webSocket.CloseAsync (WebSocketCloseStatus.NormalClosure, "", token).ConfigureAwait (false);
|
||||||
|
sessionCts.Cancel ();
|
||||||
|
}
|
||||||
|
else if (receiveResult.MessageType == WebSocketMessageType.Binary) {
|
||||||
|
await webSocket.CloseAsync (WebSocketCloseStatus.InvalidMessageType, "Cannot accept binary frame", token).ConfigureAwait (false);
|
||||||
|
sessionCts.Cancel ();
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
var size = receiveResult.Count;
|
||||||
|
while (!receiveResult.EndOfMessage) {
|
||||||
|
if (size >= receiveBuffer.Length) {
|
||||||
|
await webSocket.CloseAsync (WebSocketCloseStatus.MessageTooBig, "Message too big", token).ConfigureAwait (false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
receiveResult = await webSocket.ReceiveAsync (new ArraySegment<byte> (receiveBuffer, size, receiveBuffer.Length - size), token).ConfigureAwait (false);
|
||||||
|
size += receiveResult.Count;
|
||||||
|
}
|
||||||
|
var receivedString = Encoding.UTF8.GetString (receiveBuffer, 0, size);
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Console.WriteLine ("RECEIVED: {0}", receivedString);
|
||||||
|
var message = Newtonsoft.Json.JsonConvert.DeserializeObject<Message> (receivedString);
|
||||||
|
element.Receive (message);
|
||||||
|
}
|
||||||
|
catch (Exception ex) {
|
||||||
|
Error ("Failed to process received message", ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
element.MessageSent -= handleElementMessageSent;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void QueueStateMessagesLocked (EventTarget target)
|
||||||
|
{
|
||||||
|
if (target == null) return;
|
||||||
|
var created = false;
|
||||||
|
foreach (var m in target.StateMessages) {
|
||||||
|
if (m.MessageType == MessageType.Create) {
|
||||||
|
createdIds.Add (m.TargetId);
|
||||||
|
created = true;
|
||||||
|
}
|
||||||
|
if (created) {
|
||||||
|
QueueMessageLocked (m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void QueueMessageLocked (Message message)
|
||||||
|
{
|
||||||
|
//
|
||||||
|
// Make sure all the referenced objects have been created
|
||||||
|
//
|
||||||
|
if (!createdIds.Contains (message.TargetId)) {
|
||||||
|
QueueStateMessagesLocked (element.GetElementById (message.TargetId));
|
||||||
|
}
|
||||||
|
if (message.Value is EventTarget ve) {
|
||||||
|
if (!createdIds.Contains (ve.Id)) {
|
||||||
|
QueueStateMessagesLocked (ve);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (message.Value is Array a) {
|
||||||
|
for (var i = 0; i < a.Length; i++) {
|
||||||
|
// Console.WriteLine ($"A{i} = {a.GetValue(i)}");
|
||||||
|
if (a.GetValue (i) is EventTarget e && !createdIds.Contains (e.Id)) {
|
||||||
|
QueueStateMessagesLocked (e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Add it to the queue
|
||||||
|
//
|
||||||
|
//Console.WriteLine ($"QM {message.MessageType} {message.TargetId} {message.Key} {message.Value}");
|
||||||
|
queuedMessages.Add (message);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected override void QueueMessage (Message message)
|
||||||
|
{
|
||||||
|
base.QueueMessage (message);
|
||||||
|
sendThrottle.Enabled = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
async void TransmitQueuedMessages ()
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
//
|
||||||
|
// Dequeue as many messages as we can
|
||||||
|
//
|
||||||
|
var messagesToSend = new List<Message> ();
|
||||||
|
System.Runtime.CompilerServices.ConfiguredTaskAwaitable task;
|
||||||
|
lock (queuedMessages) {
|
||||||
|
messagesToSend.AddRange (queuedMessages);
|
||||||
|
queuedMessages.Clear ();
|
||||||
|
|
||||||
|
if (messagesToSend.Count == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Now actually send this message
|
||||||
|
// Do this while locked to make sure SendAsync is called in the right order
|
||||||
|
//
|
||||||
|
var json = Newtonsoft.Json.JsonConvert.SerializeObject (messagesToSend);
|
||||||
|
var outputBuffer = new ArraySegment<byte> (Encoding.UTF8.GetBytes (json));
|
||||||
|
//Console.WriteLine ("TRANSMIT " + json);
|
||||||
|
task = webSocket.SendAsync (outputBuffer, WebSocketMessageType.Text, true, token).ConfigureAwait (false);
|
||||||
|
}
|
||||||
|
await task;
|
||||||
|
}
|
||||||
|
catch (Exception ex) {
|
||||||
|
Error ("Failed to send queued messages, aborting session", ex);
|
||||||
|
element.MessageSent -= handleElementMessageSent;
|
||||||
|
sessionCts.Cancel ();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
Loading…
Reference in New Issue