display scanning message whenever reconcileStaged has enough files to chew on

Clear visible progress bar first.

Removed showSideActionAfter because it can't be used in reconcileStaged
(import loop). Instead, it counts the number of files it
processes and displays it after it's seen a sufficient to know it's
taking a while.

Sponsored-by: Dartmouth College's Datalad project
This commit is contained in:
Joey Hess 2021-06-08 12:48:30 -04:00
parent ecbaa52571
commit 7b6deb1109
No known key found for this signature in database
GPG key ID: DB12DB0FF05F8F38
9 changed files with 53 additions and 36 deletions

View file

@ -100,23 +100,3 @@ mergeState st = do
uncurry addCleanupAction uncurry addCleanupAction
Annex.Queue.mergeFrom st' Annex.Queue.mergeFrom st'
changeState $ \s -> s { errcounter = errcounter s + errcounter st' } changeState $ \s -> s { errcounter = errcounter s + errcounter st' }
{- Display a message, only when the action runs for a long enough
- amount of time.
-
- The action should not display any other messages, progress, etc;
- if it did there could be some scrambling of the display since the
- message display could happen at the same time as other output,
- or after it.
-}
showSideActionAfter :: Microseconds -> String -> Annex a -> Annex a
showSideActionAfter t m a = do
waiter <- liftIO $ async $ unboundDelay t
let display = liftIO (waitCatch waiter) >>= \case
Left _ -> return ()
Right _ -> showSideAction m
displayer <- liftIO . async =<< forkState display
let cleanup = do
liftIO $ cancel waiter
join (liftIO (wait displayer))
a `finally` cleanup

View file

@ -15,8 +15,6 @@ import Annex.Content
import Annex.ReplaceFile import Annex.ReplaceFile
import Annex.CurrentBranch import Annex.CurrentBranch
import Annex.InodeSentinal import Annex.InodeSentinal
import Annex.Concurrent
import Utility.ThreadScheduler
import Utility.InodeCache import Utility.InodeCache
import Git.FilePath import Git.FilePath
import Git.CatFile import Git.CatFile
@ -81,7 +79,7 @@ ifAnnexed file yes no = maybe no yes =<< lookupKey file
- as-is. - as-is.
-} -}
scanAnnexedFiles :: Bool -> Annex () scanAnnexedFiles :: Bool -> Annex ()
scanAnnexedFiles initscan = showSideActionAfter oneSecond "scanning for annexed files" $ do scanAnnexedFiles initscan = do
-- This gets the keys database populated with all annexed files, -- This gets the keys database populated with all annexed files,
-- by running Database.Keys.reconcileStaged. -- by running Database.Keys.reconcileStaged.
Database.Keys.runWriter (const noop) Database.Keys.runWriter (const noop)

View file

