From: Michael Tremer <michael.tremer@ipfire.org>
Date: Wed, 22 Jan 2025 14:46:52 +0000 (+0000)
Subject: ratelimiter: Fix cleanup
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=b5fe257a93278cead4ab31c1f244796a10eb24ef;p=pbs.git

ratelimiter: Fix cleanup

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>
---

diff --git a/src/buildservice/ratelimiter.py b/src/buildservice/ratelimiter.py
index c647744e..cf5a6844 100644
--- a/src/buildservice/ratelimiter.py
+++ b/src/buildservice/ratelimiter.py
@@ -60,7 +60,7 @@ class RateLimiter(base.Object):
 		"""
 		return RateLimiterRequest(self.backend, *args, **kwargs)
 
-	async def cleanup(backend):
+	async def cleanup(self):
 		"""
 			Called to cleanup the ratelimiter from expired entries
 		"""
@@ -69,12 +69,13 @@ class RateLimiter(base.Object):
 			ratelimiter
 			.delete()
 			.where(
-				ratelimiter.c.expired_at <= sqlalchemy.func.current_timestamp(),
+				ratelimiter.c.expires_at <= sqlalchemy.func.current_timestamp(),
 			)
 		)
 
 		# Run the query
-		await self.db.execute(stmt)
+		async with await self.db.transaction():
+			await self.db.execute(stmt)
 
 
 class RateLimiterRequest(base.Object):