咨询Django/DRF实现类似Spring JPA级联插入的方案

当然可以实现这种级联创建父子模型的逻辑！在Django和Django Rest Framework（DRF）里，咱们可以通过嵌套序列化器+重写create方法来达成和你熟悉的Spring JPA级联插入一样的效果——保存父模型时自动创建关联的子模型，并且自动关联父模型的自增ID。

第一步：定义模型结构 #

首先咱们先把Conversation和Message的模型建好，注意外键的关联配置：

from django.db import models

class Conversation(models.Model):
    title = models.CharField(max_length=255)
    # 不需要手动定义关联message的字段，Django会通过Message的外键自动生成反向关联

class Message(models.Model):
    # 外键关联Conversation，on_delete设置为CASCADE表示删除会话时同时删除消息
    conversation = models.ForeignKey(
        Conversation, 
        on_delete=models.CASCADE, 
        related_name='messages'  # 反向关联的名称，方便后续查询
    )
    content = models.TextField()  # 这里用content代替示例里的message字段，避免和模型名冲突

第二步：编写嵌套序列化器 #

这是实现级联创建的核心部分。咱们需要在ConversationSerializer里嵌套MessageSerializer，然后重写create方法来手动处理父子模型的创建逻辑：

from rest_framework import serializers
from .models import Conversation, Message

class MessageSerializer(serializers.ModelSerializer):
    class Meta:
        model = Message
        fields = ['content']  # 不需要包含conversation_id，咱们会自动关联父模型

class ConversationSerializer(serializers.ModelSerializer):
    # 嵌套单个Message的序列化器，如果需要支持多个消息，可以改成many=True
    message = MessageSerializer()

    class Meta:
        model = Conversation
        fields = ['title', 'message']

    def create(self, validated_data):
        # 先从验证后的参数中取出子模型的 data
        message_data = validated_data.pop('message')
        # 先创建父模型Conversation实例
        conversation = Conversation.objects.create(**validated_data)
        # 再创建子模型Message，自动关联刚生成的conversation
        Message.objects.create(conversation=conversation, **message_data)
        return conversation

第三步：配置ViewSet #

用DRF的ModelViewSet就能快速完成CRUD操作，不需要额外编写保存逻辑——因为序列化器已经处理好了级联创建：

from rest_framework import viewsets
from .models import Conversation
from .serializers import ConversationSerializer

class ConversationViewSet(viewsets.ModelViewSet):
    queryset = Conversation.objects.all()
    serializer_class = ConversationSerializer

测试请求 #

现在你可以用类似这样的JSON请求来测试级联插入：

{
  "title": "chat",
  "message": {
    "content": "hi"
  }
}

发送POST请求后，DRF会自动创建Conversation实例，然后创建关联的Message实例，并且把Message的conversation_id设置为刚生成的Conversation的自增ID。

扩展：支持多个子模型 #

如果需要一次创建多个Message，只需要修改序列化器和create方法：

# 修改ConversationSerializer的message字段
messages = MessageSerializer(many=True)

# 重写create方法处理多个消息
def create(self, validated_data):
    messages_data = validated_data.pop('messages')
    conversation = Conversation.objects.create(**validated_data)
    # 循环创建多个Message实例
    for msg_data in messages_data:
        Message.objects.create(conversation=conversation, **msg_data)
    return conversation

对应的请求JSON就变成：

{
  "title": "group chat",
  "messages": [
    {"content": "hi"},
    {"content": "hello there"}
  ]
}

简单来说，Django ORM不像Spring JPA那样支持自动级联创建嵌套对象，但通过DRF的序列化器重写create方法，咱们可以灵活实现同样的逻辑，而且可控性更强。

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