老旧测试用例生成平台异步任务与用户通知优化

在现代 Web 开发中，异步任务处理和用户通知是两个重要的功能。由于老旧测试平台【测试用例生成平台，源码分享】进行智能化升级后，未采用异步任务处理，大模型推理时间较长，导致任务阻塞，无法处理其他任务，体验较差。本文将以 Django 框架为基础，结合 Celery 和 Redis，完成一个完整的异步任务系统，并实现用户通知功能。我们还会对接阿里云百炼 DeepSeek-R1 模型，通过大模型生成测试用例，并提供数据的增删改查功能。以下是详细的实现步骤和完整代码示例。

---

一、需求分析与功能拆解

1.1 功能需求

本项目的主要需求如下：

1. 用户通过 API 输入任务名称和字段信息，系统为任务生成唯一的 UUID，并提示用户任务已创建。
2. 异步任务系统调用阿里云百炼 DeepSeek-R1 模型，传递字段信息生成 JSON 格式的测试用例。
3. 将模型生成的测试用例存储到数据库的测试用例预览表中。
4. 通知用户任务完成，支持通过 WebSocket 或邮件通知。
5. 提供一个页面对生成的测试用例数据进行增删改查操作。

1.2 技术选型

• Django：作为后端框架，负责处理 API 请求、任务状态管理和数据库操作。
• Celery：用于实现任务的异步处理。
• Redis：作为 Celery 的消息队列。
• 阿里云百炼 DeepSeek-R1：调用大模型接口生成测试用例。
• Django Channels：实现 WebSocket 通知功能。
• 前端页面：用于展示和管理测试用例数据。

---

二、项目创建与环境配置

2.1 安装依赖

在开始开发之前，需要安装以下依赖工具和库：

pip install django celery redis requests channels

2.2 项目目录结构

项目的目录结构如下：

your_project_name/
├── your_project_name/
│   ├── __init__.py
│   ├── settings.py
│   ├── urls.py
│   ├── asgi.py
│   ├── celery.py
├── your_app_name/
│   ├── __init__.py
│   ├── models.py
│   ├── views.py
│   ├── tasks.py
│   ├── signals.py
│   ├── urls.py
├── manage.py

---

三、系统实现

3.1 配置 Celery 和 Redis

3.1.1 在 settings.py 中配置 Celery

# Celery 配置
CELERY_BROKER_URL = 'redis://localhost:6379/0'
CELERY_ACCEPT_CONTENT = ['json']
CELERY_TASK_SERIALIZER = 'json'

# Django Channels 配置
ASGI_APPLICATION = 'your_project_name.asgi.application'
CHANNEL_LAYERS = {
   
    'default': {
   
        'BACKEND': 'channels_redis.core.RedisChannelLayer',
        'CONFIG': {
   
            'hosts': [('127.0.0.1', 6379)],
        },
    },
}

3.1.2 创建 celery.py

在项目根目录下创建 celery.py 文件，用于初始化 Celery：

from __future__ import absolute_import, unicode_literals
import os
from celery import Celery

# 设置 Django 默认的 settings 模块
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'your_project_name.settings')

app = Celery('your_project_name')

# 从 Django 的 settings.py 加载配置
app.config_from_object('django.conf:settings', namespace='CELERY')

# 自动发现任务模块
app.autodiscover_tasks()

3.1.3 配置 __init__.py

在项目目录下的 __init__.py 文件中添加以下内容：

from __future__ import absolute_import, unicode_literals

# Celery 应用
from .celery import app as celery_app

__all__ = ('celery_app',)

3.1.4 配置 asgi.py

在项目根目录下的 asgi.py 文件中配置 WebSocket 支持：

import os
from django.core.asgi import get_asgi_application
from channels.routing import ProtocolTypeRouter, URLRouter
from channels.auth import AuthMiddlewareStack

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'your_project_name.settings')

application = ProtocolTypeRouter({
   
    'http': get_asgi_application(),
    'websocket': AuthMiddlewareStack(
        URLRouter([
            # WebSocket 路由
        ]