@ -7,6 +7,7 @@
{-# LANGUAGE ScopedTypeVariables #-} {-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE OverloadedStrings #-} {-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE BangPatterns #-}
module Database.Keys ( module Database.Keys (
DbHandle, DbHandle,
@ -403,7 +404,7 @@ reconcileStaged qh = do
proct <- liftIO $ async $ proct <- liftIO $ async $
procthread mdreader catfeeder procthread mdreader catfeeder
`finally` void catcloser `finally` void catcloser
dbchanged <- dbwriter False catreader dbchanged <- dbwriter False largediff catreader
-- Flush database changes now -- Flush database changes now
-- so other processes can see them. -- so other processes can see them.
when dbchanged $ when dbchanged $
@ -420,8 +421,27 @@ reconcileStaged qh = do
Just _ -> procthread mdreader catfeeder Just _ -> procthread mdreader catfeeder
Nothing -> return () Nothing -> return ()
dbwriter dbchanged catreader = liftIO catreader >>= \case dbwriter dbchanged n catreader = liftIO catreader >>= \case
Just (ka, content) -> do Just (ka, content) -> do
changed <- ka (parseLinkTargetOrPointerLazy =<< content) changed <- ka (parseLinkTargetOrPointerLazy =<< content)
dbwriter (dbchanged || changed) catreader !n' <- countdownToMessage n
dbwriter (dbchanged || changed) n' catreader
Nothing -> return dbchanged Nothing -> return dbchanged
-- When the diff is large, the scan can take a while,
-- so let the user know what's going on.
countdownToMessage n
| n < 1 = return 0
| n == 1 = do
showSideAction "scanning for annexed files"
return 0
| otherwise = return (pred n)
-- How large is large? Too large and there will be a long
-- delay before the message is shown; too short and the message
-- will clutter things up unncessarily. It's uncommon for 1000
-- files to change in the index, and processing that many files
-- takes less than half a second, so that seems about right.
largediff :: Int
largediff = 1000

View file

@ -124,12 +124,14 @@ showSideAction m = Annex.getState Annex.output >>= go
where where
go st go st
| sideActionBlock st == StartBlock = do | sideActionBlock st == StartBlock = do
p go' st
let st' = st { sideActionBlock = InBlock } let st' = st { sideActionBlock = InBlock }
Annex.changeState $ \s -> s { Annex.output = st' } Annex.changeState $ \s -> s { Annex.output = st' }
| sideActionBlock st == InBlock = return () | sideActionBlock st == InBlock = return ()
| otherwise = p | otherwise = go' st
p = outputMessage JSON.none $ encodeBS' $ "(" ++ m ++ "...)\n" go' st = do
liftIO $ clearProgressMeter st
outputMessage JSON.none $ encodeBS' $ "(" ++ m ++ "...)\n"
showStoringStateAction :: Annex () showStoringStateAction :: Annex ()
showStoringStateAction = showSideAction "recording state in git" showStoringStateAction = showSideAction "recording state in git"

View file

@ -11,6 +11,7 @@
module Messages.Progress where module Messages.Progress where
import Common import Common
import qualified Annex
import Messages import Messages
import Utility.Metered import Utility.Metered
import Types import Types
@ -66,7 +67,7 @@ instance MeterSize KeySizer where
{- Shows a progress meter while performing an action. {- Shows a progress meter while performing an action.
- The action is passed the meter and a callback to use to update the meter. - The action is passed the meter and a callback to use to update the meter.
--} -}
metered metered
:: MeterSize sizer :: MeterSize sizer
=> Maybe MeterUpdate => Maybe MeterUpdate
@ -75,28 +76,38 @@ metered
-> Annex a -> Annex a
metered othermeter sizer a = withMessageState $ \st -> do metered othermeter sizer a = withMessageState $ \st -> do
sz <- getMeterSize sizer sz <- getMeterSize sizer
metered' st othermeter sz showOutput a metered' st setclear othermeter sz showOutput a
where
setclear c = Annex.changeState $ \st -> st
{ Annex.output = (Annex.output st) { clearProgressMeter = c } }
metered' metered'
:: (Monad m, MonadIO m, MonadMask m) :: (Monad m, MonadIO m, MonadMask m)
=> MessageState => MessageState
-> (IO () -> m ())
-- ^ This should set clearProgressMeter when progress meters
-- are being displayed; not needed when outputType is not
-- NormalOutput.
-> Maybe MeterUpdate -> Maybe MeterUpdate
-> Maybe TotalSize -> Maybe TotalSize
-> m () -> m ()
-- ^ this should run showOutput -- ^ this should run showOutput
-> (Meter -> MeterUpdate -> m a) -> (Meter -> MeterUpdate -> m a)
-> m a -> m a
metered' st othermeter msize showoutput a = go st metered' st setclear othermeter msize showoutput a = go st
where where
go (MessageState { outputType = QuietOutput }) = nometer go (MessageState { outputType = QuietOutput }) = nometer
go (MessageState { outputType = NormalOutput, concurrentOutputEnabled = False }) = do go (MessageState { outputType = NormalOutput, concurrentOutputEnabled = False }) = do
showoutput showoutput
meter <- liftIO $ mkMeter msize $ meter <- liftIO $ mkMeter msize $
displayMeterHandle stdout bandwidthMeter displayMeterHandle stdout bandwidthMeter
let clear = clearMeterHandle meter stdout
setclear clear
m <- liftIO $ rateLimitMeterUpdate consoleratelimit meter $ m <- liftIO $ rateLimitMeterUpdate consoleratelimit meter $
updateMeter meter updateMeter meter
r <- a meter (combinemeter m) r <- a meter (combinemeter m)
liftIO $ clearMeterHandle meter stdout setclear noop
liftIO clear
return r return r
go (MessageState { outputType = NormalOutput, concurrentOutputEnabled = True }) = go (MessageState { outputType = NormalOutput, concurrentOutputEnabled = True }) =
withProgressRegion st $ \r -> do withProgressRegion st $ \r -> do

View file

@ -65,9 +65,10 @@ relaySerializedOutput getso sendsor meterreport runannex = go Nothing
loop st loop st
Left BeginProgressMeter -> do Left BeginProgressMeter -> do
ost <- runannex (Annex.getState Annex.output) ost <- runannex (Annex.getState Annex.output)
let setclear = const noop
-- Display a progress meter while running, until -- Display a progress meter while running, until
-- the meter ends or a final value is returned. -- the meter ends or a final value is returned.
metered' ost Nothing Nothing (runannex showOutput) metered' ost setclear Nothing Nothing (runannex showOutput)
(\meter meterupdate -> loop (Just (meter, meterupdate))) (\meter meterupdate -> loop (Just (meter, meterupdate)))
>>= \case >>= \case
Right r -> return (Right r) Right r -> return (Right r)

View file

@ -47,6 +47,7 @@ data MessageState = MessageState
, consoleRegionErrFlag :: Bool , consoleRegionErrFlag :: Bool
, jsonBuffer :: Maybe Aeson.Object , jsonBuffer :: Maybe Aeson.Object
, promptLock :: MVar () -- left full when not prompting , promptLock :: MVar () -- left full when not prompting
, clearProgressMeter :: IO ()
} }
newMessageState :: IO MessageState newMessageState :: IO MessageState
@ -60,6 +61,7 @@ newMessageState = do
, consoleRegionErrFlag = False , consoleRegionErrFlag = False
, jsonBuffer = Nothing , jsonBuffer = Nothing
, promptLock = promptlock , promptLock = promptlock
, clearProgressMeter = return ()
} }
-- | When communicating with a child process over a pipe while it is -- | When communicating with a child process over a pipe while it is

View file

@ -425,7 +425,8 @@ displayMeterHandle h rendermeter v msize old new = do
hPutStr h ('\r':s ++ padding) hPutStr h ('\r':s ++ padding)
hFlush h hFlush h
-- | Clear meter displayed by displayMeterHandle. -- | Clear meter displayed by displayMeterHandle. May be called before
-- outputting something else, followed by more calls to displayMeterHandle.
clearMeterHandle :: Meter -> Handle -> IO () clearMeterHandle :: Meter -> Handle -> IO ()
clearMeterHandle (Meter _ _ v _) h = do clearMeterHandle (Meter _ _ v _) h = do
olds <- readMVar v olds <- readMVar v

View file

@ -26,6 +26,8 @@ with the progress display in view:
(scanning annexed files...) (scanning annexed files...)
add newfile ok add newfile ok
> [[done]] --[[Joey]]
It might also be possible to make reconcileStaged run a less expensive It might also be possible to make reconcileStaged run a less expensive
scan in this case, eg the scan it did before scan in this case, eg the scan it did before
[[!commit 428c91606b434512d1986622e751c795edf4df44]]. In this case, it [[!commit 428c91606b434512d1986622e751c795edf4df44]]. In this case, it