huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

SSE kind of works - duplicate messages

This commit is contained in:
moozzyk 2016-09-30 14:21:37 -07:00
parent 27ddb7de90
commit 32ed7ca0c4
6 changed files with 141 additions and 103 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/WebApplication95/ConnectionManager.cs
View File

@ -45,9 +45,10 @@ namespace WebApplication95
} }
} }
// TODO: don't leak HttpContext to ConnectionManager
public string GetConnectionId(HttpContext context) public string GetConnectionId(HttpContext context)
{ {
// REVIEW: Only check the query string for longpolling
var id = context.Request.Query["id"]; var id = context.Request.Query["id"];
if (!StringValues.IsNullOrEmpty(id)) if (!StringValues.IsNullOrEmpty(id))

146
src/WebApplication95/Dispatcher.cs
View File

@ -41,10 +41,85 @@ namespace WebApplication95
public async Task Execute<TEndPoint>(string path, HttpContext context) where TEndPoint : EndPoint public async Task Execute<TEndPoint>(string path, HttpContext context) where TEndPoint : EndPoint
{ {
if (context.Request.Path.StartsWithSegments(path + "/send")) if (context.Request.Path.StartsWithSegments(path + "/getid"))
{
await ProcessGetId(context);
}
else if (context.Request.Path.StartsWithSegments(path + "/send"))
{
await ProcessSend(context);
}
else
{
var endpoint = (EndPoint)context.RequestServices.GetRequiredService<TEndPoint>();
// Outgoing channels
if (context.Request.Path.StartsWithSegments(path + "/sse"))
{
var connectionState = GetOrCreateConnection(context);
var sse = new ServerSentEvents(connectionState);
var ignore = endpoint.OnConnected(connectionState.Connection);
connectionState.Connection.TransportType = TransportType.ServerSentEvents;
await sse.ProcessRequest(context);
connectionState.Connection.Complete();
_manager.RemoveConnection(connectionState.Connection.ConnectionId);
}
else if (context.Request.Path.StartsWithSegments(path + "/ws"))
{
var connectionState = GetOrCreateConnection(context);
var ws = new WebSockets(connectionState);
var ignore = endpoint.OnConnected(connectionState.Connection);
connectionState.Connection.TransportType = TransportType.WebSockets;
await ws.ProcessRequest(context);
connectionState.Connection.Complete();
_manager.RemoveConnection(connectionState.Connection.ConnectionId);
}
else if (context.Request.Path.StartsWithSegments(path + "/poll"))
{ {
var connectionId = context.Request.Query["id"]; var connectionId = context.Request.Query["id"];
ConnectionState connectionState;
bool newConnection = false;
if (_manager.AddConnection(connectionId, out connectionState))
{
newConnection = true;
var ignore = endpoint.OnConnected(connectionState.Connection);
connectionState.Connection.TransportType = TransportType.LongPolling;
}
var longPolling = new LongPolling(connectionState);
await longPolling.ProcessRequest(newConnection, context);
_manager.MarkConnectionDead(connectionState.Connection.ConnectionId);
}
}
}
private async Task ProcessGetId(HttpContext context)
{
var connectionId = _manager.GetConnectionId(context);
ConnectionState state;
_manager.AddConnection(connectionId, out state);
context.Response.Headers["X-SignalR-ConnectionId"] = connectionId;
await context.Response.WriteAsync($"{{ \"connectionId\": \"{connectionId}\" }}");
return;
}
private async Task ProcessSend(HttpContext context)
{
var connectionId = context.Request.Query["id"];
if (StringValues.IsNullOrEmpty(connectionId)) if (StringValues.IsNullOrEmpty(connectionId))
{ {
throw new InvalidOperationException("Missing connection id"); throw new InvalidOperationException("Missing connection id");
@ -57,67 +132,26 @@ namespace WebApplication95
await context.Request.Body.CopyToAsync(state.Connection.Input); await context.Request.Body.CopyToAsync(state.Connection.Input);
} }
} }
private ConnectionState GetOrCreateConnection(HttpContext context)
{
var connectionId = context.Request.Query["id"];
ConnectionState connectionState;
if (StringValues.IsNullOrEmpty(connectionId))
{
connectionId = _manager.GetConnectionId(context);
_manager.AddConnection(connectionId, out connectionState);
}
else else
{ {
var endpoint = (EndPoint)context.RequestServices.GetRequiredService<TEndPoint>(); if (!_manager.TryGetConnection(connectionId, out connectionState))
var connectionId = _manager.GetConnectionId(context);
// Outgoing channels
if (context.Request.Path.StartsWithSegments(path + "/sse"))
{ {
ConnectionState state; throw new InvalidOperationException("Unknown connection id");
_manager.AddConnection(connectionId, out state);
var sse = new ServerSentEvents(state);
var ignore = endpoint.OnConnected(state.Connection);
state.Connection.TransportType = TransportType.ServerSentEvents;
await sse.ProcessRequest(context);
state.Connection.Complete();
_manager.RemoveConnection(connectionId);
} }
else if (context.Request.Path.StartsWithSegments(path + "/ws"))
{
ConnectionState state;
_manager.AddConnection(connectionId, out state);
var ws = new WebSockets(state);
var ignore = endpoint.OnConnected(state.Connection);
state.Connection.TransportType = TransportType.WebSockets;
await ws.ProcessRequest(context);
state.Connection.Complete();
_manager.RemoveConnection(connectionId);
}
else if (context.Request.Path.StartsWithSegments(path + "/poll"))
{
ConnectionState state;
bool newConnection = false;
if (_manager.AddConnection(connectionId, out state))
{
newConnection = true;
var ignore = endpoint.OnConnected(state.Connection);
state.Connection.TransportType = TransportType.LongPolling;
}
var longPolling = new LongPolling(state);
await longPolling.ProcessRequest(newConnection, context);
_manager.MarkConnectionDead(connectionId);
} }
} return connectionState;
} }
} }
} }

