6280af2901
Especially from borg, where the content identifier logs all end up being the same identical file! But also, for other imports, the location tracking logs can, in some cases, be identical files. Bonus optimisation: Avoid looking up (and parsing when set) GIT_ANNEX_VECTOR_CLOCK env var every time a log is written to. Although the lookup does happen at startup even when no log will be written now.
74 lines
2.1 KiB
Haskell
74 lines
2.1 KiB
Haskell
{- git-annex scheduled activities log
|
|
-
|
|
- Copyright 2013 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
module Logs.Schedule (
|
|
scheduleLog,
|
|
scheduleSet,
|
|
scheduleAdd,
|
|
scheduleRemove,
|
|
scheduleChange,
|
|
scheduleGet,
|
|
scheduleMap,
|
|
getLastRunTimes,
|
|
setLastRunTime,
|
|
) where
|
|
|
|
import qualified Data.Map as M
|
|
import qualified Data.Set as S
|
|
import Data.Time.LocalTime
|
|
import qualified Data.Attoparsec.ByteString.Lazy as A
|
|
import Data.ByteString.Builder
|
|
|
|
import Annex.Common
|
|
import Types.ScheduledActivity
|
|
import qualified Annex.Branch
|
|
import Logs
|
|
import Logs.UUIDBased
|
|
import Logs.File
|
|
|
|
scheduleSet :: UUID -> [ScheduledActivity] -> Annex ()
|
|
scheduleSet uuid@(UUID _) activities = do
|
|
c <- currentVectorClock
|
|
Annex.Branch.change scheduleLog $
|
|
buildLogOld byteString
|
|
. changeLog c uuid (encodeBS val)
|
|
. parseLogOld A.takeByteString
|
|
where
|
|
val = fromScheduledActivities activities
|
|
scheduleSet NoUUID _ = error "unknown UUID; cannot modify"
|
|
|
|
scheduleMap :: Annex (M.Map UUID [ScheduledActivity])
|
|
scheduleMap = simpleMap . parseLogOld parser <$> Annex.Branch.get scheduleLog
|
|
where
|
|
parser = either fail pure . parseScheduledActivities . decodeBS
|
|
=<< A.takeByteString
|
|
|
|
scheduleGet :: UUID -> Annex (S.Set ScheduledActivity)
|
|
scheduleGet u = do
|
|
m <- scheduleMap
|
|
return $ maybe S.empty S.fromList (M.lookup u m)
|
|
|
|
scheduleRemove :: UUID -> ScheduledActivity -> Annex ()
|
|
scheduleRemove u activity = scheduleChange u $ S.delete activity
|
|
|
|
scheduleAdd :: UUID -> ScheduledActivity -> Annex ()
|
|
scheduleAdd u activity = scheduleChange u $ S.insert activity
|
|
|
|
scheduleChange :: UUID -> (S.Set ScheduledActivity -> S.Set ScheduledActivity) -> Annex ()
|
|
scheduleChange u a = scheduleSet u . S.toList . a =<< scheduleGet u
|
|
|
|
getLastRunTimes :: Annex (M.Map ScheduledActivity LocalTime)
|
|
getLastRunTimes = do
|
|
f <- fromRawFilePath <$> fromRepo gitAnnexScheduleState
|
|
liftIO $ fromMaybe M.empty
|
|
<$> catchDefaultIO Nothing (readish <$> readFile f)
|
|
|
|
setLastRunTime :: ScheduledActivity -> LocalTime -> Annex ()
|
|
setLastRunTime activity lastrun = do
|
|
f <- fromRepo gitAnnexScheduleState
|
|
writeLogFile f . show . M.insert activity lastrun
|
|
=<< getLastRunTimes
|