MyBatis 教程

insert 返回插入数据 ID

本章节将介绍怎样使用 <insert> 插入一条数据到数据库,然后返回生成的数据主键 ID。

数据库支持自动生成主键

如果你的数据库支持自动生成主键的字段(比如 MySQL 和 SQL Server),那么你可以在 <insert> 标签中设置 useGeneratedKeys="true",然后再把 keyProperty 设置为目标属性就 OK 了。例如:

<insert id="save" useGeneratedKeys="true" keyProperty="userId">
    INSERT INTO `user`(
        `name`, `sex`, `age`
    ) VALUES (
        #{name,jdbcType=VARCHAR},
        #{sex,jdbcType=INTEGER},
        #{age,jdbcType=VARCHAR}
    )
</insert>

上面实例中,useGeneratedKeys="true" 将会告诉 MyBatis 使用 JDBC 的 getGeneratedKeys 方法来取出由数据(比如:像 MySQL和 SQL Server 这样的数据库管理系统的自动递增字段)内部生成的主键;而 keyProperty="userId" 表示将使用 JavaBean 中的 userId 属性来存放数据库内部生成的主键;更多 insert 属性请阅读“insert 标签介绍

数据库不支持自动生成主键

上面介绍怎样对支持自动生成主键的数据库插入数据,且返回数据库生成的ID。下面将介绍对那些不支持自动生成主键的数据库,如:Oracle。实例:

<insert id="insert06" useGeneratedKeys="false">
    <!-- 生成自定义的主键,这里使用当前时间毫秒数和0到 99999999的随机数 -->
    <selectKey keyProperty="userId" resultType="int" order="BEFORE">
        SELECT (UNIX_TIMESTAMP(NOW()) + FLOOR(RAND()*99999999)) FROM DUAL
    </selectKey>
    INSERT INTO `user`(
        `user_id`, `name`, `sex`, `age`
    ) VALUES (
        #{userId, jdbcType=INTEGER},
        #{name, jdbcType=VARCHAR},
        #{sex, jdbcType=VARCHAR},
        #{age, jdbcType=INTEGER}
    )
</insert>

上面实例使用 <selectKey> 标签实现自动生成主键。其中:

  • order 属性表示该语句是在插入语句之前还是之后执行(BEFORE - 之前;AFTER - 之后)

  • resultType 定义结果集类型

  • keyProperty 主键属性名称

完整示例

(1)MyBatis 配置文件 mybatis-cfg.xml 内容如下:

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
  "https://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <properties resource="database.properties"/>
    <environments default="MySqlDatabase" >
        <environment id="MySqlDatabase" >
            <transactionManager type="JDBC" />
            <dataSource type="POOLED">
                <property name="driver" value="${jdbc.driver}"/>
                <property name="url" value="${jdbc.url}"/>
                <property name="username" value="${jdbc.username}"/>
                <property name="password" value="${jdbc.password}"/>
            </dataSource>
        </environment>
    </environments>
    <mappers>
        <mapper resource="com/hxstrive/mybatis/insert/demo3/UserMapper.xml" />
    </mappers>
</configuration>

(2)Mapper 接口定义如下:

package com.hxstrive.mybatis.insert.demo3;

public interface UserMapper {
    int save(UserBean userBean);
}

(3)Mapper XML 文件定义如下:

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
   "https://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.hxstrive.mybatis.insert.demo3.UserMapper">

    <insert id="save" useGeneratedKeys="true" keyProperty="userId">
        INSERT INTO `user`(
            `name`, `sex`, `age`
        ) VALUES (
            #{name,jdbcType=VARCHAR},
            #{sex,jdbcType=INTEGER},
            #{age,jdbcType=VARCHAR}
        )
    </insert>

</mapper>

(4)JavaBean 代码如下:

package com.hxstrive.mybatis.insert.demo3;

public class UserBean {
   private Integer userId;
   private String name;
   private String sex;
   private Integer age;

   // 忽略 getter 和 setter

   @Override
   public String toString() {
      return "UserBean{" +
            "userId=" + userId +
            ", name='" + name + '\'' +
            ", sex='" + sex + '\'' +
            ", age=" + age +
            '}';
   }
}

(5)客户端代码:

package com.hxstrive.mybatis.insert.demo3;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import java.io.InputStream;

public class InsertDemo3 {

   public static void main(String[] args) throws Exception {
      String cfgName = "com/hxstrive/mybatis/insert/demo3/mybatis-cfg.xml";
      InputStream input = Resources.getResourceAsStream(cfgName);
      SqlSessionFactoryBuilder factoryBuilder = new SqlSessionFactoryBuilder();
      SqlSessionFactory sqlFactory = factoryBuilder.build(input);
      SqlSession sqlSession = sqlFactory.openSession(true);
      UserMapper userMapper = sqlSession.getMapper(UserMapper.class);

      UserBean user = new UserBean();
      user.setName("Name-" + System.currentTimeMillis());
      user.setSex("female");
      user.setAge(25);

      int result = userMapper.save(user);
      System.out.println("result=" + result);
      System.out.println("user id is " + user.getUserId());
   }

}
说说我的看法
全部评论(
没有评论
关于
本网站专注于 Java、数据库(MySQL、Oracle)、Linux、软件架构及大数据等多领域技术知识分享。涵盖丰富的原创与精选技术文章,助力技术传播与交流。无论是技术新手渴望入门,还是资深开发者寻求进阶,这里都能为您提供深度见解与实用经验,让复杂编码变得轻松易懂,携手共赴技术提升新高度。如有侵权,请来信告知:hxstrive@outlook.com
公众号