30
src/WebApplication95/LongPolling.cs
View File

@ -104,33 +104,9 @@ namespace WebApplication95
await Post(async state => await Post(async state =>
{ {
var data = ((ArraySegment<byte>)state); var data = ((ArraySegment<byte>)state);
// + 100 = laziness _context.Response.Headers["X-SignalR-ConnectionId"] = _state.Connection.ConnectionId;
var buffer = new byte[data.Count + _state.Connection.ConnectionId.Length + 100]; _context.Response.ContentLength = data.Count;
var at = 0; await _context.Response.Body.WriteAsync(data.Array, 0, data.Count);
buffer[at++] = (byte)'{';
buffer[at++] = (byte)'"';
buffer[at++] = (byte)'c';
buffer[at++] = (byte)'"';
buffer[at++] = (byte)':';
buffer[at++] = (byte)'"';
int count = Encoding.UTF8.GetBytes(_state.Connection.ConnectionId, 0, _state.Connection.ConnectionId.Length, buffer, at);
at += count;
buffer[at++] = (byte)'"';
if (data.Array != null)
{
buffer[at++] = (byte)',';
buffer[at++] = (byte)'"';
buffer[at++] = (byte)'d';
buffer[at++] = (byte)'"';
buffer[at++] = (byte)':';
//buffer[at++] = (byte)'"';
Buffer.BlockCopy(data.Array, data.Offset, buffer, at, data.Count);
}
at += data.Count;
//buffer[at++] = (byte)'"';
buffer[at++] = (byte)'}';
_context.Response.ContentLength = at;
await _context.Response.Body.WriteAsync(buffer, 0, at);
}, },
value); value);

4
src/WebApplication95/ServerSentEvents.cs
View File

