Add remaining health checks and shutdown on critical errors

This commit is contained in:
Louis Holbrook
2021-04-24 17:53:45 +00:00
parent cd0e702e3a
commit f50da54274
20 changed files with 261 additions and 42 deletions

View File

@@ -0,0 +1,18 @@
# external imports
import redis
import os
def health(*args, **kwargs):
r = redis.Redis(
host=kwargs['config'].get('REDIS_HOST'),
port=kwargs['config'].get('REDIS_PORT'),
db=kwargs['config'].get('REDIS_DB'),
)
try:
r.set(kwargs['unit'], os.getpid())
except redis.connection.ConnectionError:
return False
except redis.connection.ResponseError:
return False
return True