Don't run wait action multiple times in chain seek client

input-output-hk · Aug 8, 2022 · 4d9611a · 4d9611a
1 parent 0737318
commit 4d9611a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/marlowe-protocols/src/Network/Protocol/ChainSeek/Client.hs b/marlowe-protocols/src/Network/Protocol/ChainSeek/Client.hs
@@ -184,9 +184,9 @@ chainSeekClientPeer initialPoint (ChainSeekClient mclient) =
     -> m (ClientStNext query err result point tip m a)
     -> Peer (ChainSeek query point tip) 'AsClient ('StNext err result 'StMustReply) m a
   peerWait pos query mnext = Effect do
-    ClientStNext{..} <- mnext
+    next@ClientStNext{..} <- mnext
     pure $ Await (ServerAgency (TokNext query TokMustReply)) \case
       MsgRejectQuery err tip         -> peerIdle pos $ recvMsgQueryRejected err tip
       MsgRollForward result pos' tip -> peerIdle pos' $ recvMsgRollForward result pos' tip
       MsgRollBackward pos' tip       -> peerIdle pos' $ recvMsgRollBackward pos' tip
-      MsgPing                        -> Yield (ClientAgency TokPing) MsgPong $ peerWait pos query mnext
+      MsgPing                        -> Yield (ClientAgency TokPing) MsgPong $ peerWait pos query $ pure next