mybatis-plus 实现自增id且可手动输入id的方式

最新推荐文章于 2024-03-08 11:16:11 发布

原创

最新推荐文章于 2024-03-08 11:16:11 发布 · 1.1w 阅读

26 ·

CC 4.0 BY-SA版权

文章标签：

#mybatisplus #自增id

本文介绍了如何在mybatis-plus中实现主键自增并允许手动输入id的功能。通过创建自定义MappedStatement处理主键生成，扩展默认注入器，设置实体对象属性，以及配置DataSource和yml文件，可以成功地在插入数据时获取自增id或使用手动输入的id。

在用mybatisplus的时候可以通过注解实体字段从而实现对象和数据库字段的映射，其中主键id基本上采用的是mysql数据库自增的方式，如果此时采用

@TableId(value = "id", type = IdType.AUTO)方式，则每次插入一条数据都将只会用到数据库的自增id并把id返回注入到实体对象，此时如果想要手动插入id（基本上在同步数据的时候用到），就无法实现

如果此时采用

@TableId(value = "id", type = IdType.INPUT)方式，在对象id为空的时候确实也能实现自增，在id不为空的时候也能插入想要的id，不过无法获得插入成功返回的id，此时如果也想获得返回的id，只需要重写mybatisplus的预处理器即可，总共3个步骤，如下：

1、添加一个继承mybatisplus 抽象注入方法的类，在这个类里面重写自定义 MappedStatement方法，在方法里面对主键进行自增generate声明处理便于mybatisplus把数据插入成功的id封装到实体对象中

@SuppressWarnings("all")
public class Insert extends AbstractMethod {

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        KeyGenerator keyGenerator = new NoKeyGenerator();
        SqlMethod sqlMethod = SqlMethod.INSERT_ONE;
        String columnScript = SqlScriptUtils.convertTrim(tableInfo.getAllInsertSqlCo