redis: add support to connect using unix socket

Signed-off-by: Christoph Heiss <christoph@c8h4.io>
This commit is contained in:
Christoph Heiss 2024-03-26 22:47:09 +01:00
parent b59b0c248b
commit c2a98b3dc7
No known key found for this signature in database
GPG Key ID: 73D5E7FDEE3DE49A
2 changed files with 13 additions and 5 deletions

View File

@ -31,6 +31,9 @@ class EnvironmentSettings:
# Redis
REDIS_HOST: str = str(environ.get("REDIS_HOST"))
REDIS_PORT: int = int(environ.get("REDIS_PORT", 6379))
REDIS_SOCKET: str | None = (
str(environ.get("REDIS_SOCKET")) if "REDIS_SOCKET" in environ else None
)
REDIS_NAME_SPACE: str = str(environ.get("REDIS_NAME_SPACE", "ta:"))
# ElasticSearch

View File

@ -17,11 +17,16 @@ class RedisBase:
NAME_SPACE: str = EnvironmentSettings.REDIS_NAME_SPACE
def __init__(self):
self.conn = redis.Redis(
host=EnvironmentSettings.REDIS_HOST,
port=EnvironmentSettings.REDIS_PORT,
decode_responses=True,
)
kwargs = {
"decode_responses": True,
"host": EnvironmentSettings.REDIS_HOST,
"port": EnvironmentSettings.REDIS_PORT,
}
if EnvironmentSettings.REDIS_SOCKET:
kwargs["unix_socket_path"] = EnvironmentSettings.REDIS_SOCKET
self.conn = redis.Redis(**kwargs)
class RedisArchivist(RedisBase):