mybatis中useGeneratedKeys用法--插入数据库后获取主键值

Posted yuandian8

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了mybatis中useGeneratedKeys用法--插入数据库后获取主键值相关的知识,希望对你有一定的参考价值。

前言:今天无意在mapper文件中看到useGeneratedKeys这个词,好奇就查了下,发现能解决我之前插入有外键表数据时,这个外键获取繁琐的问题,于是学习敲DEMO记录

     在项目中经常需要获取到插入数据的主键来保障后续操作,数据库中主键一般我们使用自增或者uuid()的方式自动生成

问题:对于uuid使用Java代码生成的方式还比较容易控制,然而使用数据库生成的主键,这样我们就需要将插入的数据再查询出来得到主键,某些情况下还可能查询到多条情况,这样就比较尴尬了。

   那有什么办法来插入数据的时候就得到这个主键呢?

 

以下是Demo,分别定义了一个主键自增的表,一个uuid主键的表,对应实体Student,和UUidStudent分别说明两种情况下插入数据如何获取到主键值

一、自增对应实体Student

package com.ydcc.model;

public class Student
{

    private String id;
    private String name;
    private Integer age;

}

对应的mapper.xml如下

<mapper namespace="com.ydcc.mappers.StudentMapper">

    <resultMap type="Student" id="StudentResult">
        <id property="id" column="id"/>
        <result property="name" column="name"/>
        <result property="age" column="age"/>
    </resultMap>

    <insert id="add" parameterType="Student"  useGeneratedKeys="true" keyProperty="id" ><!--keyProperty对应的是实体中的属性不是数据库-->
        insert into t_student (name,age) values (#{name},#{age})
    </insert>

</mapper> 

 

二、uuid非自增对应实体:UUidStudent

package com.ydcc.model;

public class UUidStudent
{
    private String uuid;
    private String name;
    private int age;
}

对应的xml如下

<mapper namespace="com.ydcc.mappers.UUidStudentMapper">

    <resultMap type="Student" id="StudentResult">
        <id property="uuid" column="uuid"/>
        <result property="name" column="name"/>
        <result property="age" column="age"/>
    </resultMap>

    <insert id="add" parameterType="Student" useGeneratedKeys="true" keyProperty="uuid">
        <!-- 
     <selectKey resultType="String" order="BEFORE" keyProperty="uuid">    SELECT uuid() </selectKey> insert into t_uuidstudent (uuid,name,age) values (#{uuid},#{name},#{age})
    
--> <selectKey resultType="String" order="AFTER" keyProperty="uuid"> <!--keyProperty对应的是实体中的属性不是数据库-->   SELECT uuid() </selectKey> insert into t_uuidstudent (uuid,name,age) values ( uuid(), #{name},#{age}) </insert> </mapper>

 

来看以下测试代码和运行效果

package com.ydcc.service;


import org.apache.ibatis.session.SqlSession;
import org.apache.log4j.Logger;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.ydcc.mappers.StudentMapper;
import com.ydcc.mappers.UUidStudentMapper;
import com.ydcc.model.Student;
import com.ydcc.model.UUidStudent;
import com.ydcc.util.SqlSessionFactoryUtil;

public class StudentTest2
{

    private static Logger logger = Logger.getLogger(StudentTest2.class);
    private SqlSession sqlSession = null;
    private StudentMapper studentMapper = null;
    private UUidStudentMapper uuidstudentMapper = null;

    /**
     * 测试方法前调用
     * 
     * @throws Exception
     */
    @Before
    public void setUp() throws Exception
    {
        sqlSession = SqlSessionFactoryUtil.openSession();
        studentMapper = sqlSession.getMapper(StudentMapper.class);
        uuidstudentMapper = sqlSession.getMapper(UUidStudentMapper.class);
    }

    /**
     * 测试方法后调用
     * 
     * @throws Exception
     */
    @After
    public void tearDown() throws Exception
    {
        sqlSession.close();
    }

    @Test
    public void testAdd()
    {
        logger.info("添加学生");

        Student student = new Student();
        student.setName("王五");
        student.setAge(12);

        studentMapper.add(student);

        logger.info("Student是:" + student.toString());
        sqlSession.commit();
    }

    @Test
    public void testAddUUidStudent()
    {
        logger.info("添加uuid学生");

        UUidStudent uuidStudent = new UUidStudent();
        uuidStudent.setName("uuid王五");
        uuidStudent.setAge(12);

        uuidstudentMapper.add(uuidStudent);

        logger.info("uuidStudent是:" + uuidStudent.toString());
        sqlSession.commit();
    }

}

  运行结果:

  [main] INFO com.ydcc.service.StudentTest2 - 添加uuid学生

  [main] INFO com.ydcc.service.StudentTest2 - uuidStudent是:UUidStudent [uuid=6885de71-c32d-11e8-88e1-0c9d920ff4e1, name=uuid王五, age=12]
  [main] INFO com.ydcc.service.StudentTest2 - 添加学生
  [main] INFO com.ydcc.service.StudentTest2 - Student是:Student [id=9, name=王五, age=12]

 

   经过与插入数据对比,发现数据无误,这里总结注意的点

  1、keyProperty对应的是实体中的属性,不是数据库字段

  2、对于非自增主键,必须显式写出     

  <selectKey resultType="java.lang.String" order="AFTER" keyProperty="uuid">

     SELECT uuid()

  </selectKey>

   order的设置,在insert sql语句执行前(BEFORE)或者执行后(AFTER),
  上面注释掉的代码用的before,看测试代码中我并没有对uuidStudent进行uuid set值,但是insert sql中缺用到#{uuid},理解为在执行前已经将主键id值赋给实体,直接取值就行.经过测试如果把#{uuid}换成uuid(),就发现得到的主键和入库的数据不一致
  使用after,可以看到使用了数据库的uuid(),那么在插入数据后,把这个值赋给了实体uuidStudent.

  3、对于自增的设计,我们直接用useGeneratedKeys="true" keyProperty="id"正常些插入语句即可

附上标签中几个字段的解释:

技术分享图片




敲demo过程中因为用的电脑上之前的mysql-connector包,连接数据库时报Unknown character set index for field ‘255‘ received from server.的错,经查,因为是我本地用的mysql8的版本高、jar包版本低的原因
MYSQL 5.5 之前, UTF8 编码只支持1-3个字节;从MYSQL5.5开始,可支持4个字节UTF编码utf8mb4,升级下jar包就解决了



 

  













以上是关于mybatis中useGeneratedKeys用法--插入数据库后获取主键值的主要内容,如果未能解决你的问题,请参考以下文章

mybatis中useGeneratedKeys和keyProperty的使用

mybatis中useGeneratedKeys和keyProperty的作用

Mybatis插入语句useGeneratedKeys="true"的用法

Mybatis插入语句useGeneratedKeys="true"的用法

Mybatis配置文件之useGeneratedKeys和keyProperty搭配用法

Mybatis配置文件之useGeneratedKeys和keyProperty搭配用法