master...a5191ce · distributed-process

git.iko.soy/distributed-process

on a5191ce help sign in

distributed-process-tests/src/Control/Distributed/Process/Tests/CH.hs+6−7

since splitsnapshots

unifiedsplit

@@ -44,7 +44,7 @@ import Control.Distributed.Process.Internal.Types

44import Control.Distributed.Process.Node

45import Control.Distributed.Process.Debug

46import Control.Distributed.Process.Management.Internal.Types

47−import Control.Distributed.Process.Tests.Internal.Utils (shouldBe)

47+import Control.Distributed.Process.Tests.Internal.Utils (shouldBe, pause)

48import Control.Distributed.Process.Serializable (Serializable)

50import Test.HUnit (Assertion, assertBool, assertFailure)

@@ -87,10 +87,9 @@ verifyClient :: String -> MVar Bool -> IO ()

87verifyClient s b = takeMVar b >>= assertBool s

89expectPing :: MVar Bool -> Process ()

90−expectPing mv = expect >>= liftIO . putMVar mv . checkPing

90+expectPing mv = expect >>= liftIO . putMVar mv . checkPing

91 where

92 checkPing (Ping _) = True

93− checkPing _ = False

95-- | Quick and dirty synchronous version of whereisRemoteAsync

94-- | Quick and dirty synchronous version of whereisRemoteAsync

96whereisRemote :: NodeId -> String -> Process (Maybe ProcessId)

95whereisRemote :: NodeId -> String -> Process (Maybe ProcessId)

@@ -240,10 +239,10 @@ testPing TestTransport{..} = do

240 pid <- getSelfPid

239 pid <- getSelfPid

241 replicateM_ numPings $ do

240 replicateM_ numPings $ do

242 send pingServer (Pong pid)

241 send pingServer (Pong pid)

243− p <- expect

242+ p <- expectTimeout 3000000

244 case p of

243 case p of

245− Ping _ -> return ()

244+ Just (Ping _) -> return ()

246− _ -> die "Unexpected message"

245+ Nothing -> die "Failed to receive Ping"

247

246

248 putMVar clientDone ()

247 putMVar clientDone ()

249

248

@@ -560,7 +559,7 @@ testMergeChannels TestTransport{..} = do

560 ss <- liftIO $ mapM readMVar vs

559 ss <- liftIO $ mapM readMVar vs

561 case ss of

560 case ss of

562 [sa, sb, sc] ->

561 [sa, sb, sc] ->

563− mapM_ ((>> liftIO (threadDelay 10000)) . uncurry sendChan)

562+ mapM_ ((>> pause 10000) . uncurry sendChan)

