tubearchivist/tubearchivist
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist.git synced 2024-12-22 18:00:13 +00:00
Code Issues Projects Releases Wiki Activity

better att to queue, track score

Browse Source
This commit is contained in:
Simon 2024-05-14 18:06:27 +02:00
parent 86d157699a
commit 6ab70c7602
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tubearchivist/home/src/ta/ta_redis.py
View File

@ -127,11 +127,25 @@ class RedisQueue(RedisBase):
return False return False
def add(self, to_add: str) -> None:
"""add single item to queue"""
next_score = self._get_next_score()
self.conn.zadd(self.key, {to_add: next_score})
def add_list(self, to_add: list) -> None: def add_list(self, to_add: list) -> None:
"""add list to queue""" """add list to queue"""
mapping = {i: "+inf" for i in to_add} next_score = self._get_next_score()
mapping = {i[1]: next_score + i[0] for i in enumerate(to_add)}
self.conn.zadd(self.key, mapping) self.conn.zadd(self.key, mapping)
def _get_next_score(self) -> float:
"""get next score in queue to append"""
last = self.conn.zrange(self.key, -1, -1, withscores=True)
if not last:
return 0.0
return last[0][1] + 1
def get_next(self) -> str | bool: def get_next(self) -> str | bool:
"""return next element in the queue, if available""" """return next element in the queue, if available"""
result = self.conn.zpopmin(self.key) result = self.conn.zpopmin(self.key)