Minimize interface to ephemeral ports.

ipython-kernel/src/IHaskell/IPython/ZeroMQ.hs

@@ -10,7 +10,7 @@ module IHaskell.IPython.ZeroMQ
        , ZeroMQStdin(..)
        , serveProfile
        , serveStdin
-       , ZeroMQEphemeralPorts(..)
+       , ZeroMQEphemeralPorts
        , withEphemeralPorts
        ) where
 
@@ -101,6 +101,8 @@ serveProfile profile debug = do
 
   return channels
 
+-- | Describes ports used when creating an ephemeral ZeroMQ session.
+-- Used to generate the ipython JSON config file.
 data ZeroMQEphemeralPorts
    = ZeroMQEphemeralPorts { ephHbPort      :: !Port
                           , ephControlPort :: !Port
@@ -134,9 +136,14 @@ bindLocalEphemeralPort socket = do
     Just endpointIndex ->
       return endpointIndex
 
--- | Start responding on all ZeroMQ channels used to communicate with IPython
--- with ephemerally allocated ports.
--- Profide the callback with the ports chosen and a ZeroMQInterface.
+-- | @withFork thread main@ runs @thread@ in a separate
+-- thread, and kills it when @main@ finishes.
+withFork :: IO () -> (ThreadId -> IO a) -> IO a
+withFork thread = bracket (forkIO thread) killThread
+
+
+-- | Run session for communicating with an IPython instance on ephemerally allocated
+-- ZMQ4 sockets.  The sockets will be closed when the callback returns.
 withEphemeralPorts :: ByteString
                       -- ^ HMAC encryption key
                    -> Bool
@@ -166,13 +173,11 @@ withEphemeralPorts key debug callback = do
                                         , ephIOPubPort   = iopubPort
                                         , ephSignatureKey = key
                                         }
-
        -- Launch actions to listen to communicate between channels and cockets.
-       _ <- forkIO $ forever $ heartbeat channels heartbeatSocket
-       _ <- forkIO $ forever $ control debug channels controlportSocket
-       _ <- forkIO $ forever $ shell debug channels shellportSocket
-       _ <- forkIO $ forever $ checkedIOpub debug channels iopubSocket
-
+       withFork    (forever $ heartbeat channels heartbeatSocket)       $ \_ -> do
+        withFork   (forever $ control debug channels controlportSocket) $ \_ -> do
+         withFork  (forever $ shell debug channels shellportSocket)     $ \_ -> do
+          withFork (forever $ checkedIOpub debug channels iopubSocket)  $ \_ -> do
            -- Run callback function; provide it with both ports and channels.
            callback ports channels