MyBatis批量插入轻松搞定，MySql一键数据更新

后端

2024-01-13 09:56:07

解决 MySQL 批量插入唯一键冲突：MyBatis 的强大解决方案

批量插入的意义

在软件开发中，批量插入数据是一种常见且至关重要的操作，例如导入数据或同步数据。批量插入与逐条插入相比，可以显著提高效率。

MySQL 唯一键冲突问题

使用 MyBatis 进行批量插入时，可能会遇到 MySQL 唯一键冲突错误。这是因为 MySQL 在插入数据时会检查唯一键是否重复。当检测到重复的唯一键时，就会引发错误。

MyBatis 批量插入 + 唯一键冲突更新

为了解决这个问题，我们可以利用 MyBatis 的批量插入功能，并在遇到唯一键冲突时将数据更新为新值。这样，既可以避免错误，又可以确保数据的准确性。

实现步骤

1. 添加唯一注解

在实体类中，为除 ID 以外的所有字段添加 @Unique 注解。这将指示 MyBatis 这些字段具有唯一约束。

2. 修改 Mapper.xml

将 insert 标签更改为 insertOrUpdate 标签。insertOrUpdate 标签会在检测到唯一键冲突时将数据更新为新值。

<insertOrUpdate id="batchInsert" parameterType="java.util.List">
  insert into table (field1, field2, field3) values
  <foreach collection="list" item="item" separator=",">
    (#{item.field1}, #{item.field2}, #{item.field3})
  </foreach>
  on duplicate key update field1 = values(field1), field2 = values(field2), field3 = values(field3)
</insertOrUpdate>

3. 使用批量插入方法

在 Java 代码中，使用 MyBatis 的批量插入方法即可实现批量插入数据。

List<Entity> list = new ArrayList<>();
// 添加数据
list.add(new Entity());
list.add(new Entity());
// 执行批量插入
int[] result = sqlSession.insert("batchInsert", list);

示例代码

@Entity
public class Entity {

  @Id
  @GeneratedValue(strategy = GenerationType.IDENTITY)
  private Long id;

  @Unique
  private String field1;

  @Unique
  private String field2;

  @Unique
  private String field3;

  // getters and setters
}

@Mapper
public interface EntityMapper {

  @InsertOrUpdate
  int[] batchInsert(List<Entity> list);
}