MybatisPlus中插入数据后获取该对象主键值的实现

在MybatisPlus中，插入数据并获取对象的主键值是一个常见的需求，特别是在需要跟踪新插入记录的唯一标识时。MybatisPlus提供了一种简便的方式来实现这一功能。本篇文章将详细解释如何在MybatisPlus中实现这一操作，并通过示例代码进行演示。 我们需要创建一个实体类，例如`Employee`，并确保主键字段的`IdType`设置为`AUTO`，表示数据库会自动为该字段生成值。在Java中，可以使用Lombok库的注解来简化对象的创建，如下所示： ```java import com.baomidou.mybatisplus.annotation.TableId; import com.baomidou.mybatisplus.annotation.IdType; import lombok.Data; import lombok.EqualsAndHashCode; import lombok.Accessors; @Data @EqualsAndHashCode(callSuper = false) @Accessors(chain = true) public class Employee implements Serializable { private static final long serialVersionUID = 1L; // 设置主键IdType为 AUTO，表示数据库ID自增 @TableId(value = "id", type = IdType.AUTO) private Integer id; private String lastName; private String email; private Integer gender; private Integer age; } ``` 接下来，我们可以创建一个控制器方法，用于插入新的员工记录并返回包含主键值的对象。在下面的代码中，`employeeService.saveOrUpdate(employee)`会执行插入操作，并且返回的`employee`对象已经包含了数据库生成的主键值： ```java import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.ResponseBody; import com.baomidou.mybatisplus.extension.service.IService; import com.example.entity.Employee; // 假设EmployeeService是你的服务接口 @Service public interface EmployeeService extends IService<Employee> {} // 控制器类 @RestController public class EmployeeController { @Autowired private EmployeeService employeeService; // 插入数据并返回包含主键的实体 @PostMapping("add") @ResponseBody public Employee addEmployee() { Employee employee = new Employee(); employee.setLastName("chen") .setAge(18) .setEmail("10000@qq.com") .setGender(1); employeeService.saveOrUpdate(employee); // 插入并返回对象 return employee; } } ``` 此外，如果你更倾向于在Mapper层处理插入操作，可以创建一个自定义的`saveEmp`方法，调用`baseMapper.insert(employee)`后，返回包含主键值的`employee`对象： ```java import org.apache.ibatis.annotations.Insert; import org.apache.ibatis.annotations.SelectKey; import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.example.entity.Employee; public interface EmployeeMapper extends BaseMapper<Employee> { // 自定义保存方法，返回插入后的对象 @Insert("INSERT INTO employee(last_name, email, gender, age) VALUES (#{lastName}, #{email}, #{gender}, #{age})") @SelectKey(statement = "SELECT LAST_INSERT_ID()", keyProperty = "id", before = false, resultType = Integer.class) Employee saveEmp(Employee employee); } ``` 在上述代码中，`@SelectKey`注解用于在插入后获取数据库生成的主键值，并将其设置到`employee`对象的`id`属性上。 总结起来，MybatisPlus提供了便捷的方法来插入数据并获取对象的主键值。通过设置`IdType.AUTO`，MybatisPlus会在插入数据时自动处理数据库的自增ID，然后将生成的主键值返回给插入的对象。无论是通过Service层的`saveOrUpdate`方法还是Mapper层的自定义`insert`方法，都能轻松实现这一功能。了解并掌握这一特性，能够帮助我们在实际开发中更加高效地处理数据操作。