MyBatis-Plus中高效利用saveBatch方法进行批量数据插入的指南

MyBatis-Plus 是 MyBatis 的增强工具，在 MyBatis 的基础上只做增强不做改变，为简化开发、提高效率而生。saveBatch 方法是 MyBatis-Plus 提供的一个用于批量插入数据的便捷方法。

以下是 saveBatch 方法的使用步骤和示例：

1. 引入依赖

首先，确保你的项目中已经引入了 MyBatis-Plus 的相关依赖。以下是 Maven 依赖示例：

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>最新版本号</version>
</dependency>

2. 配置 MyBatis-Plus

在 Spring Boot 项目中，通常需要在 application.yml 或 application.properties 文件中进行配置。以下是一个简单的配置示例：

mybatis-plus:
  mapper-locations: classpath:/mapper/**/*.xml
  type-aliases-package: com.example.demo.entity

3. 创建实体类

创建一个与数据库表对应的实体类。例如，假设我们有一个 User 表：

package com.example.demo.entity;

import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;

@Data
@TableName("user")
public class User {
    @TableId
    private Long id;
    private String name;
    private Integer age;
    private String email;
}

4. 创建 Mapper 接口

创建一个 Mapper 接口，继承 BaseMapper，这样你就可以使用 MyBatis-Plus 提供的各种便捷方法了：

package com.example.demo.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.example.demo.entity.User;
import org.apache.ibatis.annotations.Mapper;

@Mapper
public interface UserMapper extends BaseMapper<User> {
}

5. 使用 saveBatch 方法

在你的服务层或控制器中使用 saveBatch 方法进行批量插入。以下是一个示例：

package com.example.demo.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.example.demo.entity.User;
import java.util.List;

public interface UserService extends IService<User> {
}

package com.example.demo.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.example.demo.entity.User;
import com.example.demo.mapper.UserMapper;
import com.example.demo.service.UserService;
import org.springframework.stereotype.Service;

import java.util.List;

@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {

    @Override
    public boolean saveBatch(List<User> userList) {
        return this.baseMapper.insertBatchSomeColumn(userList); // 或者使用 this.saveBatch(userList);
    }
}

package com.example.demo.controller;

import com.example.demo.entity.User;
import com.example.demo.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.Arrays;
import java.util.List;

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    @PostMapping("/batch")
    public String saveBatchUsers(@RequestBody List<User> userList) {
        boolean result = userService.saveBatch(userList);
        return result ? "Success" : "Failure";
    }
}

注意事项

1. 事务管理：在批量插入时，如果数据量很大，建议开启事务管理，以保证数据的一致性。
2. 批量大小：如果一次插入的数据量非常大，建议分批次插入，以避免内存溢出或数据库连接超时等问题。MyBatis-Plus 提供了 insertBatchSomeColumn 等方法，可以通过配置批量大小来优化性能。
3. 数据校验：在批量插入之前，最好对数据进行校验，以避免无效数据插入数据库。

通过上述步骤，你就可以在 MyBatis-Plus 中使用 saveBatch 方法进行批量插入操作了。