@@ -77,6 +77,7 @@ module Control.Distributed.Process.Internal.Primitives
78 , unmonitor
78 , unmonitor
79 , withMonitor
79 , withMonitor
80 -- * Logging
81 -- * Logging
82 -- * Registry
83 -- * Registry
@@ -887,12 +888,21 @@ monitor = monitor' . ProcessIdentifier
887-- messages in the queue.
888-- messages in the queue.
889withMonitor :: ProcessId -> Process a -> Process a
890withMonitor :: ProcessId -> Process a -> Process a
890−withMonitor pid code = bracket (monitor pid) unmonitor (\_ -> code)
891+withMonitor pid = withMonitorRef pid . const
891 -- unmonitor blocks waiting for the response, so there's a possibility
892 -- unmonitor blocks waiting for the response, so there's a possibility
892 -- that an exception might interrupt withMonitor before the unmonitor
893 -- that an exception might interrupt withMonitor before the unmonitor
893 -- has completed. I think that's better than making the unmonitor
894 -- has completed. I think that's better than making the unmonitor
894 -- uninterruptible.
895 -- uninterruptible.
897+-- | Establishes temporary monitoring of another process.
899+-- @withMonitorRef pid code@ sets up monitoring of @pid@ for the duration
900+-- of @code@. Note: although monitoring is no longer active when
901+-- @withMonitorRef@ returns, there might still be unreceived monitor
902+-- messages in the queue.
903+withMonitorRef :: ProcessId -> (MonitorRef -> Process a) -> Process a
904+withMonitorRef pid code = bracket (monitor pid) unmonitor code
896-- | Remove a link
906-- | Remove a link
898-- This is synchronous in the sense that once it returns you are guaranteed
908-- This is synchronous in the sense that once it returns you are guaranteed