From bba23e7cc934e01d8bb8cff1998322950e55f537 Mon Sep 17 00:00:00 2001 From: Joey Hess Date: Thu, 15 Aug 2024 12:31:27 -0400 Subject: [PATCH] do not need a db queue This database is read once and written at most once per run. --- Database/RepoSize.hs | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/Database/RepoSize.hs b/Database/RepoSize.hs index ff70affbfc..40599fddf8 100644 --- a/Database/RepoSize.hs +++ b/Database/RepoSize.hs @@ -30,7 +30,7 @@ import Annex.Common import Annex.LockFile import Types.RepoSize import Git.Types -import qualified Database.Queue as H +import qualified Database.Handle as H import Database.Init import Database.Utility import Database.Types @@ -41,7 +41,7 @@ import Database.Persist.TH import qualified System.FilePath.ByteString as P import qualified Data.Map as M -newtype RepoSizeHandle = RepoSizeHandle (Maybe H.DbQueue) +newtype RepoSizeHandle = RepoSizeHandle (Maybe H.DbHandle) share [mkPersist sqlSettings, mkMigrate "migrateRepoSizes"] [persistLowerCase| -- Corresponds to location log information from the git-annex branch. @@ -72,7 +72,7 @@ openDb = do unlessM (liftIO $ R.doesPathExist db) $ do initDb db $ void $ runMigrationSilent migrateRepoSizes - h <- liftIO $ H.openDbQueue db "repo_sizes" + h <- liftIO $ H.openDb db "repo_sizes" return $ RepoSizeHandle (Just h) where -- If permissions don't allow opening the database, @@ -83,11 +83,11 @@ openDb = do permerr _e = return (RepoSizeHandle Nothing) closeDb :: RepoSizeHandle -> Annex () -closeDb (RepoSizeHandle (Just h)) = liftIO $ H.closeDbQueue h +closeDb (RepoSizeHandle (Just h)) = liftIO $ H.closeDb h closeDb (RepoSizeHandle Nothing) = noop getRepoSizes :: RepoSizeHandle -> IO (M.Map UUID RepoSize, Maybe Sha) -getRepoSizes (RepoSizeHandle (Just h)) = H.queryDbQueue h $ do +getRepoSizes (RepoSizeHandle (Just h)) = H.queryDb h $ do sizemap <- M.fromList . map conv <$> getRepoSizes' annexbranchsha <- getAnnexBranchCommit return (sizemap, annexbranchsha) @@ -119,7 +119,7 @@ getAnnexBranchCommit = do -} setRepoSizes :: RepoSizeHandle -> M.Map UUID RepoSize -> Sha -> IO () setRepoSizes (RepoSizeHandle (Just h)) sizemap branchcommitsha = - H.queueDb h commitimmediately $ do + H.commitDb h $ do l <- getRepoSizes' forM_ (map entityVal l) $ \(RepoSizes u _) -> unless (M.member u sizemap) $ @@ -127,8 +127,6 @@ setRepoSizes (RepoSizeHandle (Just h)) sizemap branchcommitsha = forM_ (M.toList sizemap) $ uncurry setRepoSize recordAnnexBranchCommit branchcommitsha - where - commitimmediately _ _ = pure True setRepoSizes (RepoSizeHandle Nothing) _ _ = noop setRepoSize :: UUID -> RepoSize -> SqlPersistM ()