Django+Django Ninja POST接口实现文章标签关联及校验方案问询

现有代码的问题 #

现有逻辑完全不合理，核心问题如下：

• 核心流程错误：文章创建逻辑写在标签遍历循环内，只要遍历到第一个合法tag就会直接创建文章、添加该标签后返回，剩余标签完全不会被关联，完全不符合多标签关联的需求
• 冗余操作过多：models.Article.objects.create执行后已经将数据写入数据库，不需要再调用qs.save()；刚创建的文章实例可以直接使用，不需要再调用get_object_or_404重复查询数据库
• 校验逻辑缺失：
  • 没有提前校验所有传入的tags_id是否都存在，只要中间有一个tag不存在，就会出现文章已创建成功、但标签关联失败的脏数据问题
  • 没有校验传入的tags_id是否为空，现有逻辑如果传入空列表会直接走到异常分支返回无意义错误
  • 仅靠tag>0判断标签合法性完全无效，数值为正但不存在的ID照样会触发报错
• 异常处理不规范：裸except会捕获所有异常包括键盘中断等系统异常，且错误提示{'msg': 'Not'}无任何业务含义，无法定位问题
• 分支逻辑错误：如果传入的第一个tag为负数会直接返回No Tags，但如果是后面的tag为负数反而不会触发该判断

优化实现方案 #

调整执行逻辑为「前置全量校验→原子操作数据→正确返回结果」，具体代码如下：

from django.db import transaction

@router.post('/article', response={200: schema.ArticleOut, 400: schema.Error, 500: schema.Error})
def create_article(request, tags_id: List[int], payload: schema.ArticleIn):
    # 校验标签列表非空
    if not tags_id:
        return 400, {'msg': '标签列表不能为空'}
    # 批量校验所有标签是否存在
    exist_tags = models.Tag.objects.filter(id__in=tags_id)
    if exist_tags.count() != len(tags_id):
        exist_ids = set(exist_tags.values_list('id', flat=True))
        not_exist_ids = [tid for tid in tags_id if tid not in exist_ids]
        return 400, {'msg': f'标签ID {",".join(map(str, not_exist_ids))} 不存在'}
    try:
        # 加事务保证创建文章和关联标签操作原子性，避免脏数据
        with transaction.atomic():
            article = models.Article.objects.create(**payload.dict())
            # 批量关联所有标签，无需循环操作
            article.tags.add(*exist_tags)
        return 200, article
    except Exception as e:
        return 500, {'msg': f'文章创建失败：{str(e)}'}

优化点说明 #

• 所有合法性校验前置，校验不通过不会操作数据库，减少无效DB请求
• 事务保证操作原子性，不会出现文章创建成功、标签关联失败的不一致问题
• 批量查询、批量关联标签，相比循环查库、循环添加标签的操作，数据库请求次数从N次降到2次，性能提升明显
• 区分400参数错误和500服务端错误，错误提示明确，符合接口设计规范
• 去掉所有冗余数据库操作，执行效率更高

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