Implement onConnectionEvent

input-output-hk · Apr 17, 2024 · 4524157 · 4524157
1 parent 74c5e61
commit 4524157
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/hydra-node/src/Hydra/HeadLogic.hs b/hydra-node/src/Hydra/HeadLogic.hs
@@ -88,7 +88,7 @@ import Hydra.Ledger (
   applyTransactions,
   txId,
  )
-import Hydra.Network.Message (Message (..), NetworkMessage (..), Connectivity)
+import Hydra.Network.Message (Connectivity (..), Message (..), NetworkMessage (..))
 import Hydra.OnChainId (OnChainId)
 import Hydra.Party (Party (vkey))
 import Hydra.Snapshot (ConfirmedSnapshot (..), Snapshot (..), SnapshotNumber, getSnapshot)
@@ -97,7 +97,11 @@ defaultTTL :: TTL
 defaultTTL = 5
 
 onConnectionEvent :: Connectivity -> Outcome tx
-onConnectionEvent _ = noop
+onConnectionEvent = \case
+  Connected{nodeId} ->
+    causes [ClientEffect (ServerOutput.PeerConnected nodeId)]
+  Disconnected{nodeId} ->
+    causes [ClientEffect (ServerOutput.PeerDisconnected nodeId)]
 
 -- * The Coordinated Head protocol