Simplify chainDiffs computation in chainSelectionForBlock

IntersectMBO · May 3, 2024 · 1c0ab60 · 1c0ab60
1 parent 3a7e4d4
commit 1c0ab60
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/...os-consensus/src/ouroboros-consensus/Ouroboros/Consensus/Storage/ChainDB/Impl/ChainSel.hs b/...os-consensus/src/ouroboros-consensus/Ouroboros/Consensus/Storage/ChainDB/Impl/ChainSel.hs
@@ -671,10 +671,9 @@ chainSelectionForBlock cdb@CDB{..} blockCache hdr punish = electric $ do
               return $ AF.fromOldestFirst curHead (hdr : hdrs)
 
         let chainDiffs = NE.nonEmpty
-              $ NE.filter ( preferAnchoredCandidate (bcfg chainSelEnv) curChain
-                          . Diff.getSuffix
-                          )
-              $ fmap Diff.extend candidates
+              $ map Diff.extend
+              $ NE.filter (preferAnchoredCandidate (bcfg chainSelEnv) curChain)
+                candidates
         -- All candidates are longer than the current chain, so they will be
         -- preferred over it, /unless/ the block we just added is an EBB,
         -- which has the same 'BlockNo' as the block before it, so when