Fix getNodeStats function, see DP-97

haskell-distributed · Mar 2, 2015 · 16fd0e2 · 16fd0e2
1 parent 4d3472a
commit 16fd0e2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/Control/Distributed/Process/Internal/Primitives.hs b/src/Control/Distributed/Process/Internal/Primitives.hs
@@ -757,11 +757,11 @@ getNodeStats nid = do
     selfNode <- getSelfNode
     if nid == selfNode
       then Right `fmap` getLocalNodeStats -- optimisation
-      else getNodeStatsRemote
+      else getNodeStatsRemote selfNode
   where
-    getNodeStatsRemote :: Process (Either DiedReason NodeStats)
-    getNodeStatsRemote = do
-        sendCtrlMsg (Just nid) $ GetNodeStats nid
+    getNodeStatsRemote :: NodeId -> Process (Either DiedReason NodeStats)
+    getNodeStatsRemote selfNode = do
+        sendCtrlMsg (Just nid) $ GetNodeStats selfNode
         bracket (monitorNode nid) unmonitor $ \mRef ->
             receiveWait [ match (\(stats :: NodeStats) -> return $ Right stats)
                         , matchIf (\(NodeMonitorNotification ref _ _) -> ref == mRef)