Fix issue with DataContext.WaitForAsync refactor, where token cancela…

…tion from WaitFor doesn't cause the TaskCompletionSource from WaitForAsync to get canceled as well
0x0ade · Aug 21, 2024 · d0d9f6a · d0d9f6a
1 parent 629a1ce
commit d0d9f6a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CelesteNet.Shared/DataContext.cs b/CelesteNet.Shared/DataContext.cs
@@ -10,8 +10,7 @@
 using Celeste.Mod.CelesteNet.DataTypes;
 using MonoMod.Utils;
 
-namespace Celeste.Mod.CelesteNet
-{
+namespace Celeste.Mod.CelesteNet {
     public delegate void DataHandler(CelesteNetConnection con, DataType data);
     public delegate void DataHandler<T>(CelesteNetConnection con, T data) where T : DataType<T>;
     public delegate bool DataFilter(CelesteNetConnection con, DataType data);
@@ -266,6 +265,7 @@ void RemoveHandler() {
                     if (handler != null)
                         UnregisterHandler(typeof(T), handler);
                     handler = null;
+                    tcs.TrySetCanceled();
                 }
             }