summaryrefslogtreecommitdiff
path: root/Assistant/Pushes.hs
blob: 6ac19405a94bb86d73e508313df6efecac4d7947 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
{- 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

import Assistant.Common
import Assistant.Types.Pushes
import Utility.TSet

import Control.Concurrent.STM
import Control.Concurrent.MSampleVar
import Data.Time.Clock
import qualified Data.Map as M

{- Blocks until there are failed pushes.
 - Returns Remotes whose pushes failed a given time duration or more ago.
 - (This may be an empty list.) -}
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
  where
	toorecent now time = now `diffUTCTime` time < duration

{- Modifies the map. -}
changeFailedPushMap :: (PushMap -> PushMap) -> Assistant ()
changeFailedPushMap a = do
	v <- getAssistant failedPushMap
	liftIO $ atomically $ store v . a . fromMaybe M.empty =<< tryTakeTMVar v
  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

notifyPush :: [UUID] -> Assistant ()
notifyPush us = flip putTSet us <<~ (pushNotifierSuccesses . pushNotifier)

waitPush :: Assistant [UUID]
waitPush = getTSet <<~ (pushNotifierSuccesses . pushNotifier)

notifyRestart :: Assistant ()
notifyRestart = flip writeSV () <<~ (pushNotifierWaiter . pushNotifier)

waitRestart :: Assistant ()
waitRestart = readSV <<~ (pushNotifierWaiter . pushNotifier)