Query over single connection in runQueryLeadershipSchedule

IntersectMBO · Jul 14, 2023 · d803dd5 · d803dd5
1 parent 0b98bad
commit d803dd5
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/cardano-cli/src/Cardano/CLI/Shelley/Run/Query.hs b/cardano-cli/src/Cardano/CLI/Shelley/Run/Query.hs
@@ -1394,12 +1394,11 @@ runQueryLeadershipSchedule
 
             case whichSchedule of
               CurrentEpoch -> do
-                pure $ do
-                  serCurrentEpochState <- lift (executeLocalStateQueryExpr localNodeConnInfo Nothing $ queryPoolDistribution eInMode sbe (Just (Set.singleton poolid)))
-                    & onLeft (left . ShelleyQueryCmdAcquireFailure)
-                    & onLeft (left . ShelleyQueryCmdUnsupportedNtcVersion)
-                    & onLeft (left . ShelleyQueryCmdLocalStateQueryError . EraMismatchError)
+                serCurrentEpochState <- lift (queryPoolDistribution eInMode sbe (Just (Set.singleton poolid)))
+                  & onLeft (left . ShelleyQueryCmdUnsupportedNtcVersion)
+                  & onLeft (left . ShelleyQueryCmdLocalStateQueryError . EraMismatchError)
 
+                pure $ do
                   schedule <- firstExceptT ShelleyQueryCmdLeaderShipError $ hoistEither
                     $ eligibleLeaderSlotsConstaints sbe
                     $ currentEpochEligibleLeadershipSlots
@@ -1416,12 +1415,11 @@ runQueryLeadershipSchedule
                   writeSchedule mJsonOutputFile eInfo shelleyGenesis schedule
 
               NextEpoch -> do
-                pure $ do
-                  serCurrentEpochState <- lift (executeLocalStateQueryExpr localNodeConnInfo Nothing $ queryCurrentEpochState eInMode sbe)
-                    & onLeft (left . ShelleyQueryCmdAcquireFailure)
-                    & onLeft (left . ShelleyQueryCmdUnsupportedNtcVersion)
-                    & onLeft (left . ShelleyQueryCmdLocalStateQueryError . EraMismatchError)
+                serCurrentEpochState <- lift (queryCurrentEpochState eInMode sbe)
+                  & onLeft (left . ShelleyQueryCmdUnsupportedNtcVersion)
+                  & onLeft (left . ShelleyQueryCmdLocalStateQueryError . EraMismatchError)
 
+                pure $ do
                   tip <- liftIO $ getLocalChainTip localNodeConnInfo
 
                   schedule <- firstExceptT ShelleyQueryCmdLeaderShipError $ hoistEither