@@ -39,7 +39,6 @@ module Test.Ouroboros.Network.Diffusion.Node
3939import Control.Applicative (Alternative )
4040import Control.Concurrent.Class.MonadMVar (MonadMVar )
4141import Control.Concurrent.Class.MonadSTM.Strict
42- import Control.Monad ((>=>) )
4342import Control.Monad.Class.MonadAsync (MonadAsync (wait , withAsync ))
4443import Control.Monad.Class.MonadFork
4544import Control.Monad.Class.MonadSay
@@ -379,17 +378,16 @@ run blockGeneratorArgs ni na
379378 blockFetchPolicy nodeKernel =
380379 BlockFetchConsensusInterface {
381380 readCandidateChains = readTVar (nkClientChains nodeKernel)
382- >>= traverse (readTVar
383- >=> (return . toAnchoredFragment)),
384- readCurrentChain = readTVar (nkChainProducerState nodeKernel)
385- >>= (return . toAnchoredFragmentHeader . chainState),
381+ >>= fmap (fmap toAnchoredFragment) . traverse readTVar,
382+ readCurrentChain = toAnchoredFragmentHeader . chainState
383+ <$> readTVar (nkChainProducerState nodeKernel),
386384 readFetchMode = return $ PraosFetchMode FetchModeBulkSync ,
387385 readFetchedBlocks = flip Set. member <$> getBlockPointSet (nkChainDB nodeKernel),
388- readFetchedMaxSlotNo = Foldable. foldl' max NoMaxSlotNo .
389- map (maxSlotNoFromWithOrigin . pointSlot) .
390- Set. elems <$>
391- getBlockPointSet (nkChainDB nodeKernel),
392- mkAddFetchedBlock =
386+ readFetchedMaxSlotNo = Foldable. foldl' max NoMaxSlotNo
387+ . map (maxSlotNoFromWithOrigin . pointSlot)
388+ . Set. elems
389+ <$> getBlockPointSet (nkChainDB nodeKernel),
390+ mkAddFetchedBlock =
393391 pure $ \ _p b ->
394392 atomically (addBlock b (nkChainDB nodeKernel)),
395393
0 commit comments