fa62c98910
This eliminates the distinction between decodeBS and decodeBS', encodeBS and encodeBS', etc. The old implementation truncated at NUL, and the primed versions had to do extra work to avoid that problem. The new implementation does not truncate at NUL, and is also a lot faster. (Benchmarked at 2x faster for decodeBS and 3x for encodeBS; more for the primed versions.) Note that filepath-bytestring 1.4.2.1.8 contains the same optimisation, and upgrading to it will speed up to/fromRawFilePath. AFAIK, nothing relied on the old behavior of truncating at NUL. Some code used the faster versions in places where I was sure there would not be a NUL. So this change is unlikely to break anything. Also, moved s2w8 and w82s out of the module, as they do not involve filesystem encoding really. Sponsored-by: Shae Erisson on Patreon
63 lines
1.7 KiB
Haskell
63 lines
1.7 KiB
Haskell
{- git-annex fsck results log files
|
|
-
|
|
- Copyright 2013 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
module Logs.FsckResults (
|
|
writeFsckResults,
|
|
readFsckResults,
|
|
clearFsckResults,
|
|
) where
|
|
|
|
import Annex.Common
|
|
import Git.Fsck
|
|
import Git.Types
|
|
import Logs.File
|
|
import qualified Utility.RawFilePath as R
|
|
|
|
import qualified Data.Set as S
|
|
|
|
writeFsckResults :: UUID -> FsckResults -> Annex ()
|
|
writeFsckResults u fsckresults = do
|
|
logfile <- fromRepo $ gitAnnexFsckResultsLog u
|
|
case serializeFsckResults fsckresults of
|
|
Just s -> store s logfile
|
|
Nothing -> liftIO $
|
|
removeWhenExistsWith R.removeLink logfile
|
|
where
|
|
store s logfile = writeLogFile logfile s
|
|
|
|
serializeFsckResults :: FsckResults -> Maybe String
|
|
serializeFsckResults fsckresults = case fsckresults of
|
|
FsckFailed -> Just (serialize S.empty False)
|
|
FsckFoundMissing s t
|
|
| S.null s -> Nothing
|
|
| otherwise -> Just (serialize s t)
|
|
where
|
|
serialize s t =
|
|
let ls = map fromRef (S.toList s)
|
|
in if t
|
|
then unlines ("truncated":ls)
|
|
else unlines ls
|
|
|
|
readFsckResults :: UUID -> Annex FsckResults
|
|
readFsckResults u = do
|
|
logfile <- fromRepo $ gitAnnexFsckResultsLog u
|
|
liftIO $ catchDefaultIO (FsckFoundMissing S.empty False) $
|
|
deserializeFsckResults <$> readFile (fromRawFilePath logfile)
|
|
|
|
deserializeFsckResults :: String -> FsckResults
|
|
deserializeFsckResults = deserialize . lines
|
|
where
|
|
deserialize ("truncated":ls) = deserialize' ls True
|
|
deserialize ls = deserialize' ls False
|
|
deserialize' ls t =
|
|
let s = S.fromList $ map (Ref . encodeBS) ls
|
|
in if S.null s then FsckFailed else FsckFoundMissing s t
|
|
|
|
clearFsckResults :: UUID -> Annex ()
|
|
clearFsckResults = liftIO . removeWhenExistsWith R.removeLink
|
|
<=< fromRepo . gitAnnexFsckResultsLog
|
|
|