vinzenz/servicepoint-tanks
0
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix UnreachableException

Browse source
This commit is contained in:
Vinzenz Schroeter 2024-05-08 00:29:51 +02:00
parent 827b3a9330
commit c7bfceffe1
1 changed files with 6 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
tanks-backend/TanksServer/Interactivity/DroppablePackageRequestConnection.cs
View file

@ -10,27 +10,15 @@ internal abstract class DroppablePackageRequestConnection<TPackage>(
where TPackage : class, IDisposable where TPackage : class, IDisposable
{ {
private readonly AsyncAutoResetEvent _nextPackageEvent = new(false, 1); private readonly AsyncAutoResetEvent _nextPackageEvent = new(false, 1);
private int _runningMessageHandlers = 0;
private TPackage? _next; private TPackage? _next;
protected override ValueTask HandleMessageAsync(Memory<byte> _) protected override async ValueTask HandleMessageAsync(Memory<byte> _)
{ {
if (Interlocked.Increment(ref _runningMessageHandlers) == 1) await _nextPackageEvent.WaitAsync();
return Core(); var package = Interlocked.Exchange(ref _next, null);
if (package == null)
// client has requested multiple frames, ignoring duplicate requests return;
Interlocked.Decrement(ref _runningMessageHandlers); await SendPackageAsync(package);
return ValueTask.CompletedTask;
async ValueTask Core()
{
await _nextPackageEvent.WaitAsync();
var package = Interlocked.Exchange(ref _next, null);
if (package == null)
throw new UnreachableException("package should be set here");
await SendPackageAsync(package);
Interlocked.Decrement(ref _runningMessageHandlers);
}
} }
protected void SetNextPackage(TPackage next) protected void SetNextPackage(TPackage next)