Add guard to unregistering twice

LucHeart · LucHeart · commit ece1cc01c67b · 2025-01-31T20:19:15.000+01:00
diff --git a/LiveControlGateway/Controllers/HubControllerBase.cs b/LiveControlGateway/Controllers/HubControllerBase.cs
@@ -141,11 +141,17 @@ protected override async Task<bool> TryRegisterConnection()
         return await _hubLifetimeManager.TryAddDeviceConnection(5, this, LinkedToken);
     }
 
+    private bool _unregistered;
+    
     /// <summary>
     /// When the connection is destroyed
     /// </summary>
     protected override async Task UnregisterConnection()
     {
+        if (_unregistered)
+            return;
+        _unregistered = true;
+        
         await _hubLifetimeManager.RemoveDeviceConnection(this);
     }
 

Original file line number	Diff line number	Diff line change
`@@ -141,11 +141,17 @@ protected override async Task<bool> TryRegisterConnection()`
`141`	`141`	`return await _hubLifetimeManager.TryAddDeviceConnection(5, this, LinkedToken);`
`142`	`142`	`}`
`143`	`143`
	`144`	`+ private bool _unregistered;`
	`145`	`+`
`144`	`146`	`/// <summary>`
`145`	`147`	`/// When the connection is destroyed`
`146`	`148`	`/// </summary>`
`147`	`149`	`protected override async Task UnregisterConnection()`
`148`	`150`	`{`
	`151`	`+ if (_unregistered)`
	`152`	`+ return;`
	`153`	`+ _unregistered = true;`
	`154`	`+`
`149`	`155`	`await _hubLifetimeManager.RemoveDeviceConnection(this);`
`150`	`156`	`}`
`151`	`157`