Query over single connection in runQueryLeadershipSchedule

IntersectMBO · Jul 30, 2023 · f704350 · f704350
1 parent f6cb631
commit f704350
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/cardano-cli/src/Cardano/CLI/Run/Legacy/Query.hs b/cardano-cli/src/Cardano/CLI/Run/Legacy/Query.hs
@@ -1426,19 +1426,21 @@ runQueryLeadershipSchedule
                     $ nextEpochEligibleLeadershipSlots sbe shelleyGenesis
                       serCurrentEpochState ptclState poolid vrkSkey bpp
                       eInfo (tip, curentEpoch)
-
-              case mJsonOutputFile of
-                Nothing -> liftIO $ printLeadershipScheduleAsText schedule eInfo (SystemStart $ sgSystemStart shelleyGenesis)
-                Just (File jsonOutputFile) ->
-                  liftIO $ LBS.writeFile jsonOutputFile $
-                    printLeadershipScheduleAsJson schedule eInfo (SystemStart $ sgSystemStart shelleyGenesis)
+              writeSchedule mJsonOutputFile eInfo shelleyGenesis schedule
           mode ->
             pure $ do
               left . ShelleyQueryCmdUnsupportedMode $ AnyConsensusMode mode
     )
     & onLeft (left . ShelleyQueryCmdAcquireFailure)
     & onLeft left
   where
+    writeSchedule mOutFile eInfo shelleyGenesis schedule =
+      case mOutFile of
+        Nothing -> liftIO $ printLeadershipScheduleAsText schedule eInfo (SystemStart $ sgSystemStart shelleyGenesis)
+        Just (File jsonOutputFile) ->
+          liftIO $ LBS.writeFile jsonOutputFile $
+            printLeadershipScheduleAsJson schedule eInfo (SystemStart $ sgSystemStart shelleyGenesis)
+
     printLeadershipScheduleAsText
       :: Set SlotNo
       -> EpochInfo (Either Text)