Django gRPC服务PostgreSQL连接已关闭问题优化求助

针对你遇到的django.db.utils.InterfaceError: connection already closed错误，同时要避免每次调用django.db.close_old_connections()带来的响应延迟，试试以下几个方案：

1. 用请求前连接校验替代请求后关闭 #

django-grpc-framework支持自定义拦截器，我们可以在每个gRPC请求处理前检查数据库连接有效性，而不是请求结束后强制关闭。这样只有当连接真的失效时才会重建，不会每次都额外消耗连接建立的时间。

实现一个数据库连接拦截器：

from django.db import connection
from grpc_interceptor import ServerInterceptor
from grpc import ServicerContext

class DBConnectionInterceptor(ServerInterceptor):
    def intercept(self, method, request, context: ServicerContext, method_name):
        # 确保连接有效，失效时自动重连
        connection.ensure_connection()
        return method(request, context)

注册拦截器到gRPC服务：

from django_grpc_framework.server import Server
server = Server(interceptors=[DBConnectionInterceptor()])

2. 优化数据库连接池配置 #

调整Django的数据库连接参数，让连接池自动管理失效连接，避免被数据库或K8s网络策略主动断开后还被复用：

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.postgresql',
        'NAME': 'your_db_name',
        'USER': 'db_user',
        'PASSWORD': 'db_password',
        'HOST': 'db_host',
        'PORT': '5432',
        'CONN_MAX_AGE': 300,  # 连接最大存活时间，要小于数据库的超时设置
        'OPTIONS': {
            'connect_timeout': 5,
            # 针对psycopg2的连接池配置，需安装psycopg2-binary
            'pool_size': 10,
            'max_overflow': 20,
            'pool_recycle': 280,  # 提前回收连接，避免数据库主动断开
        }
    }
}

关键是pool_recycle要设置得比数据库的idle_in_transaction_session_timeout小，确保Django在数据库断开前就回收连接。

3. 调整K8s探针配置，自动清理失效Pod #

给K8s的Pod配置带数据库校验的存活/就绪探针，当数据库连接失效时自动重启Pod，避免无效连接被持续复用：

livenessProbe:
  exec:
    command:
    - python
    - -c
    - "from django.db import connection; connection.cursor().execute('SELECT 1');"
  initialDelaySeconds: 30
  periodSeconds: 10
  failureThreshold: 3
readinessProbe:
  exec:
    command:
    - python
    - -c
    - "from django.db import connection; connection.cursor().execute('SELECT 1');"
  initialDelaySeconds: 5
  periodSeconds: 5
  failureThreshold: 2

4. 定时刷新连接池，保持连接活跃 #

如果Pod长时间空闲导致数据库断开连接，可以添加定时任务定期执行简单查询，维持连接池里的连接活跃：

from apscheduler.schedulers.background import BackgroundScheduler
from django.db import connection
from django.apps import apps

def keep_db_connections_alive():
    with connection.cursor() as cursor:
        cursor.execute('SELECT 1')

def start_scheduler():
    scheduler = BackgroundScheduler()
    scheduler.add_job(keep_db_connections_alive, 'interval', minutes=4)
    scheduler.start()

# 在Django启动时初始化定时任务
class YourAppConfig(apps.AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app'

    def ready(self):
        start_scheduler()

内容的提问来源于stack exchange，提问作者Mahsa Fathi