@ -40,14 +40,12 @@ namespace WebApplication95
public async Task ProcessRequest(HttpContext context) public async Task ProcessRequest(HttpContext context)
{ {
context.Response.ContentType = "text/event-stream"; context.Response.ContentType = "text/event-stream";
context.Response.Headers["Cache-Control"] = "no-cache";
// End the connection if the client goes away // End the connection if the client goes away
context.RequestAborted.Register(state => OnConnectionAborted(state), this); context.RequestAborted.Register(state => OnConnectionAborted(state), this);
_context = context; _context = context;
await _context.Response.WriteAsync($"data: {_state.Connection.ConnectionId}\n\n");
// Set the initial TCS when everything is setup // Set the initial TCS when everything is setup
_initTcs.TrySetResult(null); _initTcs.TrySetResult(null);

1
src/WebApplication95/WebSockets.cs
View File

@ -78,6 +78,7 @@ namespace WebApplication95
else if (result.MessageType == WebSocketMessageType.Close) else if (result.MessageType == WebSocketMessageType.Close)
{ {
await ws.CloseAsync(WebSocketCloseStatus.NormalClosure, "", CancellationToken.None); await ws.CloseAsync(WebSocketCloseStatus.NormalClosure, "", CancellationToken.None);
// TODO: needs to remove itself from connection mamanger?
break; break;
} }
} }

54
src/WebApplication95/wwwroot/index.html
View File

@ -4,12 +4,8 @@
<meta charset="utf-8" /> <meta charset="utf-8" />
<title></title> <title></title>
<script> <script>
document.addEventListener('DOMContentLoaded', () => {
var connectionId; function send(connectionId, data) {
function send() {
var body = document.getElementById('data').value;
var xhr = new XMLHttpRequest(); var xhr = new XMLHttpRequest();
var url = '/chat/send?id=' + connectionId; var url = '/chat/send?id=' + connectionId;
xhr.open("POST", url, true); xhr.open("POST", url, true);
@ -18,11 +14,39 @@
if (xhr.readyState == 4 && xhr.status == 200) { if (xhr.readyState == 4 && xhr.status == 200) {
} }
} }
var data = JSON.stringify(body); var data = JSON.stringify(data);
xhr.send(data); xhr.send(data);
} }
var source = new EventSource('/chat/sse'); function xhr(method, url) {
return new Promise((resolve, reject) => {
let xhr = new XMLHttpRequest();
xhr.open(method, url);
xhr.send();
xhr.onload = () => {
if (xhr.status >= 200 && xhr.status < 300) {
resolve(xhr.response);
} else {
reject({
status: xhr.status,
statusText: xhr.statusText
});
}
};
xhr.onerror = () => {
reject({
status: xhr.status,
statusText: xhr.statusText
});
};
});
}
xhr('GET', '/chat/getid').then(getidPayload => {
let connectionId = JSON.parse(getidPayload).connectionId;
let source = new EventSource(`/chat/sse?id=${connectionId}`);
source.onopen = function () { source.onopen = function () {
console.log('Opened!'); console.log('Opened!');
@ -33,20 +57,24 @@
}; };
source.onmessage = function (data) { source.onmessage = function (data) {
if (!connectionId) {
connectionId = data.data;
return;
}
var child = document.createElement('li'); var child = document.createElement('li');
child.innerText = data.data; child.innerText = data.data;
document.getElementById('messages').appendChild(child); document.getElementById('messages').appendChild(child);
}; };
document.getElementById('sendmessage').addEventListener('click', () => {
let data = document.getElementById('data').value;
send(connectionId, data);
});
});
});
</script> </script>
</head> </head>
<body> <body>
<h1>Server Sent Events</h1> <h1>Server Sent Events</h1>
<input type="text" id="data" /> <input type="text" id="data" />
<input type="button" value="Send" onclick="send()" /> <input type="button" id="sendmessage" value="Send" />
<ul id="messages"> <ul id="messages">