From: Michael Tremer <michael.tremer@ipfire.org>
Date: Fri, 14 Oct 2022 14:19:52 +0000 (+0000)
Subject: messages: Add __len__ to Queue class
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=9d856910f37d38e2ed9d11192c651dab445bdccc;p=pbs.git

messages: Add __len__ to Queue class

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

diff --git a/src/buildservice/messages.py b/src/buildservice/messages.py
index 747385d3..084637b7 100644
--- a/src/buildservice/messages.py
+++ b/src/buildservice/messages.py
@@ -205,6 +205,19 @@ class Queue(base.Object):
 
 		return iter(messages)
 
+	def __len__(self):
+		res = self.db.get("""
+			SELECT
+				COUNT(*) AS length
+			FROM
+				messages
+			WHERE
+				sent_at IS NULL
+			""",
+		)
+
+		return res.length
+
 	def enqueue(self, message, priority=None):
 		"""
 			Enqueues a new message