Add publishing to the server

This commit is contained in:
Frank A. Krueger 2017-06-12 16:24:40 -07:00
parent daeb375c16
commit 317fc97652
2 changed files with 60 additions and 9 deletions

View File

@ -1,4 +1,6 @@
using System; using System;
using System.Collections.Generic;
using System.IO;
using System.Text; using System.Text;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
@ -9,6 +11,9 @@ namespace Ooui
{ {
public class Server public class Server
{ {
readonly Dictionary<string, Func<Element>> publishedPaths =
new Dictionary<string, Func<Element>> ();
public Task RunAsync (string listenerPrefix) public Task RunAsync (string listenerPrefix)
{ {
return RunAsync (listenerPrefix, CancellationToken.None); return RunAsync (listenerPrefix, CancellationToken.None);
@ -22,22 +27,66 @@ namespace Ooui
Console.WriteLine ($"Listening at {listenerPrefix}..."); Console.WriteLine ($"Listening at {listenerPrefix}...");
while (!token.IsCancellationRequested) { while (!token.IsCancellationRequested) {
var listenerContext = await listener.GetContextAsync (); var listenerContext = await listener.GetContextAsync ().ConfigureAwait (false);
if (listenerContext.Request.IsWebSocketRequest) { if (listenerContext.Request.IsWebSocketRequest) {
ProcessWebSocketRequest (listenerContext, token); ProcessWebSocketRequest (listenerContext, token);
} }
else { else {
listenerContext.Response.StatusCode = 400; ProcessRequest (listenerContext, token);
}
}
}
public void Publish (string path, Func<Element> elementCtor)
{
System.Console.WriteLine($"PUBLISH {path}");
publishedPaths[path] = elementCtor;
}
public void Publish (string path, Element element)
{
Publish (path, () => element);
}
void ProcessRequest (HttpListenerContext listenerContext, CancellationToken token)
{
var url = listenerContext.Request.Url;
var path = url.LocalPath;
Console.WriteLine ($"{listenerContext.Request.HttpMethod} {url.LocalPath}");
Func<Element> ctor;
if (publishedPaths.TryGetValue (path, out ctor)) {
var element = ctor ();
RegisterElement (element);
WriteElementHtml (element, listenerContext.Response);
}
else {
listenerContext.Response.StatusCode = 404;
listenerContext.Response.Close (); listenerContext.Response.Close ();
} }
} }
void RegisterElement (Element element)
{
}
void WriteElementHtml (Element element, HttpListenerResponse response)
{
response.StatusCode = 200;
using (var s = response.OutputStream) {
using (var w = new StreamWriter (s, Encoding.UTF8)) {
w.WriteLine ($"Hello {element}");
}
}
response.Close ();
} }
async void ProcessWebSocketRequest (HttpListenerContext listenerContext, CancellationToken token) async void ProcessWebSocketRequest (HttpListenerContext listenerContext, CancellationToken token)
{ {
WebSocketContext webSocketContext = null; WebSocketContext webSocketContext = null;
try { try {
webSocketContext = await listenerContext.AcceptWebSocketAsync(subProtocol: null); webSocketContext = await listenerContext.AcceptWebSocketAsync(subProtocol: null).ConfigureAwait (false);
Console.WriteLine ("Accepted WebSocket: {0}", webSocketContext); Console.WriteLine ("Accepted WebSocket: {0}", webSocketContext);
} }
catch (Exception e) { catch (Exception e) {
@ -58,26 +107,26 @@ namespace Ooui
var receiveResult = await webSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), token); var receiveResult = await webSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), token);
if (receiveResult.MessageType == WebSocketMessageType.Close) { if (receiveResult.MessageType == WebSocketMessageType.Close) {
await webSocket.CloseAsync (WebSocketCloseStatus.NormalClosure, "", token); await webSocket.CloseAsync (WebSocketCloseStatus.NormalClosure, "", token).ConfigureAwait (false);
} }
else if (receiveResult.MessageType == WebSocketMessageType.Binary) { else if (receiveResult.MessageType == WebSocketMessageType.Binary) {
await webSocket.CloseAsync (WebSocketCloseStatus.InvalidMessageType, "Cannot accept binary frame", token); await webSocket.CloseAsync (WebSocketCloseStatus.InvalidMessageType, "Cannot accept binary frame", token).ConfigureAwait (false);
} }
else { else {
var size = receiveResult.Count; var size = receiveResult.Count;
while (!receiveResult.EndOfMessage) { while (!receiveResult.EndOfMessage) {
if (size >= receiveBuffer.Length) { if (size >= receiveBuffer.Length) {
await webSocket.CloseAsync (WebSocketCloseStatus.MessageTooBig, "Message too big", token); await webSocket.CloseAsync (WebSocketCloseStatus.MessageTooBig, "Message too big", token).ConfigureAwait (false);
return; return;
} }
receiveResult = await webSocket.ReceiveAsync (new ArraySegment<byte>(receiveBuffer, size, receiveBuffer.Length - size), token); receiveResult = await webSocket.ReceiveAsync (new ArraySegment<byte>(receiveBuffer, size, receiveBuffer.Length - size), token).ConfigureAwait (false);
size += receiveResult.Count; size += receiveResult.Count;
} }
var receivedString = Encoding.UTF8.GetString (receiveBuffer, 0, size); var receivedString = Encoding.UTF8.GetString (receiveBuffer, 0, size);
Console.WriteLine ("RECEIVED: {0}", receivedString); Console.WriteLine ("RECEIVED: {0}", receivedString);
var outputBuffer = new ArraySegment<byte> (Encoding.UTF8.GetBytes ($"You said: {receivedString}")); var outputBuffer = new ArraySegment<byte> (Encoding.UTF8.GetBytes ($"You said: {receivedString}"));
await webSocket.SendAsync (outputBuffer, WebSocketMessageType.Text, true, token); await webSocket.SendAsync (outputBuffer, WebSocketMessageType.Text, true, token).ConfigureAwait (false);
} }
} }
} }

View File

@ -9,6 +9,8 @@ namespace Samples
{ {
Console.WriteLine ("Hello World!"); Console.WriteLine ("Hello World!");
var server = new Server (); var server = new Server ();
var button = new Button();
server.Publish ("/button", button);
server.RunAsync ("http://*:8080/").Wait (); server.RunAsync ("http://*:8080/").Wait ();
return 0; return 0;
} }