]> git.ipfire.org Git - pbs.git/commitdiff
ratelimiter: Fix cleanup
authorMichael Tremer <michael.tremer@ipfire.org>
Wed, 22 Jan 2025 14:46:52 +0000 (14:46 +0000)
committerMichael Tremer <michael.tremer@ipfire.org>
Wed, 22 Jan 2025 14:46:52 +0000 (14:46 +0000)
Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>
src/buildservice/ratelimiter.py

index c647744ef7616cd9fd962fa96c388d48c6955636..cf5a68445573f3edcb84d23c316f31a4373663ae 100644 (file)
@@ -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):