17 lines
548 B
Python
17 lines
548 B
Python
|
from django.db import connection
|
||
|
|
from django.http import HttpResponse
|
||
|
|
|
||
|
|
|
||
|
|
class HealthCheckMiddleware:
|
||
|
|
"""Answer /healthz before host validation, so Fly's checks pass whatever Host they send."""
|
||
|
|
|
||
|
|
def __init__(self, get_response):
|
||
|
|
self.get_response = get_response
|
||
|
|
|
||
|
|
def __call__(self, request):
|
||
|
|
if request.path == "/healthz":
|
||
|
|
with connection.cursor() as cursor:
|
||
|
|
cursor.execute("SELECT 1")
|
||
|
|
return HttpResponse("ok", content_type="text/plain")
|
||
|
|
return self.get_response(request)
|