Adding CTs to Receiving

Author: PaciStardust

CoreOSC/IOscListener.cs

@@ -1,15 +1,16 @@
 using System.Net;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace LucHeart.CoreOSC;
 
 public interface IOscListener
 {
-    public Task<OscMessage> ReceiveMessageAsync();
+    public Task<OscMessage> ReceiveMessageAsync(CancellationToken ct = default);
 
-    public Task<(OscMessage Message, IPEndPoint EndPoint)> ReceiveMessageExAsync();
+    public Task<(OscMessage Message, IPEndPoint EndPoint)> ReceiveMessageExAsync(CancellationToken ct = default);
 
-    public Task<OscBundle> ReceiveBundleAsync();
+    public Task<OscBundle> ReceiveBundleAsync(CancellationToken ct = default);
 
-    public Task<(OscBundle Bundle, IPEndPoint EndPoint)> ReceiveBundleExAsync();
+    public Task<(OscBundle Bundle, IPEndPoint EndPoint)> ReceiveBundleExAsync(CancellationToken ct = default);
 }

CoreOSC/OscListener.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Net;
 using System.Net.Sockets;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace LucHeart.CoreOSC;
@@ -42,14 +43,14 @@ public OscListener(IPEndPoint listenerEndPoint)
         }
     }
 
-    public async Task<OscMessage> ReceiveMessageAsync()
+    public async Task<OscMessage> ReceiveMessageAsync(CancellationToken ct = default)
     {
         if (EnableTransparentBundleToMessageConversion)
         {
             if (MessageQueue.Count > 0)
                 return MessageQueue.Dequeue();
 
-            var receiveResult = await UdpClient.ReceiveAsync();
+            var receiveResult = await UdpClient.ReceiveAsync(ct);
 
             if (!OscBundle.IsBundle(receiveResult.Buffer))
                 return OscMessage.ParseMessage(receiveResult.Buffer);
@@ -62,26 +63,26 @@ public async Task<OscMessage> ReceiveMessageAsync()
         }
         else
         {
-            var receiveResult = await UdpClient.ReceiveAsync();
+            var receiveResult = await UdpClient.ReceiveAsync(ct);
             return OscMessage.ParseMessage(receiveResult.Buffer);
         }
     }
 
-    public async Task<(OscMessage Message, IPEndPoint EndPoint)> ReceiveMessageExAsync()
+    public async Task<(OscMessage Message, IPEndPoint EndPoint)> ReceiveMessageExAsync(CancellationToken ct = default)
     {
-        var receiveResult = await UdpClient.ReceiveAsync();
+        var receiveResult = await UdpClient.ReceiveAsync(ct);
         return (OscMessage.ParseMessage(receiveResult.Buffer), receiveResult.RemoteEndPoint);
     }
 
-    public async Task<OscBundle> ReceiveBundleAsync()
+    public async Task<OscBundle> ReceiveBundleAsync(CancellationToken ct = default)
     {
-        var receiveResult = await UdpClient.ReceiveAsync();
+        var receiveResult = await UdpClient.ReceiveAsync(ct);
         return OscBundle.ParseBundle(receiveResult.Buffer);
     }
 
-    public async Task<(OscBundle Bundle, IPEndPoint EndPoint)> ReceiveBundleExAsync()
+    public async Task<(OscBundle Bundle, IPEndPoint EndPoint)> ReceiveBundleExAsync(CancellationToken ct = default)
     {
-        var receiveResult = await UdpClient.ReceiveAsync();
+        var receiveResult = await UdpClient.ReceiveAsync(ct);
         return (OscBundle.ParseBundle(receiveResult.Buffer), receiveResult.RemoteEndPoint);
     }