Implement unsafeNSendRemote.

facundominguez · facundominguez · commit bb52b3b9a9bc · 2015-11-27T15:57:43.000-03:00
diff --git a/src/Control/Distributed/Process.hs b/src/Control/Distributed/Process.hs
@@ -43,6 +43,7 @@ module Control.Distributed.Process
   , unsafeUSend
   , unsafeSendChan
   , unsafeNSend
+  , unsafeNSendRemote
   , unsafeWrapMessage
     -- * Advanced messaging
   , Match
@@ -216,6 +217,7 @@ import Control.Distributed.Process.Internal.Primitives
   , unsafeUSend
   , unsafeSendChan
   , unsafeNSend
+  , unsafeNSendRemote
     -- Advanced messaging
   , Match
   , receiveWait
diff --git a/src/Control/Distributed/Process/Internal/Primitives.hs b/src/Control/Distributed/Process/Internal/Primitives.hs
@@ -26,6 +26,7 @@ module Control.Distributed.Process.Internal.Primitives
   , unsafeUSend
   , unsafeSendChan
   , unsafeNSend
+  , unsafeNSendRemote
     -- * Advanced messaging
   , Match
   , receiveWait
@@ -1184,6 +1185,13 @@ nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()
 nsendRemote nid label msg =
   sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))
 
+-- | Named send to a process in a remote registry (asynchronous)
+-- This function makes /no/ attempt to serialize and (in the case when the
+-- destination process resides on the same local node) therefore ensure that
+-- the payload is fully evaluated before it is delivered.
+unsafeNSendRemote :: Serializable a => NodeId -> String -> a -> Process ()
+unsafeNSendRemote = Unsafe.nsendRemote
+
 --------------------------------------------------------------------------------
 -- Closures                                                                   --
 --------------------------------------------------------------------------------
diff --git a/src/Control/Distributed/Process/UnsafePrimitives.hs b/src/Control/Distributed/Process/UnsafePrimitives.hs
@@ -45,6 +45,7 @@ module Control.Distributed.Process.UnsafePrimitives
     send
   , sendChan
   , nsend
+  , nsendRemote
   , usend
   , wrapMessage
   ) where
@@ -57,6 +58,7 @@ import Control.Distributed.Process.Internal.Messaging
 
 import Control.Distributed.Process.Internal.Types
   ( ProcessId(..)
+  , NodeId(..)
   , LocalNode(..)
   , LocalProcess(..)
   , Process(..)
@@ -80,6 +82,14 @@ nsend :: Serializable a => String -> a -> Process ()
 nsend label msg =
   sendCtrlMsg Nothing (NamedSend label (unsafeCreateUnencodedMessage msg))
 
+-- | Named send to a process in a remote registry (asynchronous)
+nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()
+nsendRemote nid label msg = do
+  proc <- ask
+  if localNodeId (processNode proc) == nid
+    then nsend label msg
+    else sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))
+
 -- | Send a message
 send :: Serializable a => ProcessId -> a -> Process ()
 send them msg = do
