如何使用Hibernate在MySQL自增主键列中插入指定值？

针对数据迁移时需要插入固定ID的需求，这里给你几个实用方案：

方案一：直接用原生SQL插入（最稳妥，适合迁移） #

Hibernate的ORM映射会自动处理自增列的生成逻辑，迁移时绕过这套逻辑直接用原生SQL是最省心的方式，不同数据库的操作细节如下：

• MySQL：如果列是AUTO_INCREMENT类型，直接执行插入语句即可，只要ID不冲突，数据库会自动处理后续自增序列：

  INSERT INTO your_table(id, name, other_column) VALUES(1001, '迁移数据', 'xxx');
  

  在Hibernate中执行原生SQL：

  String sql = "INSERT INTO your_table(id, name, other_column) VALUES(?, ?, ?)";
  session.createNativeQuery(sql)
         .setParameter(1, 1001L)
         .setParameter(2, "迁移数据")
         .setParameter(3, "xxx")
         .executeUpdate();
  

• PostgreSQL：如果列是SERIAL或IDENTITY类型，需要先开启手动插入权限，插入后再关闭：

  SET IDENTITY_INSERT your_table ON;
  INSERT INTO your_table(id, name, other_column) VALUES(1001, '迁移数据', 'xxx');
  SET IDENTITY_INSERT your_table OFF;
  

方案二：临时调整实体类注解，允许手动赋值 #

如果不想用原生SQL，可以临时修改实体类的ID字段配置，让Hibernate允许手动传入ID值：

@Id
@Column(name = "id", insertable = true) // 确保允许插入时传入ID
// 注释掉原来的@GeneratedValue，或者保留但插入时手动设置ID
private Long id;

然后在迁移代码中手动设置实体的ID，再调用session.save(entity)即可。注意：迁移完成后记得恢复原来的@GeneratedValue配置，避免影响正常业务逻辑。

方案三：自定义ID生成器（适合需要频繁切换场景） #

如果需要在日常业务和迁移场景中灵活切换ID生成逻辑，可以自定义Hibernate的IdentifierGenerator：

1. 实现自定义生成器：

public class CustomIdGenerator implements IdentifierGenerator {
    @Override
    public Serializable generate(SharedSessionContractImplementor session, Object object) throws HibernateException {
        if (object instanceof YourEntity entity && entity.getId() != null) {
            // 如果手动设置了ID，直接使用该值
            return entity.getId();
        }
        // 否则使用数据库自增逻辑
        var defaultGenerator = session.getFactory()
                                      .getIdentifierGenerator(YourEntity.class.getName());
        return defaultGenerator.generate(session, object);
    }
}

2. 在实体类中配置该生成器：

@Id
@GeneratedValue(generator = "customIdGenerator")
@GenericGenerator(name = "customIdGenerator", strategy = "com.yourpackage.CustomIdGenerator")
private Long id;

这样平时插入时不设置ID就用自增，迁移时设置ID就会使用手动传入的值。

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