From 6cca320c3441cc5fe8833176af34986db123407a Mon Sep 17 00:00:00 2001
From: voidlizard <dzuikov@gmail.com>
Date: Fri, 1 Nov 2024 10:52:29 +0300
Subject: [PATCH] some TCP cleanup

---
 hbs2-core/lib/HBS2/Net/Messaging/TCP.hs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/hbs2-core/lib/HBS2/Net/Messaging/TCP.hs b/hbs2-core/lib/HBS2/Net/Messaging/TCP.hs
index ef5a13ae..27db86f6 100644
--- a/hbs2-core/lib/HBS2/Net/Messaging/TCP.hs
+++ b/hbs2-core/lib/HBS2/Net/Messaging/TCP.hs
@@ -268,9 +268,10 @@ spawnConnection tp env so sa = liftIO do
               sendLazy so frame --(LBS.toStrict frame)
               next
 
-      void $ waitAnyCatchCancel [rd,wr]
+      ContT $ bracket none $ \_ -> mapM cancel [rd,wr]
+      ContT $ bracket (pure connId) cleanupConn
 
-    -- lift $ cleanupConn connId
+      void $ waitAnyCatchCancel [rd,wr]
 
   -- gracefulClose so 1000
   debug $ "spawnConnection exit" <+> pretty sa