Spring JPA保存父实体后子实体ID为空的问题求助

这是我职业生涯中第一个采用数据库端ID生成策略的项目，不清楚如何同时兼顾该策略与整洁的代码库。

典型场景：定义了Parent Entity和Child Entity，代码如下：

Parent Entity

@Entity(name = "parent")
@Table(name = "parent", schema = "schema")
class ParentEntity(
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    var id: Long? = null,

    @OneToMany(
        mappedBy = "parent",
        cascade = [CascadeType.ALL],
        orphanRemoval = true,
        fetch = FetchType.LAZY
    )
    private val _children: MutableSet<ChildEntity> = mutableSetOf()

    val children
        get() = _children.toList()

    fun addChild(child: ChildEntity): ChildEntity {
        _children.add(child)
        child.parent = this
        return child
    }

    fun removeChild(child: ChildEntity): Boolean {
        return _children.remove(child)
    }

    override fun equals(other: Any?): Boolean {
        if (this === other) return true
        if (other == null || other !is ParentEntity) return false

        return id != null && id == other.id
    }

    override fun hashCode(): Int {
        return javaClass.hashCode()
    }
}

Child Entity

@Entity(name = "child")
@Table(name = "child", schema = "schema")
class ChildEntity(
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    var id: Long? = null,

    var title: String? = null,

    @ManyToOne(optional = false, fetch = FetchType.LAZY)
    @JoinColumn
    var parent: ParentEntity
) {

    override fun equals(other: Any?): Boolean {
        if (this === other) return true
        if (other == null || other !is ChildEntity) return false

        return id != null && id == other.id
    }

    override fun hashCode(): Int {
        return javaClass.hashCode()
    }
}

服务层添加子实体的代码：

@Transactional
override fun addChildToParent(parentId: Long, input: ChildInputDto): ChildDto {
    val currentParent = parentService.findParentById(parentId) ?: throw NotFoundException("Parent not found")
            
    val newChild = childMapper.dtoToEntity(input)

    val addedChild = currentParent.addChild(newChild)

    val savedParent = parentService.update(currentParent) // 内部调用`parentRepository.save(currentParent)`

    return childMapper.entityToDto(addedChild) // 此处addedChild的id为null
}

已知savedParent中的子实体已生成新ID，但子实体唯一标识是id，无法以此从savedParent中筛选目标子实体。

补充：

• 已遵循JPA实体equals/hashcode的最佳实践，但子实体状态转换时表现不一致；
• 暂时无法重构为应用端生成UUID的方案。

方案一：利用JPA持久态对象的自动同步特性 #

处于事务中的持久态ParentEntity，添加临时态ChildEntity到集合后，由于配置了CascadeType.ALL，JPA会自动持久化子实体，并将生成的ID直接赋值给原addedChild对象，无需手动调用save。修改服务层代码：

@Transactional
override fun addChildToParent(parentId: Long, input: ChildInputDto): ChildDto {
    val currentParent = parentService.findParentById(parentId) ?: throw NotFoundException("Parent not found")
            
    val newChild = childMapper.dtoToEntity(input)
    val addedChild = currentParent.addChild(newChild)

    // 事务提交时自动同步，addedChild的ID已被JPA填充
    return childMapper.entityToDto(addedChild)
}

方案二：优化equals/hashcode实现，兼容临时态与持久态 #

针对临时态（未保存）实体，补充业务键比较逻辑，避免仅依赖ID导致的匹配问题。以ChildEntity为例（假设title+parent可作为临时态唯一标识）：

class ChildEntity(
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    var id: Long? = null,

    var title: String? = null,

    @ManyToOne(optional = false, fetch = FetchType.LAZY)
    @JoinColumn
    var parent: ParentEntity
) {

    override fun equals(other: Any?): Boolean {
        if (this === other) return true
        if (other == null || other !is ChildEntity) return false

        // 持久态用ID比较
        if (id != null && other.id != null) {
            return id == other.id
        }

        // 临时态用业务键+父实体ID比较
        return title == other.title && parent.id == other.parent.id
    }

    override fun hashCode(): Int {
        return if (id != null) {
            id.hashCode()
        } else {
            var result = title?.hashCode() ?: 0
            result = 31 * result + parent.id.hashCode()
            result
        }
    }
}

方案三：通过对象引用匹配子实体 #

若必须依赖savedParent，可利用对象引用相等性直接匹配（addedChild与savedParent集合中的实例为同一对象）：

@Transactional
override fun addChildToParent(parentId: Long, input: ChildInputDto): ChildDto {
    val currentParent = parentService.findParentById(parentId) ?: throw NotFoundException("Parent not found")
            
    val newChild = childMapper.dtoToEntity(input)
    val addedChild = currentParent.addChild(newChild)

    val savedParent = parentService.update(currentParent)
    val savedChild = savedParent.children.firstOrNull { it === addedChild } 
        ?: throw IllegalStateException("Child not found in saved parent")

    return childMapper.entityToDto(savedChild)
}

内容的提问来源于stack exchange，提问作者mr.nothing