private async void CleanUpMothershipsLoop()
{
while (true)
{
// Every 5 minutes, clean up any stale motherships
// Note that mothership will pause for up to 1.5 minutes
await Task.Delay(TimeSpan.FromMinutes(5));
try
{
MothershipModel[] motherships;
lock (AllMothershipsModel.Motherships)
{
motherships = AllMothershipsModel.Motherships.ToArray();
}
foreach (var mothership in motherships)
{
// If haven't received a message in last 90 seconds
if (mothership.LastTimeMessageReceived < DateTime.UtcNow.AddSeconds(-90))
{
mothership.CloseAndRemove();
}
// If has been disconnected for more than 2 minutes
else if (mothership.DisconnectedTime != null && mothership.DisconnectedTime < DateTime.Now.AddMinutes(-2))
{
AllMothershipsModel.RemoveMothership(mothership);
}
}
}
catch { }
}
}
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
public void Configure(IApplicationBuilder app, IHostingEnvironment env)
{
if (env.IsDevelopment())
{
app.UseDeveloperExceptionPage();
}
app.UseMvc();
app.UseWebSockets(new WebSocketOptions()
{
ReceiveBufferSize = RECEIVE_BUFFER_SIZE
});
CleanUpMothershipsLoop();
app.Use(async(context, next) =>
{
if (context.Request.Path == "/wsMothership" || context.Request.Path.ToString().StartsWith("/wsClient/"))
{
if (!context.WebSockets.IsWebSocketRequest)
{
context.Response.StatusCode = 400;
return;
}
WebSocket webSocket = await context.WebSockets.AcceptWebSocketAsync();
if (context.Request.Path == "/wsMothership")
{
var mothership = AllMothershipsModel.CreateNewMothership(webSocket);
await mothership.RunReceiveLoopAsync();
}
else if (context.Request.Path.ToString().StartsWith("/wsMothership/"))
{
var mothership = AllMothershipsModel.ReconnectMothership(webSocket, context.Request.Path.ToString().Substring("/wsMothership/".Length));
await mothership.RunReceiveLoopAsync();
}
else
{
string path = context.Request.Path.ToString();
string mothershipName = path.Substring("/wsClient/".Length).TrimEnd('/');
var client = AllMothershipsModel.TryCreateClient(webSocket, mothershipName);
if (client == null)
{
context.Response.StatusCode = 400;
return;
}
await client.RunReceiveLoopAsync();
}
}
else
{
await next();
}
});
}
public IEnumerable <string> Get()
{
return(AllMothershipsModel.GetAllMothershipNames());
}
