mybatis插入數(shù)據(jù)后返回自增主鍵ID

開發(fā)過程中我們經(jīng)常性的會(huì)用到許多的中間表，mybatis插入數(shù)據(jù)后返回自增主鍵的兩種方式，

用于數(shù)據(jù)之間的對應(yīng)和關(guān)聯(lián)，這個(gè)時(shí)候我們關(guān)聯(lián)最多的就是ID，我們在一張表中插入數(shù)據(jù)后級(jí)聯(lián)增加到關(guān)聯(lián)表中，我們數(shù)值的mybatis在插入數(shù)據(jù)后返回的是插入成功的條數(shù)，那么這個(gè)時(shí)候我們想要得到相應(yīng)的新增數(shù)據(jù)的ID，該怎么辦呢？

下面我們介紹二種方式來實(shí)現(xiàn)，一種是在xml文件種設(shè)置屬性，另外一種就是使用注解來實(shí)現(xiàn)。

整體兩種方法流程

1.使用注解@SelectKey,繼承原生的方法insert來實(shí)現(xiàn)

這種方法是使用注解，不使用xml方法，使用繼承原生的方法insert來實(shí)現(xiàn)，在注解里面寫號(hào)SQL以及每個(gè)字段綁定的變量即可。

注解的方法為

@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，這種方式需要把數(shù)據(jù)庫表的id設(shè)置成自增ID，這樣注解就生效了

在使用的時(shí)候直接用

result = getBaseMapper().insert(user);

最后也得到了這個(gè) 自增的ID,當(dāng)然了，我這個(gè)用的是pgsql,如果是mysql，在執(zhí)行select LAST_INSERT_ID()得到的結(jié)果那么注解就得改一下：

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

其中select LAST_INSERT_ID()就是mysql插入數(shù)據(jù)后，獲取的當(dāng)前的自增id。

2.在xml中定義useGeneratedKeys為true,返回主鍵id的值

keyProperty和keyColum分別代表

數(shù)據(jù)庫記錄主鍵字段和java對象成員屬性名：在XML文件中要設(shè)置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自動(dòng)生成主鍵，返回自增主鍵 insert語句

對于表的主鍵自動(dòng)生成的 insert語句 , 在不需要返回自增主鍵時(shí) , 就和普通的insert語句沒有區(qū)別 , 在需要的返回的時(shí)候 , 情況稍微有點(diǎn)不一樣 , 故在此記錄一下 . 在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 '創(chuàng)建時(shí)間',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='用戶表';

在使用主鍵自增的時(shí)候 ,  插入數(shù)據(jù)庫后可能需要得到自增的主鍵值 , 然后使用這個(gè)值進(jìn)行一些其他的操作 .

有些數(shù)據(jù)庫提供主鍵自增(MYSQL , SQLSERVER) , 有些不提供(ORACLE) , 而是使用序列得到一個(gè)值 , 然后將這個(gè)值賦給 id , 再講數(shù)據(jù)插入.

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設(shè)置為true后 , MyBatis會(huì)使用JDBC的getGeneratedKeys方法取出由數(shù)據(jù)庫內(nèi)部生成的主鍵 .

獲得主鍵值之后將其賦給keyProperty配置的 id 屬性 .

當(dāng)需要設(shè)置多個(gè)屬性時(shí) , 使用逗號(hào)隔開 , 這種情況下 還需要配置 keyColumn屬性 , 按順序指定數(shù)據(jù)庫的列 , 這里列的值會(huì)和 keyProperty 配置的屬性一一對應(yīng) .

由于要使用數(shù)據(jù)庫返回的主鍵值 , 所以SQL上下兩部分的列中去掉了 id 列和對于的 #{id} 屬性

• 1.2返回非自增主鍵

采用 <selectKey>標(biāo)簽獲取主鍵的值 , 這種方式對提供和不提供主鍵自增功能的數(shù)據(jù)庫同樣適用

<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標(biāo)簽的 keyColumn,keyProperty和上面useGeneratedKeys的用法和含義一樣 , 這里的resultType用于設(shè)置返回值類型.

order 屬性和數(shù)據(jù)庫相關(guān) , 在MYSQL 中 , order是AFTER , 因?yàn)楫?dāng)前及記錄的主鍵值在insert語句執(zhí)行成功之后才能拿到 , 而在ORACLE中，oder是BEFORE , 因?yàn)镺RACLE需要先從序列取到值 , 再將其作為主鍵插入到數(shù)據(jù)庫

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 注解之外 , 多了一個(gè) @Option 注解 , 在這個(gè)注解上配置了useGeneratedKeys 和 keyProperty屬性，用法和 XML 的方式一樣 , 當(dāng)需要配置多個(gè)列的時(shí)候 , 這個(gè)注解也提供了 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時(shí) , 功能和order = before一樣 , before = false時(shí), 功能和order = after一樣

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評論