peer-selection: renamed peerStateToCounters

Also export it from `Governor` module, it is useful in `cardano-node`.
IntersectMBO · Apr 16, 2024 · f7277cb · f7277cb
1 parent 3e4190a
commit f7277cb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/ouroboros-network/src/Ouroboros/Network/PeerSelection/Governor.hs b/ouroboros-network/src/Ouroboros/Network/PeerSelection/Governor.hs
@@ -30,6 +30,7 @@ module Ouroboros.Network.PeerSelection.Governor
   , PublicPeerSelectionState (..)
   , makePublicPeerSelectionStateVar
   , PeerSelectionCounters (..)
+  , peerSelectionStateToCounters
   , emptyPeerSelectionCounters
   , nullPeerSelectionTargets
   , emptyPeerSelectionState
@@ -557,7 +558,7 @@ peerSelectionGovernorLoop tracer
       now <- getMonotonicTime
       let Decision { decisionTrace, decisionJobs, decisionState } =
             timedDecision now
-          !newCounters = peerStateToCounters decisionState
+          !newCounters = peerSelectionStateToCounters decisionState
 
       atomically $ do
         -- Update counters

diff --git a/ouroboros-network/src/Ouroboros/Network/PeerSelection/Governor/Types.hs b/ouroboros-network/src/Ouroboros/Network/PeerSelection/Governor/Types.hs
@@ -39,7 +39,7 @@ module Ouroboros.Network.PeerSelection.Governor.Types
   , Completion (..)
   , PeerSelectionCounters (.., PeerSelectionCountersHWC, numberOfColdPeers, numberOfWarmPeers, numberOfHotPeers, numberOfColdBigLedgerPeers, numberOfWarmBigLedgerPeers, numberOfHotBigLedgerPeers, numberOfColdLocalRootPeers, numberOfWarmLocalRootPeers, numberOfHotLocalRootPeers, localRootsHWC)
   , emptyPeerSelectionCounters
-  , peerStateToCounters
+  , peerSelectionStateToCounters
     -- * Peer Sharing Auxiliary data type
   , PeerSharingResult (..)
     -- * Traces
@@ -752,16 +752,20 @@ peerSelectionCountersHWC PeerSelectionCounters {..} =
     }
 
 
-peerStateToCounters :: Ord peeraddr => PeerSelectionState peeraddr peerconn -> PeerSelectionCounters
-peerStateToCounters PeerSelectionState { knownPeers,
-                                         establishedPeers,
-                                         activePeers,
-                                         publicRootPeers,
-                                         localRootPeers,
-                                         inProgressPromoteCold,
-                                         inProgressPromoteWarm,
-                                         inProgressDemoteWarm,
-                                         inProgressDemoteHot } =
+peerSelectionStateToCounters :: Ord peeraddr => PeerSelectionState peeraddr peerconn -> PeerSelectionCounters
+peerSelectionStateToCounters
+    PeerSelectionState {
+        knownPeers,
+        establishedPeers,
+        activePeers,
+        publicRootPeers,
+        localRootPeers,
+        inProgressPromoteCold,
+        inProgressPromoteWarm,
+        inProgressDemoteWarm,
+        inProgressDemoteHot
+      }
+    =
     PeerSelectionCounters {
       numberOfRootPeers                          = Set.size rootPeersSet,