Docker Compose环境下Celery执行任务时找不到Dispatch模型

可能原因及对应解决办法 #

1. 数据库连接不一致 #

Docker Compose环境里，Celery容器和Web服务可能没连同一个数据库，或者数据库还没启动就绪Celery就开始执行任务了。

• 检查docker-compose.yml里Celery的数据库连接配置，确保和Web服务使用相同的数据库主机（比如用容器名db作为主机名）。
• 给Celery服务添加depends_on和数据库健康检查，保证数据库就绪后再启动Celery：

  services:
    db:
      image: postgres:15
      healthcheck:
        test: ["CMD-SHELL", "pg_isready -U your_user -d your_db"]
        interval: 5s
        timeout: 5s
        retries: 5
    celery:
      build: .
      command: celery -A src worker --loglevel=info
      depends_on:
        db:
          condition: service_healthy
        web:
          condition: service_started
  

2. 事务未提交就触发任务 #

如果Dispatch是在Web服务的事务中创建的，Celery任务在事务提交前就执行，自然查不到记录。

• 要么创建Dispatch后立刻提交事务，要么用transaction.on_commit()延迟触发任务，示例：

  from django.db import transaction
  
  def create_dispatch(request):
      dispatch = Dispatch.objects.create(...)
      # 等事务提交后再发送Celery任务
      transaction.on_commit(lambda: send_message_to_client.delay(dispatch.id))
      return Response(...)
  

3. 任务参数未传原始ID #

要是给Celery任务传的是序列化后的对象（比如用pickle），可能出现ID不匹配的情况。

• 确保任务只接收Dispatch的ID，任务内部用ID重新查询：

  # tasks.py
  @shared_task
  def send_message_to_client(dispatch_id):
      try:
          dispatch = Dispatch.objects.get(id=dispatch_id)
          # 执行发送逻辑
      except Dispatch.DoesNotExist:
          # 添加日志方便排查
          logger.error(f"Dispatch {dispatch_id} 不存在")
          raise
  

4. 多数据库配置错误 #

如果项目使用了多数据库，Celery可能没走正确的数据库路由查询Dispatch。

• 检查settings.py里的DATABASE_ROUTERS配置，确保Celery使用的数据库能访问Dispatch表。
• 也可以在任务里显式指定数据库：

  dispatch = Dispatch.objects.using('default').get(id=dispatch_id)
  

5. Docker网络不通 #

Celery容器无法连接到数据库容器，导致查询失败。

• 检查Docker Compose的网络配置，确保Celery和数据库在同一个自定义网络里（默认Compose会创建默认网络，所有服务都加入其中）。
• 进入Celery容器测试能否ping通数据库容器：

  docker exec -it 你的celery容器名 ping db
  

调试小技巧 #

• 在Celery任务里添加详细日志，打印要查询的ID、数据库连接信息，确认查询的是否为正确的数据库：

  import logging
  
  logger = logging.getLogger(__name__)
  
  @shared_task
  def send_message_to_client(dispatch_id):
      logger.info(f"尝试查询Dispatch ID: {dispatch_id}")
      logger.info(f"当前使用数据库: {settings.DATABASES['default']['NAME']}")
      try:
          dispatch = Dispatch.objects.get(id=dispatch_id)
          logger.info(f"成功查到Dispatch: {dispatch}")
      except Dispatch.DoesNotExist:
          logger.error(f"数据库 {settings.DATABASES['default']['NAME']} 中找不到Dispatch {dispatch_id}")
          raise
  

• 直接进入Celery容器启动Django shell，手动查询该ID的Dispatch，确认记录是否存在：

  docker exec -it 你的celery容器名 python manage.py shell
  >>> from src.newsletter.models import Dispatch
  >>> Dispatch.objects.get(id=你的dispatch_id)
  

内容的提问来源于stack exchange，提问作者Юрий