[FIX] Redis initialization

redis.Redis.from_url(url) doesn't check if the url is valid

Before this commit: actual error are detected later when the client is actually used.
With this commit, client() makes sure to return a valid Redis client or None.

Also, the code makes sure not to log the password of the Redis URL
This commit is contained in:
Alexandre FLAMENT 2022-11-07 21:20:48 +00:00
parent 8f19bdaf17
commit 73b5a58d9e
1 changed files with 21 additions and 7 deletions

View File

@ -40,17 +40,31 @@ def client() -> redis.Redis:
def initialize(): def initialize():
global _CLIENT # pylint: disable=global-statement global _CLIENT # pylint: disable=global-statement
redis_url = get_setting('redis.url') redis_url = get_setting('redis.url')
if not redis_url:
return False
try: try:
if redis_url: # create a client, but no connection is done
_CLIENT = redis.Redis.from_url(redis_url) _CLIENT = redis.Redis.from_url(redis_url)
logger.info("connected redis: %s", redis_url)
# log the parameters as seen by the redis lib, without the password
kwargs = _CLIENT.get_connection_kwargs()
kwargs.pop('password', None)
kwargs = ' '.join([f'{k}={v!r}' for k, v in kwargs.items()])
logger.info("connecting to Redis %s", kwargs)
# check the connection
_CLIENT.ping()
# no error: the redis connection is working
logger.info("connected to Redis")
return True return True
except redis.exceptions.ConnectionError: except redis.exceptions.RedisError as e:
_CLIENT = None
_pw = pwd.getpwuid(os.getuid()) _pw = pwd.getpwuid(os.getuid())
logger.exception("[%s (%s)] can't connect redis DB ...", _pw.pw_name, _pw.pw_uid) logger.exception("[%s (%s)] can't connect redis DB ...", _pw.pw_name, _pw.pw_uid)
if redis_url == OLD_REDIS_URL_DEFAULT_URL: if redis_url == OLD_REDIS_URL_DEFAULT_URL and isinstance(e, redis.exceptions.ConnectionError):
logger.info( logger.info(
"You can safely ignore the above Redis error if you don't use Redis." "You can safely ignore the above Redis error if you don't use Redis. "
"You can remove this error by setting redis.url to false in your settings.yml." "You can remove this error by setting redis.url to false in your settings.yml."
) )
return False return False