t0w0bi/ClubPenguinServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

release connected async akways

Browse Source
This commit is contained in:
Stanley Dimant
2022-09-23 12:45:24 +02:00
parent 588eb9f4ed
commit 8236d355a2
1 changed files with 22 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
MareSynchronosServer/MareSynchronosServer/Hubs/SignalRLimitFilter.cs
View File

@@ -56,28 +56,29 @@ public class SignalRLimitFilter : IHubFilter
public async Task OnConnectedAsync(HubLifetimeContext context, Func<HubLifetimeContext, Task> next) public async Task OnConnectedAsync(HubLifetimeContext context, Func<HubLifetimeContext, Task> next)
{ {
await ConnectionLimiterSemaphore.WaitAsync().ConfigureAwait(false); await ConnectionLimiterSemaphore.WaitAsync().ConfigureAwait(false);
var ip = accessor.GetIpAddress();
var client = new ClientRequestIdentity
{
ClientIp = ip,
Path = "Connect",
HttpVerb = "ws",
};
foreach (var rule in await _processor.GetMatchingRulesAsync(client).ConfigureAwait(false))
{
var counter = await _processor.ProcessRequestAsync(client, rule).ConfigureAwait(false);
if (counter.Count > rule.Limit)
{
var retry = counter.Timestamp.RetryAfterFrom(rule);
logger.LogWarning("Connection rate limit triggered from {ip}", ip);
ConnectionLimiterSemaphore.Release();
throw new HubException($"Connection rate limit {retry}");
}
}
try try
{ {
await Task.Delay(100).ConfigureAwait(false); var ip = accessor.GetIpAddress();
var client = new ClientRequestIdentity
{
ClientIp = ip,
Path = "Connect",
HttpVerb = "ws",
};
foreach (var rule in await _processor.GetMatchingRulesAsync(client).ConfigureAwait(false))
{
var counter = await _processor.ProcessRequestAsync(client, rule).ConfigureAwait(false);
if (counter.Count > rule.Limit)
{
var retry = counter.Timestamp.RetryAfterFrom(rule);
logger.LogWarning("Connection rate limit triggered from {ip}", ip);
ConnectionLimiterSemaphore.Release();
throw new HubException($"Connection rate limit {retry}");
}
}
await Task.Delay(25).ConfigureAwait(false);
await next(context).ConfigureAwait(false); await next(context).ConfigureAwait(false);
} }
catch (Exception ex) catch (Exception ex)
@@ -102,7 +103,7 @@ public class SignalRLimitFilter : IHubFilter
try try
{ {
await next(context, exception).ConfigureAwait(false); await next(context, exception).ConfigureAwait(false);
await Task.Delay(250).ConfigureAwait(false); await Task.Delay(25).ConfigureAwait(false);
} }
catch (Exception e) catch (Exception e)
{ {