Adds logging domain to ZMQ resource operations

1 files changed, 13 insertions, 11 deletions

diff --git a/hsm-core/Hsm/Core/Zmq/Client.hs b/hsm-core/Hsm/Core/Zmq/Client.hs
index 97728f7..aa74ed3 100644
--- a/hsm-core/Hsm/Core/Zmq/Client.hs
+++ b/hsm-core/Hsm/Core/Zmq/Client.hs
@@ -75,14 +75,16 @@ runClient ::
      )
   => Eff (Client : es) a
   -> Eff es a
-runClient action =
-  withSocket Z.Sub >>= \sock ->
-    E.evalStaticRep (Client sock) $ do
-      localDomain domain $ do
-        logInfo_ "Initializing ZMQ client"
-        env <- ask @env
-        forM_ env.subEps $ E.unsafeEff_ . Z.connect sock . unpack
-        forM_ env.topics $ E.unsafeEff_ . Z.subscribe sock . encodeUtf8
-        logTrace_ $ "Listening to " <> pack (show env.subEps)
-        logTrace_ $ "Subscribed to " <> pack (show env.topics)
-      action
+runClient action = do
+  sock <- localDomain domain $ withSocket Z.Sub
+  E.evalStaticRep (Client sock) $ do
+    localDomain domain $ initialize sock
+    action
+  where
+    initialize sock = do
+      logInfo_ "Initializing ZMQ client"
+      env <- ask @env
+      forM_ env.subEps $ E.unsafeEff_ . Z.connect sock . unpack
+      forM_ env.topics $ E.unsafeEff_ . Z.subscribe sock . encodeUtf8
+      logTrace_ $ "Listening to " <> pack (show env.subEps)
+      logTrace_ $ "Subscribed to " <> pack (show env.topics)