5afbea25e7
In calcRepoSizes and also git-annex info, when a key was in the journal, it was passed to the callback twice, so the calculated size was wrong.
37 lines
1.1 KiB
Haskell
37 lines
1.1 KiB
Haskell
{- git-annex repo sizes
|
|
-
|
|
- Copyright 2024 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
module Annex.RepoSize where
|
|
|
|
import Annex.Common
|
|
import Annex.Branch (UnmergedBranches(..))
|
|
import Types.RepoSize
|
|
import Logs.Location
|
|
import Logs.UUID
|
|
|
|
import qualified Data.Map.Strict as M
|
|
|
|
{- Sum up the sizes of all keys in all repositories, from the information
|
|
- in the git-annex branch. New keys that only appear in the journal are
|
|
- not included. Can be slow.
|
|
-
|
|
- The map includes the UUIDs of all known repositories, including
|
|
- repositories that are empty.
|
|
-}
|
|
calcRepoSizes :: Annex (M.Map UUID RepoSize)
|
|
calcRepoSizes = do
|
|
knownuuids <- M.keys <$> uuidDescMap
|
|
let startmap = M.fromList $ map (\u -> (u, RepoSize 0)) knownuuids
|
|
overLocationLogs True startmap accum >>= \case
|
|
UnmergedBranches m -> return m
|
|
NoUnmergedBranches m -> return m
|
|
where
|
|
addksz ksz (Just (RepoSize sz)) = Just $ RepoSize $ sz + ksz
|
|
addksz ksz Nothing = Just $ RepoSize ksz
|
|
accum k locs m = return $
|
|
let sz = fromMaybe 0 $ fromKey keySize k
|
|
in foldl' (flip $ M.alter $ addksz sz) m locs
|