61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
import json
|
|
|
|
import redis.asyncio as redis
|
|
|
|
|
|
class RedisClient:
|
|
SET_NAME = "queue"
|
|
TASKS_NAME = "tasks_working"
|
|
TASKS_DONE_NAME = "tasks_done"
|
|
|
|
def __init__(self):
|
|
self.connection = redis.Redis(host="localhost", port=6379, db=0)
|
|
|
|
async def _set_task(self, task: dict, queue_name) -> int:
|
|
async with self.connection as connection:
|
|
res = await connection.sadd(f'{queue_name}:1', json.dumps(task, indent=4).encode('utf-8'))
|
|
return res
|
|
|
|
async def _del_task(self, task: dict, queue_name) -> int:
|
|
async with self.connection as connection:
|
|
res = await connection.srem(f'{queue_name}:1', json.dumps(task, indent=4).encode('utf-8'))
|
|
return res
|
|
|
|
async def set_task_to_queue(self, task: dict) -> int:
|
|
async with self.connection as connection:
|
|
res = await connection.sadd(self.SET_NAME, json.dumps(task, indent=4).encode('utf-8'))
|
|
return res
|
|
|
|
async def get_queue(self) -> set:
|
|
async with self.connection as connection:
|
|
res = await connection.smembers(self.SET_NAME + f":1")
|
|
return res
|
|
|
|
async def get_tasks(self) -> set:
|
|
async with self.connection as connection:
|
|
res = await connection.smembers(self.TASKS_NAME + f":1")
|
|
return res
|
|
|
|
async def get_task_done_queue(self) -> set:
|
|
async with self.connection as connection:
|
|
res = await connection.smembers(self.TASKS_DONE_NAME + f":1")
|
|
return res
|
|
|
|
async def del_task_from_queue(self, task: dict) -> int:
|
|
async with self.connection as connection:
|
|
res = await connection.srem(self.SET_NAME, json.dumps(task, indent=4).encode('utf-8'))
|
|
return res
|
|
|
|
async def del_task_from_queue_and_add_to_tasks(self, task: dict) -> int:
|
|
await self._del_task(task, self.SET_NAME)
|
|
return await self._set_task(task, self.TASKS_NAME)
|
|
|
|
async def del_task_from_tasks_and_add_to_task_done(self, task: dict, working_task: dict) -> int:
|
|
await self._del_task(working_task, self.TASKS_NAME)
|
|
return await self._set_task(task, self.TASKS_DONE_NAME)
|
|
|
|
async def del_task_from_task_done_queue(self, task) -> int:
|
|
async with self.connection as connection:
|
|
res = await connection.srem(self.TASKS_DONE_NAME + f":1", json.dumps(task, indent=4).encode('utf-8'))
|
|
return res
|