sim: Fix size tracking for balanced preferred content

This commit is contained in:
Joey Hess 2024-09-23 12:28:18 -04:00
parent a6b8082119
commit 6cf9a101b8
No known key found for this signature in database
GPG key ID: DB12DB0FF05F8F38
5 changed files with 98 additions and 32 deletions

View file

@ -33,6 +33,7 @@ module Database.RepoSize (
removeStaleLiveSizeChanges,
recordedRepoOffsets,
liveRepoOffsets,
setSizeChanges,
) where
import Annex.Common
@ -311,6 +312,11 @@ setSizeChangeFor u sz =
(SizeChanges u sz)
[SizeChangesRollingtotal =. sz]
setSizeChanges :: RepoSizeHandle -> M.Map UUID FileSize -> IO ()
setSizeChanges (RepoSizeHandle (Just h) _) sizemap =
H.commitDb h $ forM_ (M.toList sizemap) $ uncurry setSizeChangeFor
setSizeChanges (RepoSizeHandle Nothing _) _ = noop
addRecentChange :: UUID -> Key -> SizeChange -> SqlPersistM ()
addRecentChange u k sc =
void $ upsertBy