git-annex/Assistant/Pushes.hs

41 lines
1.2 KiB
Haskell
Raw Normal View History

2012-06-25 20:38:12 +00:00
{- git-annex assistant push tracking
-
- Copyright 2012 Joey Hess <joey@kitenet.net>
-
- Licensed under the GNU GPL version 3 or higher.
-}
module Assistant.Pushes where
2012-10-29 21:52:43 +00:00
import Assistant.Common
import Assistant.Types.Pushes
2012-06-25 20:38:12 +00:00
import Control.Concurrent.STM
2012-06-25 20:38:12 +00:00
import Data.Time.Clock
import qualified Data.Map as M
2012-06-25 20:38:12 +00:00
{- Blocks until there are failed pushes.
- Returns Remotes whose pushes failed a given time duration or more ago.
- (This may be an empty list.) -}
2012-10-29 21:52:43 +00:00
getFailedPushesBefore :: NominalDiffTime -> Assistant [Remote]
getFailedPushesBefore duration = do
v <- getAssistant failedPushMap
liftIO $ do
m <- atomically $ readTMVar v
now <- getCurrentTime
return $ M.keys $ M.filter (not . toorecent now) m
2012-10-31 06:34:03 +00:00
where
toorecent now time = now `diffUTCTime` time < duration
{- Modifies the map. -}
2012-10-29 21:52:43 +00:00
changeFailedPushMap :: (PushMap -> PushMap) -> Assistant ()
changeFailedPushMap a = do
v <- getAssistant failedPushMap
liftIO $ atomically $ store v . a . fromMaybe M.empty =<< tryTakeTMVar v
2012-10-31 06:34:03 +00:00
where
{- tryTakeTMVar empties the TMVar; refill it only if
- the modified map is not itself empty -}
store v m
| m == M.empty = noop
| otherwise = putTMVar v $! m