mirror of
https://github.com/urbit/shrub.git
synced 2024-12-01 06:35:32 +03:00
Display a progress bar on startup when we have to replay the log.
Use the terminal-progress-bar package to display a progress bar during replay, when there are jobs that need to be replayed. Bumps the lts resolver so we get access to this package.
This commit is contained in:
parent
9b8ce8d9de
commit
860f59d46f
@ -19,6 +19,7 @@ import Data.Conduit
|
||||
import Data.Void
|
||||
import Noun
|
||||
import System.Process
|
||||
import System.ProgressBar
|
||||
import Vere.Pier.Types
|
||||
|
||||
import Data.Bits (setBit)
|
||||
@ -400,18 +401,38 @@ bootFromSeq serf (BootSeq ident nocks ovums) = do
|
||||
until it is caught up.
|
||||
-}
|
||||
replayJobs :: HasLogFunc e
|
||||
=> Serf e -> SerfState -> ConduitT Job Void (RIO e) SerfState
|
||||
replayJobs serf = go
|
||||
=> Serf e -> Int -> SerfState -> ConduitT Job Void (RIO e) SerfState
|
||||
replayJobs serf lastEv = go Nothing
|
||||
where
|
||||
go ss = await >>= maybe (pure ss) (lift . replayJob serf >=> go)
|
||||
go pb ss = do
|
||||
await >>= \case
|
||||
Nothing -> pure ss
|
||||
Just job -> do
|
||||
pb <- case pb of
|
||||
Nothing -> do
|
||||
-- We only construct the progress bar on the first time that we
|
||||
-- process an event so that we don't display an empty progress
|
||||
-- bar when the snapshot is caught up to the log.
|
||||
let toReplay = lastEv - (fromIntegral (ssNextEv ss))
|
||||
let style = defStyle { stylePostfix = exact }
|
||||
putStrLn $ pack ("Replaying events #" ++ (show (ssNextEv ss)) ++
|
||||
" to #" ++ (show lastEv))
|
||||
io $ newProgressBar style 10 (Progress 0 toReplay lastEv)
|
||||
Just pb -> do
|
||||
io $ incProgress pb 1
|
||||
pure pb
|
||||
played <- lift $ replayJob serf job
|
||||
go (Just pb) played
|
||||
|
||||
|
||||
replay :: HasLogFunc e => Serf e -> Log.EventLog -> RIO e SerfState
|
||||
replay serf log = do
|
||||
ss <- handshake serf (Log.identity log)
|
||||
|
||||
lastEv <- Log.lastEv log
|
||||
runConduit $ Log.streamEvents log (ssNextEv ss)
|
||||
.| toJobs (Log.identity log) (ssNextEv ss)
|
||||
.| replayJobs serf ss
|
||||
.| replayJobs serf (fromIntegral lastEv) ss
|
||||
|
||||
toJobs :: HasLogFunc e
|
||||
=> LogIdentity -> EventId -> ConduitT ByteString Job (RIO e) ()
|
||||
|
@ -80,6 +80,7 @@ dependencies:
|
||||
- tasty-th
|
||||
- template-haskell
|
||||
- terminfo
|
||||
- terminal-progress-bar
|
||||
- text
|
||||
- these
|
||||
- time
|
||||
|
@ -1,4 +1,4 @@
|
||||
resolver: lts-13.10
|
||||
resolver: lts-14.4
|
||||
|
||||
packages:
|
||||
- pkg/king
|
||||
|
Loading…
Reference in New Issue
Block a user