首页 > 软件编程 > java > mybatis插入数据后返回自增主键ID

mybatis插入数据后返回自增主键ID的两种实现方式

2024-05-18 09:31:43 作者：随码去

这篇文章主要介绍了mybatis插入数据后返回自增主键ID的两种实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

mybatis插入数据后返回自增主键ID

开发过程中我们经常性的会用到许多的中间表，mybatis插入数据后返回自增主键的两种方式，

用于数据之间的对应和关联，这个时候我们关联最多的就是ID，我们在一张表中插入数据后级联增加到关联表中，我们数值的mybatis在插入数据后返回的是插入成功的条数，那么这个时候我们想要得到相应的新增数据的ID，该怎么办呢？

下面我们介绍二种方式来实现，一种是在xml文件种设置属性，另外一种就是使用注解来实现。

整体两种方法流程

1.使用注解@SelectKey,继承原生的方法insert来实现

这种方法是使用注解，不使用xml方法，使用继承原生的方法insert来实现，在注解里面写号SQL以及每个字段绑定的变量即可。

注解的方法为

@Override
        @Insert(" INSERT INTO `air_quality_index` (`districtId`, `monitorTime`, `Pm10`,`Pm25`,`monitoringStation`) VALUES (#{districtId}, NOW(), #{Pm10},#{Pm25},#{monitoringStation}))
@SelectKey(statement="select currval('air_quality_index_id_seq')"),keyProperty = "id",before=false,resultType = Long.class)
int insert(User user);

@SelectKey(statement="select currval('air_quality_index_id_seq')"),keyProperty = "id",before=false,resultType = Long.class)

其中select currval('air_quality_index_id_seq')")就是pgsql插入后返回的自增id，这种方式需要把数据库表的id设置成自增ID，这样注解就生效了

在使用的时候直接用

result = getBaseMapper().insert(user);

最后也得到了这个自增的ID,当然了，我这个用的是pgsql,如果是mysql，在执行select LAST_INSERT_ID()得到的结果那么注解就得改一下：

@SelectKey(statement="select LAST_INSERT_ID()",keyProperty = "id",before = false,resultType = Long.class)

其中select LAST_INSERT_ID()就是mysql插入数据后，获取的当前的自增id。

2.在xml中定义useGeneratedKeys为true,返回主键id的值

keyProperty和keyColum分别代表

数据库记录主键字段和java对象成员属性名：在XML文件中要设置userGeneratedKeys="true" keyProperty="id"

<insert id="insertUser" useGeneratedKeys="true" keyProperty="id"> 
@Insert(" INSERT INTO `air_quality_index` (`districtId`, `monitorTime`, `Pm10`,`Pm25`,`monitoringStation`) VALUES (#{districtId}, NOW(), #{Pm10},#{Pm25},#{monitoringStation}))
</insert>

mybatis自动生成主键，返回自增主键 insert语句

对于表的主键自动生成的 insert语句 , 在不需要返回自增主键时 , 就和普通的insert语句没有区别 , 在需要的返回的时候 , 情况稍微有点不一样 , 故在此记录一下 . 在MyBatis中 , 有两种方式编写SQL , 一种XML方式 , 一种是注解方式 .

首先在MYSQL中新建一张表 , 建表语句如下

CREATE TABLE `sys_user` (
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '用户id',
  `user_name` varchar(64) DEFAULT NULL COMMENT '用户名',
  `password` varchar(64) DEFAULT NULL COMMENT '密码',
  `email` varchar(100) DEFAULT NULL COMMENT '邮箱',
  `info` text COMMENT '简介',
  `create_time` datetime DEFAULT NULL COMMENT '创建时间',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='用户表';

在使用主键自增的时候 , 插入数据库后可能需要得到自增的主键值 , 然后使用这个值进行一些其他的操作 .

有些数据库提供主键自增(MYSQL , SQLSERVER) , 有些不提供(ORACLE) , 而是使用序列得到一个值 , 然后将这个值赋给 id , 再讲数据插入.

1.XML方式

1.1返回自增主键

<insert id="insert1" useGeneratedKeys="true" keyProperty="id">
        insert into sys_user(user_name,password,email,info,create_time)
        values(#{userName},#{password},#{email},#{info},#{createTime})
    </insert>

使用useGeneratedKeys设置为true后 , MyBatis会使用JDBC的getGeneratedKeys方法取出由数据库内部生成的主键 .

获得主键值之后将其赋给keyProperty配置的 id 属性 .

当需要设置多个属性时 , 使用逗号隔开 , 这种情况下还需要配置 keyColumn属性 , 按顺序指定数据库的列 , 这里列的值会和 keyProperty 配置的属性一一对应 .

由于要使用数据库返回的主键值 , 所以SQL上下两部分的列中去掉了 id 列和对于的 #{id} 属性

1.2返回非自增主键

采用 <selectKey>标签获取主键的值 , 这种方式对提供和不提供主键自增功能的数据库同样适用

<insert id="insert2">
        insert into sys_user(user_name,password,email,info,create_time)
        values(#{userName},#{password},#{email},#{info},#{createTime})
        <selectKey keyColumn="id" resultType="int" keyProperty="id" order="AFTER">
            SELECT LAST_INSERT_ID()
        </selectKey>
    </insert>

selectKey标签的 keyColumn,keyProperty和上面useGeneratedKeys的用法和含义一样 , 这里的resultType用于设置返回值类型.

order 属性和数据库相关 , 在MYSQL 中 , order是AFTER , 因为当前及记录的主键值在insert语句执行成功之后才能拿到 , 而在ORACLE中，oder是BEFORE , 因为ORACLE需要先从序列取到值 , 再将其作为主键插入到数据库

2.注解方式

2.1返回自增主键

@Insert("insert into sys_user(user_name,password,email,info,create_time)"
            +"values(#{userName},#{password},#{email},#{info},#{createTime})")
    @Options(keyColumn="id",keyProperty="id",useGeneratedKeys=true)
    int insert(SysUser user);

insert方法上除了 @Insert 注解之外 , 多了一个 @Option 注解 , 在这个注解上配置了useGeneratedKeys 和 keyProperty属性，用法和 XML 的方式一样 , 当需要配置多个列的时候 , 这个注解也提供了 keyColumn 属性 , 可以像XML中那样配置使用 .

2.2返回非自增

@Insert("insert into sys_user(user_name,password,email,info,create_time)"
            +"values(#{userName},#{password},#{email},#{info},#{createTime})")
    @SelectKey(statement="SELECT LAST_INSERT_ID()",
               keyProperty="id",
               resultType=Integer.class,
               before = false)
    int insert(SysUser user);

和XML的方式对比 , 只是XML中的order属性换成了 before属性 , before = true时 , 功能和order = before一样 , before = false时, 功能和order = after一样

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

mybatis插入数据后返回自增主键ID的两种实现方式