p2p-governor: update inProgressPromoteWarm

The connection monitoring needs to update inProgressPromoteWarm, by removing peers that fail to be promoted.
IntersectMBO · Oct 19, 2020 · 944dc8e · 944dc8e
1 parent f1210d0
commit 944dc8e
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/ouroboros-network/src/Ouroboros/Network/PeerSelection/Governor/Monitor.hs b/ouroboros-network/src/Ouroboros/Network/PeerSelection/Governor/Monitor.hs
@@ -98,7 +98,8 @@ connections PeerSelectionActions{peerStateActions = PeerStateActions {monitorPee
               establishedPeers,
               establishedStatus,
               inProgressDemoteHot,
-              inProgressDemoteWarm
+              inProgressDemoteWarm,
+              inProgressPromoteWarm
             } =
     Guarded Nothing $ do
       establishedStatus' <- traverse monitorPeerConnection establishedPeers
@@ -130,7 +131,15 @@ connections PeerSelectionActions{peerStateActions = PeerStateActions {monitorPee
                                             . foldr
                                                 ((snd .) . KnownPeers.incrementFailCount)
                                                 (knownPeers st)
-                                            $ (Map.keysSet demotedToCold)
+                                            $ (Map.keysSet demotedToCold),
+
+                          -- When promoting a warm peer, it might happen
+                          -- that the connection will break (or one of the
+                          -- established protocols will error).  For that
+                          -- reason we need to adjust 'inProgressPromoteWarm'.
+                          inProgressPromoteWarm
+                                            = inProgressPromoteWarm
+                                                Set.\\ Map.keysSet demotions
                         }
       }
   where