【MyBatis框架点滴】——mybatis插入数据返回主键（mysql、oracle）

迈不过友情╰ 2022-07-24 07:11 198阅读 0赞

向数据库中插入数据时，大多数情况都会使用自增列或者UUID做为主键。主键的值都是插入之前无法知道的，但很多情况下我们在插入数据后需要使用刚刚插入数据的主键，比如向两张关联表A、B中插入数据（A的主键是B的外键），向A表中插入数据之后，向B表中插入数据时需要用到A的主键。

比如添加一个用户，同时返回插入用户后得到的用户id：

/** * 添加用户信息 * @param user * @throws Exception */
        public int insertUser(User user) throws Exception {
            SqlSession session=sqlSessionFactory.openSession();
            session.insert("com.danny.mybatis.insertUser", user);
            session.commit();
            return user.getUserId();//返回插入数据库后得到的用户id
        }

这里总结一下mybatis插入数据时返回主键的4种情况：mysql环境下主键自增、mysql环境下主键为uuid、mysql环境下主键自增、mysql环境下主键为uuid。

以下全文均以User实体来举例说明，字段有userId、userName、sex、birthday、address 五个属性，其中userId有可能是int类型，也有可能是String类型。

数据库为mysql

--------------------

主键为自增时（主键为数值类型且自增）

利用mysql的`LAST_INSERT_ID()`方法获取插入记录的主键，`select LAST_INSERT_ID()`可以在插入数据后，查询并返回刚插入数据的主键（但是单独执行这条语句只会返回0）。

<insert id="insertUser" parameterType="com.danny.mybatis.po.User">
            <selectKey keyProperty="userId" order="AFTER" resultType="java.lang.Integer">
                select LAST_INSERT_ID()
            </selectKey>
            insert into T_USER(userName,birthday,sex,address) values (#{userName},#{birthday},#{sex},#{address})
        </insert>

parameterType：指定insert执行语句接收的参数类型为pojo（这里的user）。

keyProperty：将查询到的主键值设置到parameterType指定对象的哪个属性。

order：`<selectKey>` 标签内的sql语句相对于`insert`语句的执行顺序，`AFTER`表示`select LAST_INSERT_ID()` 这个语句将在`insert`语句之后执行。

主键为UUID时（主键必须为字符类型）

使用mysql的方法`UUID()`方法获取随机的UUID作为主键，`select UUID()`可以在插入数据前，生成随机的UUID并通过keyProperty赋值给将要插入记录的主键。

<insert id="insertPerson" parameterType="com.danny.mybatis.po.User">
            <selectKey keyProperty="userId" order="BEFORE" resultType="java.lang.String">
                select UUID()
            </selectKey>
            insert into user(userId,userName,birthday,sex,address) values (#{
       id},#{
       userName},#{
       birthday},#{
       sex},#{
       address})
        </insert>

在上述代码中可以看到`order` 的属性值为`BEFORE` ，说明在插入之前就已经生成了UUID，并且已经把UUID赋值给user的id。

背 景数据库为oracle：

主键为自增时（主键为数值类型）：

在oracle中实现主键自增，需要先创建序列，相当于创建一个全局变量，用来存储对应表的主键的当前最大值（主键为数值类型时）。

先为user表创建一个序列：

CREATE SEQUENCE USER_ID_SEQ INCREMENT BY 1 -- 每次递增1 START WITH 1 -- 从1开始 MINVALUE 1 -- 最小值=1 NOCYCLE; -- 不循环

利用`USER_ID_SEQ.NEXTVAL` 获得要插入数据的主键：

<insert id="insertUser" parameterType="com.danny.mybatis.po.User">
            <selectKey keyProperty="userId" order="BEFORE" resultType="java.lang.Integer">
                select USER_ID_SEQ.NEXTVAL as userId from DUAL
            </selectKey>
            insert into T_USER(userId,userName,birthday,sex,address) values (#{userId},#{userName},#{birthday},#{sex},#{address})
        </insert>

主键为UUID时

用oracle自带的`SYS_GUID()`方法获得随机的GUID作为主键：

<insert id="insertUser" parameterType="com.danny.mybatis.po.User">
            <selectKey keyProperty="userId" order="BEFORE" resultType="java.lang.Integer">
                select SYS_GUID() as userId from DUAL
            </selectKey>
            insert into T_USER(userId,userName,birthday,sex,address) values (#{userId},#{userName},#{birthday},#{sex},#{address})
        </insert>