run vacuum task in the background

2 files changed, 22 insertions, 0 deletions

diff --git a/pypaste/server/__init__.py b/pypaste/server/__init__.py
index da5e538..6f79b65 100644
--- a/pypaste/server/__init__.py
+++ b/pypaste/server/__init__.py
@@ -262,6 +262,26 @@ class App:
 
         return web.HTTPOk(text=url)
 
+    async def background_vacuum(self) -> None:
+        while True:
+            try:
+                before = await self.storage.storage_use()
+                await self.storage.vacuum(self.config.storage_max_bytes)
+                after = await self.storage.storage_use()
+
+                match [before, after]:
+                    case [int(before), int(after)]:
+                        log_info(f"vacuumed {before-after} bytes")
+                    case [int(before), None]:
+                        log_info(f"vacuumed {before} bytes")
+                    case [None, None]:
+                        pass
+
+            except Exception as e:
+                log_error(f"failed to vacuum: {e}")
+            finally:
+                await asyncio.sleep(60 * 5)
+
 
 def test_humanize_dehumanize_roundtrip() -> None:
     key = keygen(6)
diff --git a/pypaste/server/__main__.py b/pypaste/server/__main__.py
index be83ba2..d1dcdf4 100644
--- a/pypaste/server/__main__.py
+++ b/pypaste/server/__main__.py
@@ -155,6 +155,8 @@ async def main() -> int:
 
     log_info("starting pypaste")
 
+    asyncio.create_task(pypaste.background_vacuum())
+
     try:
         await asyncio.Event().wait()
     except asyncio.exceptions.CancelledError: