Spring Boot 3.3.1下如何实现实体ID自动生成与导入CSV时保留指定ID的兼容需求

我完全懂你这种折腾到深夜的痛苦——升级Spring Boot后，原来用得好好的自定义ID生成器突然不能用了，既要让系统自动生成ID（像IDENTITY那样），又要在导入CSV时严格保留文件里的ID，这种两头兼顾的需求确实让人头大。不过别担心，在Spring Boot 3.3.1（对应Hibernate 6.x）的生态下，我们完全可以通过调整自定义生成器的实现来解决这个问题。

核心思路 #

Hibernate 6.x对ID生成器的API做了不少调整，原来的IdentityGenerator继承方式不再可行，但我们可以换个思路：自定义一个实现Generator接口的生成器，内部同时处理「手动设置ID」和「自动生成ID」两种场景：

1. 先检查当前实体的ID是否已经被赋值（比如导入CSV时手动设置的）
2. 如果ID非空，直接返回这个ID
3. 如果ID为空，就调用Hibernate内置的IdentityGenerator来触发数据库自增逻辑

具体实现步骤 #

1. 编写兼容Hibernate 6.x的自定义ID生成器

import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.generator.EventType;
import org.hibernate.generator.Generator;
import org.hibernate.id.IdentityGenerator;
import org.hibernate.type.Type;

import java.io.Serializable;
import java.util.Properties;

public class DualModeIdGenerator implements Generator {

    // 内置的Identity生成器，用于处理自动生成ID的场景
    private IdentityGenerator identityGenerator;

    @Override
    public void configure(Type type, Properties params, org.hibernate.service.ServiceRegistry serviceRegistry) {
        // 初始化内置的Identity生成器，复用它的自增逻辑
        identityGenerator = new IdentityGenerator();
        identityGenerator.configure(type, params, serviceRegistry);
    }

    @Override
    public Serializable generate(SharedSessionContractImplementor session, Object entity, EventType eventType) {
        // 获取当前实体已设置的ID值
        Serializable entityId = session.getEntityPersister(null, entity)
                .getClassMetadata()
                .getIdentifier(entity, session);

        // 如果ID已被手动设置（比如导入CSV时赋值），直接返回该ID
        if (entityId != null) {
            return entityId;
        }

        // 否则调用内置的Identity生成器，触发数据库自增获取ID
        return identityGenerator.generate(session, entity, eventType);
    }
}

2. 实体类上配置自定义生成器

在你的实体类的ID字段上，指定使用这个自定义生成器，同时关联IDENTITY策略：

import jakarta.persistence.*;

@Entity
@Table(name = "your_entity_table")
public class YourEntity {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY, generator = "dual-mode-id-generator")
    @GenericGenerator(
            name = "dual-mode-id-generator",
            type = DualModeIdGenerator.class
    )
    private Long id;

    // 其他业务字段...
    // getter、setter方法...
}

3. 关键Hibernate配置

Hibernate 6.x默认会忽略手动设置的IDENTITY类型ID，所以需要在application.properties（或application.yml）中添加以下配置，允许手动插入ID值：

# 允许Hibernate接受手动设置的IDENTITY类型ID
spring.jpa.properties.hibernate.id.allow_manual_insert=true

验证你的流程 #

现在按照你原来的三步流程测试：

1. 导出数据库内容到CSV
2. 导入空CSV清空表
3. 导入之前导出的CSV
   你会发现数据库内容和操作前完全一致——手动设置的ID被正确保留，空ID的实体也会自动生成自增ID。

额外注意事项 #

• 确保实体的ID字段类型和数据库自增列类型匹配（比如Java的Long对应MySQL的BIGINT AUTO_INCREMENT）
• 导入CSV时，要保证ID值的格式正确，避免类型转换错误
• 如果你的项目中有多个实体需要这个逻辑，可以把生成器抽象成通用类，避免重复代码

内容来源于stack exchange