1 files changed, 11 insertions, 9 deletions
diff --git a/hsm-core/Hsm/Core/Zmq/Server.hs b/hsm-core/Hsm/Core/Zmq/Server.hs
index 6ea8aeb..fd2f159 100644
--- a/hsm-core/Hsm/Core/Zmq/Server.hs
+++ b/hsm-core/Hsm/Core/Zmq/Server.hs
@@ -54,12 +54,14 @@ runServer ::
      )
   => Eff (Server : es) a
   -> Eff es a
-runServer action =
-  withSocket Z.Pub >>= \sock ->
-    E.evalStaticRep (Server sock) $ do
-      localDomain domain $ do
-        logInfo_ "Initializing ZMQ server"
-        env <- ask @env
-        E.unsafeEff_ $ Z.bind sock $ unpack env.pubEp
-        logTrace_ $ "Publishing to " <> env.pubEp
-      action
+runServer action = do
+  sock <- localDomain domain $ withSocket Z.Pub
+  E.evalStaticRep (Server sock) $ do
+    localDomain domain $ initialize sock
+    action
+  where
+    initialize sock = do
+      logInfo_ "Initializing ZMQ server"
+      env <- ask @env
+      E.unsafeEff_ $ Z.bind sock $ unpack env.pubEp
+      logTrace_ $ "Publishing to " <> env.pubEp