From de672241f7b8824533cbdc9adf258b396e28ce8e Mon Sep 17 00:00:00 2001
From: Oliver Booth <git@olivr.me>
Date: Wed, 14 Feb 2024 15:23:19 +0000
Subject: [PATCH] fix: set tcs result on callback response

---
 TcpDotNet/ClientNode.cs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/TcpDotNet/ClientNode.cs b/TcpDotNet/ClientNode.cs
index bb29390..46aea66 100644
--- a/TcpDotNet/ClientNode.cs
+++ b/TcpDotNet/ClientNode.cs
@@ -16,6 +16,7 @@ public abstract class ClientNode : Node
 {
     private readonly ConcurrentDictionary<int, List<TaskCompletionSource<Packet>>> _packetCompletionSources = new();
     private readonly ConcurrentDictionary<long, TaskCompletionSource<ResponsePacket>> _callbackCompletionSources = new();
+
     private EndPoint? _remoteEP;
 
     /// <summary>
@@ -160,6 +161,9 @@ public abstract class ClientNode : Node
             }
         }
 
+        if (packet is ResponsePacket response && _callbackCompletionSources.TryGetValue(response.CallbackId, out TaskCompletionSource<ResponsePacket>? callback))
+            callback.SetResult(response);
+
         return packet;
     }