564 [ -- a, b, c

563 [ -- a, b, c

565 (sa, 'a')

564 (sa, 'a')

566 , (sb, 'b')

565 , (sb, 'b')

distributed-process-tests/src/Control/Distributed/Process/Tests/Internal/Utils.hs+2−1

@@ -74,12 +74,13 @@ import Control.Concurrent

74import Control.Concurrent.MVar

75 ( putMVar

76 )

77−import Control.Distributed.Process

77+import Control.Distributed.Process hiding (finally, catch)

78import Control.Distributed.Process.Node

79import Control.Distributed.Process.Serializable()

81import Control.Exception (AsyncException(ThreadKilled), SomeException)

82import Control.Monad (forever, void)

83+import Control.Monad.Catch (finally, catch)

83import Control.Monad.STM (atomically)

84import Control.Monad.STM (atomically)

84import Control.Rematch hiding (match)

85import Control.Rematch hiding (match)

85import Control.Rematch.Run

86import Control.Rematch.Run

distributed-process-tests/src/Control/Distributed/Process/Tests/Mx.hs+148−16

@@ -1,11 +1,23 @@

1−{-# LANGUAGE DeriveGeneric #-}

1+{-# LANGUAGE DeriveGeneric #-}

2+{-# LANGUAGE ParallelListComp #-}

2module Control.Distributed.Process.Tests.Mx (tests) where

4module Control.Distributed.Process.Tests.Mx (tests) where

4import Control.Distributed.Process.Tests.Internal.Utils

6import Control.Distributed.Process.Tests.Internal.Utils

5import Network.Transport.Test (TestTransport(..))

7import Network.Transport.Test (TestTransport(..))

6−

8+import Control.Exception (SomeException)

9+import Control.Distributed.Process hiding (bracket, finally, try)

7−import Control.Distributed.Process

10+import Control.Distributed.Process.Internal.Types

11+ ( ProcessExitException(..)

12+ , unsafeCreateUnencodedMessage

13+ )

8import Control.Distributed.Process.Node

14import Control.Distributed.Process.Node

15+import qualified Control.Distributed.Process.UnsafePrimitives as Unsafe

16+ ( send

17+ , nsend

18+ , nsendRemote

19+ , usend

20+ )

9import Control.Distributed.Process.Management

21import Control.Distributed.Process.Management

10 ( MxEvent(..)

22 ( MxEvent(..)

11 , MxAgentId(..)

23 , MxAgentId(..)

@@ -22,12 +34,13 @@ import Control.Distributed.Process.Management

22 , mxBroadcast

34 , mxBroadcast

23 )

35 )

24import Control.Monad (void, unless)

36import Control.Monad (void, unless)

37+import Control.Monad.Catch(finally, bracket, try)

25import Control.Rematch (equalTo)

38import Control.Rematch (equalTo)

26import Data.Binary

39import Data.Binary

27−import Data.List (find, sort)

40+import Data.List (find, sort, intercalate)

28−import Data.Maybe (isJust, isNothing)

41+import Data.Maybe (isJust, fromJust, isNothing)

29import Data.Typeable

42import Data.Typeable

30−import GHC.Generics

43+import GHC.Generics hiding (from)

31#if ! MIN_VERSION_base(4,6,0)

44#if ! MIN_VERSION_base(4,6,0)

32import Prelude hiding (catch, log)

45import Prelude hiding (catch, log)

33#endif

46#endif

@@ -51,7 +64,7 @@ awaitExit pid =

51 (\_ -> return ())

64 (\_ -> return ())

52 ]

65 ]

53 where

66 where

54− withMonitorRef pid = bracket (monitor pid) unmonitor

67+ withMonitorRef p = bracket (monitor p) unmonitor

56testAgentBroadcast :: TestResult () -> Process ()

69testAgentBroadcast :: TestResult () -> Process ()

57testAgentBroadcast result = do

70testAgentBroadcast result = do

@@ -313,43 +326,162 @@ testMxRegMon remoteNode result = do

313ensure :: Process () -> Process () -> Process ()

326ensure :: Process () -> Process () -> Process ()

314ensure = flip finally

327ensure = flip finally

315

328

329+testNSend :: (String -> () -> Process ())

330+ -> Maybe LocalNode

331+ -> TestResult Bool

332+ -> Process ()

333+testNSend op n r = testMxSend n r $ \p1 sink -> do

334+ let delay = 5000000

335+ let label = "testMxSend"

336+ let isValid = isValidLabel n label

337+

338+ register label p1

339+ reg1 <- receiveChanTimeout delay sink

340+ case reg1 of

341+ Just (MxRegistered pd lb)

342+ | pd == p1 && isValid lb -> return ()

343+ _ -> die $ "Reg-Failed: " ++ show reg1

344+

345+ op label ()

346+

347+ us <- getSelfPid

348+ sent <- receiveChanTimeout delay sink

349+ case sent of

350+ Just (MxSentToName lb by _)

351+ | by == us && isValid lb -> return True

352+ _ -> die $ "Send-Failed: " ++ show sent

353+

354+ where

355+ isValidLabel n l1 l2

356+ | l2 == l2 = True

357+ | isJust n = l2 == l1 ++ "@" ++ show (localNodeId $ fromJust n)

358+ | otherwise = False

359+

360+testSend :: (ProcessId -> () -> Process ())

361+ -> Maybe LocalNode

362+ -> TestResult Bool

363+ -> Process ()

364+testSend op n r = testMxSend n r $ \p1 sink -> do

365+ -- initiate a send

366+ op p1 ()

367+

368+ -- verify the management event

369+ us <- getSelfPid

370+ sent <- receiveChanTimeout 5000000 sink

371+ case sent of

372+ Just (MxSent pidTo pidFrom _)

373+ | pidTo == p1 && pidFrom == us -> return True

374+ _ -> return False

375+

376+type SendTest = ProcessId -> ReceivePort MxEvent -> Process Bool

377+

378+testMxSend :: Maybe LocalNode -> TestResult Bool -> SendTest -> Process ()

379+testMxSend mNode result test = do

380+ us <- getSelfPid

381+ (chan, sink) <- newChan

382+ agent <- mxAgent (MxAgentId $ agentLabel us) () [

383+ mxSink $ \ev -> do

384+ case ev of

385+ m@(MxSent _ fromPid _)

386+ | fromPid == us -> liftMX $ sendChan chan m

387+ m@(MxSentToName _ fromPid _)

388+ | fromPid == us -> liftMX $ sendChan chan m

389+ m@(MxRegistered _ name)

390+ | name == label -> liftMX $ sendChan chan m

391+ _ -> return ()

392+ mxReady

393+ ]

394+

395+ (sp, rp) <- newChan

396+ case mNode of

397+ Nothing -> void $ spawnLocal (proc sp)

398+ Just remoteNode -> void $ liftIO $ forkProcess remoteNode $ proc sp

399+

400+ p1 <- receiveChan rp

401+ res <- try (test p1 sink)

402+ case res of

403+ Left (ProcessExitException _ m) -> (liftIO $ putStrLn $ "SomeException-" ++ show m) >> stash result False

404+ Right tr -> stash result tr

405+ kill agent "bye"

406+ kill p1 "bye"

407+

408+ where

409+ label = "testMxSend"

410+ agentLabel s = "mx-unsafe-check-agent-" ++ show s

411+ proc sp' = getSelfPid >>= sendChan sp' >> expect :: Process ()

412+

316tests :: TestTransport -> IO [Test]

413tests :: TestTransport -> IO [Test]

317tests TestTransport{..} = do

414tests TestTransport{..} = do

318 node1 <- newLocalNode testTransport initRemoteTable

415 node1 <- newLocalNode testTransport initRemoteTable

319 node2 <- newLocalNode testTransport initRemoteTable

416 node2 <- newLocalNode testTransport initRemoteTable

320 return [

417 return [

321− testGroup "Mx Agents" [

418+ testGroup "MxAgents" [

322− testCase "Event Handling"

419+ testCase "EventHandling"

323 (delayedAssertion

420 (delayedAssertion

324 "expected True, but events where not as expected"

421 "expected True, but events where not as expected"

325 node1 True testAgentEventHandling)

422 node1 True testAgentEventHandling)

326− , testCase "Inter-Agent Broadcast"

423+ , testCase "InterAgentBroadcast"

327 (delayedAssertion

424 (delayedAssertion

328 "expected (), but no broadcast was received"

425 "expected (), but no broadcast was received"

329 node1 () testAgentBroadcast)

426 node1 () testAgentBroadcast)

330− , testCase "Agent Mailbox Handling"

427+ , testCase "AgentMailboxHandling"

331 (delayedAssertion

428 (delayedAssertion

332 "expected (Just ()), but no regular (mailbox) input was handled"

429 "expected (Just ()), but no regular (mailbox) input was handled"

333 node1 (Just ()) testAgentMailboxHandling)

430 node1 (Just ()) testAgentMailboxHandling)

334− , testCase "Agent Dual Input Handling"

431+ , testCase "AgentDualInputHandling"

335 (delayedAssertion

432 (delayedAssertion

336 "expected sum = 15, but the result was Nothing"

433 "expected sum = 15, but the result was Nothing"

337 node1 (Just 15 :: Maybe Int) testAgentDualInput)

434 node1 (Just 15 :: Maybe Int) testAgentDualInput)

338− , testCase "Agent Input Prioritisation"

435+ , testCase "AgentInputPrioritisation"

339 (delayedAssertion

436 (delayedAssertion

340 "expected [first, second, third, fourth, fifth], but result diverged"

437 "expected [first, second, third, fourth, fifth], but result diverged"

341 node1 (sort ["first", "second",

438 node1 (sort ["first", "second",

342 "third", "fourth",

439 "third", "fourth",

343 "fifth"]) testAgentPrioritisation)

440 "fifth"]) testAgentPrioritisation)

344 ]

441 ]

345− , testGroup "Mx Events" [

442+ , testGroup "MxEvents" [

346− testCase "Name Registration Events"

443+ testCase "NameRegistrationEvents"

347 (delayedAssertion

444 (delayedAssertion

348 "expected registration events to map to the correct ProcessId"

445 "expected registration events to map to the correct ProcessId"

349 node1 () testMxRegEvents)

446 node1 () testMxRegEvents)

350− , testCase "Post Death Name UnRegistration Events"

447+ , testCase "PostDeathNameUnRegistrationEvents"

351 (delayedAssertion

448 (delayedAssertion

352 "expected process deaths to result in unregistration events"

449 "expected process deaths to result in unregistration events"

353 node1 () (testMxRegMon node2))

450 node1 () (testMxRegMon node2))

451+ , testGroup "SendEvents" $ buildTestCases node1 node2

354 ]

452 ]

355 ]

453 ]

454+ where

455+ buildTestCases n1 n2 =

456+ let nid = localNodeId n2 in build n2 [

457+ ("NSend", [

458+ ("Unsafe.nsend", testNSend Unsafe.nsend)

459+ , ("Unsafe.nsendRemote", testNSend (Unsafe.nsendRemote nid))

460+ , ("nsend", testNSend nsend)

461+ , ("nsendRemote", testNSend (nsendRemote nid))

462+ ])

463+ , ("Send", [

464+ ("Unsafe.send", testSend Unsafe.send)

465+ , ("Unsafe.usend", testSend Unsafe.usend)

466+ , ("send", testSend send)

467+ , ("usend", testSend usend)

468+ ])

469+ , ("Forward", [

470+ ("forward", testSend (\p m -> forward (unsafeCreateUnencodedMessage m) p))

471+ , ("uforward", testSend (\p m -> uforward (unsafeCreateUnencodedMessage m) p))

472+ ])

473+ ]

474+

475+ build :: LocalNode

476+ -> [(String, [(String, (Maybe LocalNode -> TestResult Bool -> Process ()))])]

477+ -> [Test]

478+ build ln specs =

479+ [ testGroup (intercalate "-" [groupName, caseSuffix]) [

480+ testCase (intercalate "-" [caseName, caseSuffix])

481+ (delayedAssertion "expected mx events failed"

482+ ln True (caseImpl caseNode))

483+ | (caseName, caseImpl) <- groupCases

484+ ]

485+ | (groupName, groupCases) <- specs

486+ , (caseSuffix, caseNode) <- [("RemotePid", Just ln), ("LocalPid", Nothing)]

487+ ]

src/Control/Distributed/Process/Internal/Primitives.hs+47−40

@@ -254,17 +254,15 @@ send them msg = do

254 node = processNode proc

255 nodeId = localNodeId node

256 destNode = (processNodeId them) in do

257− case destNode == nodeId of

258− True -> sendLocal them msg

259− False -> liftIO $ sendMessage (processNode proc)

260− (ProcessIdentifier (processId proc))

261− (ProcessIdentifier them)

262− NoImplicitReconnect

263− msg

264− -- We do not fire the trace event until after the sending is complete;

265− -- In the remote case, 'sendMessage' can block in the networking stack.

266 liftIO $ traceEvent (localEventBus node)

257 liftIO $ traceEvent (localEventBus node)

267 (MxSent them us (createUnencodedMessage msg))

258 (MxSent them us (createUnencodedMessage msg))

259+ if destNode == nodeId

260+ then sendLocal them msg

261+ else liftIO $ sendMessage (processNode proc)

262+ (ProcessIdentifier (processId proc))

263+ (ProcessIdentifier them)

264+ NoImplicitReconnect

265+ msg

268

266

269-- | /Unsafe/ variant of 'send'. This function makes /no/ attempt to serialize

267-- | /Unsafe/ variant of 'send'. This function makes /no/ attempt to serialize

270-- and (in the case when the destination process resides on the same local

268-- and (in the case when the destination process resides on the same local

@@ -284,9 +282,12 @@ unsafeSend = Unsafe.send

284--

282--

285usend :: Serializable a => ProcessId -> a -> Process ()

283usend :: Serializable a => ProcessId -> a -> Process ()

286usend them msg = do

284usend them msg = do

287− here <- getSelfNode

285+ proc <- ask

288 let there = processNodeId them

286 let there = processNodeId them

289− if here == there

287+ let (us, node) = (processId proc, processNode proc)

288+ let msg' = wrapMessage msg

289+ liftIO $ traceEvent (localEventBus node) (MxSent them us msg')

290+ if localNodeId (processNode proc) == there

290 then sendLocal them msg

291 then sendLocal them msg

291 else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)

292 else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)

292 (createMessage msg)

293 (createMessage msg)

@@ -334,13 +335,16 @@ newChan = do

334sendChan :: Serializable a => SendPort a -> a -> Process ()

335sendChan :: Serializable a => SendPort a -> a -> Process ()

335sendChan (SendPort cid) msg = do

336sendChan (SendPort cid) msg = do

336 proc <- ask

337 proc <- ask

337− let node = localNodeId (processNode proc)

338+ let node = processNode proc

339+ pid = processId proc

340+ us = localNodeId node

338− destNode = processNodeId (sendPortProcessId cid) in do

341+ them = processNodeId (sendPortProcessId cid) in do

342+ liftIO $ traceEvent (localEventBus node) (MxSentToPort pid cid $ wrapMessage msg)

339− case destNode == node of

343+ case them == us of

340 True -> sendChanLocal cid msg

344 True -> sendChanLocal cid msg

341 False -> do

345 False -> do

342− liftIO $ sendBinary (processNode proc)

346+ liftIO $ sendBinary node

343− (ProcessIdentifier (processId proc))

347+ (ProcessIdentifier pid)

344 (SendPortIdentifier cid)

348 (SendPortIdentifier cid)

345 NoImplicitReconnect

349 NoImplicitReconnect

346 msg

350 msg

@@ -484,17 +488,14 @@ forward msg them = do

484 us = processId proc

488 us = processId proc

485 nid = localNodeId node

489 nid = localNodeId node

486 destNode = (processNodeId them) in do

490 destNode = (processNodeId them) in do

491+ liftIO $ traceEvent (localEventBus node) (MxSent them us msg)

487− case destNode == nid of

492+ if destNode == nid

488− True -> sendCtrlMsg Nothing (LocalSend them msg)

493+ then sendCtrlMsg Nothing (LocalSend them msg)

489− False -> liftIO $ sendPayload (processNode proc)

494+ else liftIO $ sendPayload (processNode proc)

490− (ProcessIdentifier (processId proc))

495+ (ProcessIdentifier (processId proc))

491− (ProcessIdentifier them)

496+ (ProcessIdentifier them)

492− NoImplicitReconnect

497+ NoImplicitReconnect

493− (messageToPayload msg)

498+ (messageToPayload msg)

494− -- We do not fire the trace event until after the sending is complete;

495− -- In the remote case, 'sendMessage' can block in the networking stack.

496− liftIO $ traceEvent (localEventBus node)

497− (MxSent them us msg)

498

499

499-- | Forward a raw 'Message' to the given 'ProcessId'.

500-- | Forward a raw 'Message' to the given 'ProcessId'.

500--

501--

@@ -508,14 +509,10 @@ uforward msg them = do

508 us = processId proc

509 us = processId proc

509 nid = localNodeId node

510 nid = localNodeId node

510 destNode = (processNodeId them) in do

511 destNode = (processNodeId them) in do

512+ liftIO $ traceEvent (localEventBus node) (MxSent them us msg)

511− case destNode == nid of

513+ if destNode == nid

512− True -> sendCtrlMsg Nothing (LocalSend them msg)

514+ then sendCtrlMsg Nothing (LocalSend them msg)

513− False -> sendCtrlMsg (Just destNode) $ UnreliableSend (processLocalId them)

515+ else sendCtrlMsg (Just destNode) $ UnreliableSend (processLocalId them) msg

514− msg

515− -- We do not fire the trace event until after the sending is complete;

516− -- In the remote case, 'sendCtrlMsg' can block in the networking stack.

517− liftIO $ traceEvent (localEventBus node)

518− (MxSent them us msg)

519

516

520-- | Wrap a 'Serializable' value in a 'Message'. Note that 'Message's are

517-- | Wrap a 'Serializable' value in a 'Message'. Note that 'Message's are

521-- 'Serializable' - like the datum they contain - but also note, deserialising

518-- 'Serializable' - like the datum they contain - but also note, deserialising

@@ -1224,8 +1221,12 @@ whereisRemoteAsync nid label = do

1224

1221

1225-- | Named send to a process in the local registry (asynchronous)

1222-- | Named send to a process in the local registry (asynchronous)

1226nsend :: Serializable a => String -> a -> Process ()

1223nsend :: Serializable a => String -> a -> Process ()

1227−nsend label msg =

1224+nsend label msg = do

1225+ proc <- ask

1226+ let msg' = createUnencodedMessage msg

1227+ liftIO $ traceEvent (localEventBus (processNode proc))

1228+ (MxSentToName label (processId proc) msg')

1228− sendCtrlMsg Nothing (NamedSend label (createUnencodedMessage msg))

1229+ sendCtrlMsg Nothing (NamedSend label msg')

1229

1230

1230-- | Named send to a process in the local registry (asynchronous).

1231-- | Named send to a process in the local registry (asynchronous).

1231-- This function makes /no/ attempt to serialize and (in the case when the

1232-- This function makes /no/ attempt to serialize and (in the case when the

@@ -1237,9 +1238,15 @@ unsafeNSend = Unsafe.nsend

1237-- | Named send to a process in a remote registry (asynchronous)

1238-- | Named send to a process in a remote registry (asynchronous)

1238nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()

1239nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()

1239nsendRemote nid label msg = do

1240nsendRemote nid label msg = do

1240− here <- getSelfNode

1241+ proc <- ask

1242+ let us = processId proc

1243+ let node = processNode proc

1244+ if localNodeId node == nid

1241− if here == nid then nsend label msg

1245+ then nsend label msg

1246+ else let lbl = label ++ "@" ++ show nid in do

1247+ liftIO $ traceEvent (localEventBus node)

1248+ (MxSentToName lbl us (wrapMessage msg))

1242− else sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))

1249+ sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))

1243

1250

1244-- | Named send to a process in a remote registry (asynchronous)

1251-- | Named send to a process in a remote registry (asynchronous)

1245-- This function makes /no/ attempt to serialize and (in the case when the

1252-- This function makes /no/ attempt to serialize and (in the case when the

src/Control/Distributed/Process/Management/Internal/Types.hs+7

@@ -21,6 +21,7 @@ import Control.Concurrent.STM

21import Control.Distributed.Process.Internal.Types

22 ( Process

23 , ProcessId

24+ , SendPortId

24 , Message

25 , Message

25 , DiedReason

26 , DiedReason

26 , NodeId

27 , NodeId

@@ -57,8 +58,14 @@ data MxEvent =

57 -- ^ fired whenever a node /dies/ (i.e., the connection is broken/disconnected)

58 -- ^ fired whenever a node /dies/ (i.e., the connection is broken/disconnected)

58 | MxSent ProcessId ProcessId Message

59 | MxSent ProcessId ProcessId Message

59 -- ^ fired whenever a message is sent from a local process

60 -- ^ fired whenever a message is sent from a local process

61+ | MxSentToName String ProcessId Message

62+ -- ^ fired whenever a named send occurs

63+ | MxSentToPort ProcessId SendPortId Message

64+ -- ^ fired whenever a sendChan occurs

60 | MxReceived ProcessId Message

65 | MxReceived ProcessId Message

61 -- ^ fired whenever a message is received by a local process

66 -- ^ fired whenever a message is received by a local process

67+ | MxReceivedPort SendPortId Message

68+ -- ^ fired whenever a message is received via a typed channel

62 | MxConnected ConnectionId EndPointAddress

69 | MxConnected ConnectionId EndPointAddress

63 -- ^ fired when a network-transport connection is first established

70 -- ^ fired when a network-transport connection is first established

64 | MxDisconnected ConnectionId EndPointAddress

71 | MxDisconnected ConnectionId EndPointAddress

src/Control/Distributed/Process/Node.hs+15−10

@@ -492,7 +492,7 @@ type IncomingConnection = (NT.EndPointAddress, IncomingTarget)

492data IncomingTarget =

493 Uninit

494 | ToProc ProcessId (Weak (CQueue Message))

495− | ToChan TypedChannel

495+ | ToChan SendPortId TypedChannel

496 | ToNode

497

498data ConnectionState = ConnectionState {

@@ -547,13 +547,17 @@ handleIncomingMessages node = go initConnectionState

547 enqueue queue msg -- 'enqueue' is strict

548 trace node (MxReceived pid msg)

549 go st

550− Just (_, ToChan (TypedChannel chan')) -> do

550+ Just (_, ToChan chId (TypedChannel chan')) -> do

551 mChan <- deRefWeak chan'

552 -- If mChan is Nothing, the process has given up the read end of

553 -- the channel and we simply ignore the incoming message

554− forM_ mChan $ \chan -> atomically $

554+ forM_ mChan $ \chan -> do

555+ msg' <- atomically $ do

556+ msg <- return $! decode (BSL.fromChunks payload)

555− -- We make sure the message is fully decoded when it is enqueued

557+ -- We make sure the message is fully decoded when it is enqueued

556− writeTQueue chan $! decode (BSL.fromChunks payload)

558+ writeTQueue chan msg

559+ return msg

560+ trace node $ MxReceivedPort chId $ unsafeCreateUnencodedMessage msg'

557 go st

561 go st

558 Just (_, ToNode) -> do

562 Just (_, ToNode) -> do

559 let ctrlMsg = decode . BSL.fromChunks $ payload

563 let ctrlMsg = decode . BSL.fromChunks $ payload

@@ -580,7 +584,7 @@ handleIncomingMessages node = go initConnectionState

580 mChannel <- withMVar (processState proc) $ return . (^. typedChannelWithId lcid)

584 mChannel <- withMVar (processState proc) $ return . (^. typedChannelWithId lcid)

581 case mChannel of

585 case mChannel of

582 Just channel ->

586 Just channel ->

583− go (incomingAt cid ^= Just (src, ToChan channel) $ st)

587+ go (incomingAt cid ^= Just (src, ToChan chId channel) $ st)

584 Nothing ->

588 Nothing ->

585 invalidRequest cid st $

589 invalidRequest cid st $

586 "incoming attempt to connect to unknown channel of"

590 "incoming attempt to connect to unknown channel of"

@@ -1050,11 +1054,12 @@ ncEffectLocalPortSend from msg = do

1050 -- If ch is Nothing, the process has given up the read end of

1054 -- If ch is Nothing, the process has given up the read end of

1051 -- the channel and we simply ignore the incoming message - this

1055 -- the channel and we simply ignore the incoming message - this

1052 ch <- deRefWeak chan'

1056 ch <- deRefWeak chan'

1053− forM_ ch $ \chan -> deliverChan msg chan

1057+ forM_ ch $ \chan -> deliverChan node from msg chan

1054− where deliverChan :: forall a . Message -> TQueue a -> IO ()

1058+ where deliverChan :: forall a . LocalNode -> SendPortId -> Message -> TQueue a -> IO ()

1055− deliverChan (UnencodedMessage _ raw) chan' =

1059+ deliverChan n p (UnencodedMessage _ raw) chan' = do

1056 atomically $ writeTQueue chan' ((unsafeCoerce raw) :: a)

1060 atomically $ writeTQueue chan' ((unsafeCoerce raw) :: a)

1061+ trace n (MxReceivedPort p $ unsafeCreateUnencodedMessage raw)

1057− deliverChan (EncodedMessage _ _) _ =

1062+ deliverChan _ _ (EncodedMessage _ _) _ =

1058 -- this will not happen unless someone screws with Primitives.hs

1063 -- this will not happen unless someone screws with Primitives.hs

1059 error "invalid local channel delivery"

1064 error "invalid local channel delivery"

1060

1065

src/Control/Distributed/Process/UnsafePrimitives.hs+82−33

@@ -32,7 +32,24 @@

32-- the /normal/ strategy).

33--

34-- Use of the functions in this module can potentially change the runtime

35+-- behaviour of your application. In addition, messages passed between Cloud

36+-- Haskell processes are written to a tracing infrastructure on the local node,

37+-- to provide improved introspection and debugging facilities for complex actor

38+-- based systems. This module makes no attempt to force evaluation in these

39+-- cases either, thus evaluation problems in passed data structures could not

40+-- only crash your processes, but could also bring down critical internal

41+-- services on which the node relies to function correctly.

42+--

43+-- If you wish to repudiate such issues, you are advised to consider the use

44+-- of NFSerialisable in the distributed-process-extras package, which type

45+-- class brings NFData into scope along with Serializable, such that we can

46+-- force evaluation. Intended for use with modules such as this one, this

47+-- approach guarantees correct evaluatedness in terms of @NFData@. Please note

48+-- however, that we /cannot/ guarantee that an @NFData@ instance will behave the

49+-- same way as a @Binary@ one with regards evaluation, so it is still possible

50+-- to introduce unexpected behaviour by using /unsafe/ primitives in this way.

51+--

35−-- behaviour of your application. You have been warned!

52+-- You have been warned!

36--

53--

37-- This module is exported so that you can replace the use of Cloud Haskell's

54-- This module is exported so that you can replace the use of Cloud Haskell's

38-- /safe/ messaging primitives. If you want to use both variants, then you can

55-- /safe/ messaging primitives. If you want to use both variants, then you can

@@ -55,7 +72,12 @@ import Control.Distributed.Process.Internal.Messaging

55 , sendBinary

72 , sendBinary

56 , sendCtrlMsg

73 , sendCtrlMsg

57 )

74 )

58−

75+import Control.Distributed.Process.Management.Internal.Types

76+ ( MxEvent(..)

77+ )

78+import Control.Distributed.Process.Management.Internal.Trace.Types

79+ ( traceEvent

80+ )

59import Control.Distributed.Process.Internal.Types

81import Control.Distributed.Process.Internal.Types

60 ( ProcessId(..)

82 ( ProcessId(..)

61 , NodeId(..)

83 , NodeId(..)

@@ -79,34 +101,48 @@ import Control.Monad.Reader (ask)

101

80-- | Named send to a process in the local registry (asynchronous)

102-- | Named send to a process in the local registry (asynchronous)

81nsend :: Serializable a => String -> a -> Process ()

103nsend :: Serializable a => String -> a -> Process ()

82−nsend label msg =

104+nsend label msg = do

105+ proc <- ask

106+ let us = processId proc

107+ let msg' = wrapMessage msg

108+ -- see [note: tracing]

109+ liftIO $ traceEvent (localEventBus (processNode proc))

110+ (MxSentToName label us msg')

83− sendCtrlMsg Nothing (NamedSend label (unsafeCreateUnencodedMessage msg))

111+ sendCtrlMsg Nothing (NamedSend label msg')

112

85-- | Named send to a process in a remote registry (asynchronous)

113-- | Named send to a process in a remote registry (asynchronous)

86nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()

114nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()

87nsendRemote nid label msg = do

115nsendRemote nid label msg = do

88 proc <- ask

116 proc <- ask

117+ let us = processId proc

118+ let node = processNode proc

89− if localNodeId (processNode proc) == nid

119+ if localNodeId node == nid

90 then nsend label msg

120 then nsend label msg

121+ else

122+ let lbl = label ++ "@" ++ show nid in do

123+ -- see [note: tracing] NB: this is a remote call to another NC...

124+ liftIO $ traceEvent (localEventBus node)

125+ (MxSentToName lbl us (wrapMessage msg))

91− else sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))

126+ sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))

127

93-- | Send a message

128-- | Send a message

94send :: Serializable a => ProcessId -> a -> Process ()

129send :: Serializable a => ProcessId -> a -> Process ()

95send them msg = do

130send them msg = do

96 proc <- ask

131 proc <- ask

97 let node = localNodeId (processNode proc)

132 let node = localNodeId (processNode proc)

98− destNode = (processNodeId them) in do

133+ destNode = (processNodeId them)

134+ us = (processId proc)

135+ msg' = wrapMessage msg in do

136+ -- see [note: tracing]

137+ liftIO $ traceEvent (localEventBus (processNode proc))

138+ (MxSent them us msg')

99− case destNode == node of

139+ if destNode == node

100− True -> unsafeSendLocal them msg

101− False -> liftIO $ sendMessage (processNode proc)

102− (ProcessIdentifier (processId proc))

103− (ProcessIdentifier them)

104− NoImplicitReconnect

105− msg

106− where

107− unsafeSendLocal :: (Serializable a) => ProcessId -> a -> Process ()

108− unsafeSendLocal pid msg' =

109− sendCtrlMsg Nothing $ LocalSend pid (unsafeCreateUnencodedMessage msg')

140+ then sendCtrlMsg Nothing $ LocalSend them msg'

141+ else liftIO $ sendMessage (processNode proc)

142+ (ProcessIdentifier (processId proc))

143+ (ProcessIdentifier them)

144+ NoImplicitReconnect

145+ msg

110

146

111-- | Send a message unreliably.

147-- | Send a message unreliably.

112--

148--

@@ -121,30 +157,43 @@ usend :: Serializable a => ProcessId -> a -> Process ()

121usend them msg = do

157usend them msg = do

122 proc <- ask

158 proc <- ask

123 let there = processNodeId them

159 let there = processNodeId them

160+ let (us, node) = (processId proc, processNode proc)

161+ let msg' = wrapMessage msg

162+ -- see [note: tracing]

163+ liftIO $ traceEvent (localEventBus node) (MxSent them us msg')

124 if localNodeId (processNode proc) == there

164 if localNodeId (processNode proc) == there

125− then sendCtrlMsg Nothing $

165+ then sendCtrlMsg Nothing $ LocalSend them msg'

126− LocalSend them (unsafeCreateUnencodedMessage msg)

127 else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)

166 else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)

128 (createMessage msg)

167 (createMessage msg)

129

168

169+-- [note: tracing]

170+-- Note that tracing writes to the local node's control channel, and this

171+-- module explicitly specifies to its clients that it does unsafe message

172+-- encoding. The same is true for the messages it puts onto the Management

173+-- event bus, however we do *not* want unevaluated thunks hitting the event

174+-- bus control thread. Hence the word /Unsafe/ in this module's name!

175+--

176+

130-- | Send a message on a typed channel

177-- | Send a message on a typed channel

131sendChan :: Serializable a => SendPort a -> a -> Process ()

178sendChan :: Serializable a => SendPort a -> a -> Process ()

132sendChan (SendPort cid) msg = do

179sendChan (SendPort cid) msg = do

133 proc <- ask

180 proc <- ask

181+ let node = processNode proc

182+ pid = processId proc

134− let node = localNodeId (processNode proc)

183+ us = localNodeId node

135− destNode = processNodeId (sendPortProcessId cid) in do

184+ them = processNodeId (sendPortProcessId cid)

136− case destNode == node of

185+ msg' = wrapMessage msg in do

186+ liftIO $ traceEvent (localEventBus node) (MxSentToPort pid cid msg')

187+ if them == us

137− True -> unsafeSendChanLocal cid msg

188+ then unsafeSendChanLocal cid msg' -- NB: we wrap to P.Message !!!

138− False -> do

139− liftIO $ sendBinary (processNode proc)

189+ else liftIO $ sendBinary node

140− (ProcessIdentifier (processId proc))

190+ (ProcessIdentifier pid)

141− (SendPortIdentifier cid)

191+ (SendPortIdentifier cid)

142− NoImplicitReconnect

192+ NoImplicitReconnect

143− msg

193+ msg

144 where

194 where

145− unsafeSendChanLocal :: (Serializable a) => SendPortId -> a -> Process ()

195+ unsafeSendChanLocal :: SendPortId -> Message -> Process ()

146− unsafeSendChanLocal spId msg' =

147− sendCtrlMsg Nothing $ LocalPortSend spId (unsafeCreateUnencodedMessage msg')

196+ unsafeSendChanLocal p m = sendCtrlMsg Nothing $ LocalPortSend p m

148

197

149-- | Create an unencoded @Message@ for any @Serializable@ type.

198-- | Create an unencoded @Message@ for any @Serializable@ type.

150wrapMessage :: Serializable a => a -> Message

199wrapMessage :: Serializable a => a -> Message