Files
core/services/redis.py

83 lines
2.5 KiB
Python
Raw Normal View History

2024-04-08 10:38:58 +03:00
import logging
2024-10-14 13:08:43 +03:00
from fakeredis.aioredis import FakeRedis
from redis.asyncio import Redis
2023-12-17 23:30:20 +03:00
2024-10-14 13:08:43 +03:00
from settings import MODE, REDIS_URL
2024-01-24 15:36:34 +03:00
2024-02-26 15:56:13 +03:00
# Set redis logging level to suppress DEBUG messages
2024-04-17 18:32:23 +03:00
logger = logging.getLogger("redis")
2024-02-26 15:56:13 +03:00
logger.setLevel(logging.WARNING)
2024-08-07 09:51:09 +03:00
class RedisService:
2023-10-05 21:46:18 +03:00
def __init__(self, uri=REDIS_URL):
self._uri: str = uri
self.pubsub_channels = []
2023-10-13 13:48:17 +03:00
self._client = None
2023-10-05 21:46:18 +03:00
async def connect(self):
2024-10-14 13:08:43 +03:00
if MODE == "development":
self._client = FakeRedis(decode_responses=True)
else:
self._client = await Redis.from_url(self._uri, decode_responses=True)
2023-10-05 21:46:18 +03:00
async def disconnect(self):
2023-11-22 19:38:39 +03:00
if self._client:
await self._client.close()
2023-10-05 21:46:18 +03:00
async def execute(self, command, *args, **kwargs):
2023-11-22 19:38:39 +03:00
if self._client:
try:
2024-06-02 17:01:22 +03:00
logger.debug(f"{command}") # {args[0]}") # {args} {kwargs}")
2024-02-21 16:06:24 +03:00
for arg in args:
if isinstance(arg, dict):
2024-04-17 18:32:23 +03:00
if arg.get("_sa_instance_state"):
del arg["_sa_instance_state"]
2023-11-22 19:38:39 +03:00
r = await self._client.execute_command(command, *args, **kwargs)
2024-06-02 16:14:01 +03:00
# logger.debug(type(r))
# logger.debug(r)
2023-11-22 19:38:39 +03:00
return r
except Exception as e:
2024-01-24 15:36:34 +03:00
logger.error(e)
2023-10-05 21:46:18 +03:00
async def subscribe(self, *channels):
2023-11-22 19:38:39 +03:00
if self._client:
async with self._client.pubsub() as pubsub:
for channel in channels:
await pubsub.subscribe(channel)
self.pubsub_channels.append(channel)
2023-10-05 21:46:18 +03:00
async def unsubscribe(self, *channels):
2023-10-13 13:48:17 +03:00
if not self._client:
2023-10-05 21:46:18 +03:00
return
2023-10-13 13:48:17 +03:00
async with self._client.pubsub() as pubsub:
2023-10-13 13:13:45 +03:00
for channel in channels:
await pubsub.unsubscribe(channel)
self.pubsub_channels.remove(channel)
2023-10-05 21:46:18 +03:00
async def publish(self, channel, data):
2023-10-13 13:48:17 +03:00
if not self._client:
2023-10-05 21:46:18 +03:00
return
2023-10-13 13:48:17 +03:00
await self._client.publish(channel, data)
2023-10-05 21:46:18 +03:00
2024-09-27 10:18:08 +03:00
async def set(self, key, data, ex=None):
# Prepare the command arguments
args = [key, data]
# If an expiration time is provided, add it to the arguments
if ex is not None:
args.append("EX")
args.append(ex)
# Execute the command with the provided arguments
await self.execute("set", *args)
2024-08-08 17:55:34 +03:00
2024-08-08 18:13:51 +03:00
async def get(self, key):
2024-08-08 18:14:49 +03:00
return await self.execute("get", key)
2024-08-08 17:55:34 +03:00
2024-01-25 22:41:27 +03:00
2024-08-07 09:51:09 +03:00
redis = RedisService()
2023-10-05 21:46:18 +03:00
2024-04-17 18:32:23 +03:00
__all__ = ["redis"]