"""健康检查：服务存活与数据库连通性。"""

from django.db import connection
from django.db.utils import OperationalError
from rest_framework.decorators import api_view, permission_classes
from rest_framework.permissions import AllowAny
from rest_framework.response import Response


@api_view(['GET', 'HEAD'])
@permission_classes([AllowAny])
def ping(request):
    """服务存活探测，通则返回 ok。"""
    return Response('ok')


@api_view(['GET', 'HEAD'])
@permission_classes([AllowAny])
def test_mysql(request):
    """MySQL 连通探测：使用 Django 连接池建立连接，通则 ok，否则返回没连通。"""
    try:
        connection.ensure_connection()
    except OperationalError:
        return Response('没连通', status=503)
    except Exception:
        return Response('没连通', status=503)
    return Response('ok')