MyBatis:获取插入记录的自增主键

为什么80%的码农都做不了架构师?>>>   hot3.png

我们知道MySql中主键可以设成auto_increment,也就是自增主键,每当新增一条记录,主键值自动加1。

在MyBatis中,执行insert,返回值为受影响的语句行数,此操作跟JDBC一致。

但是如果应用中需要将插入的值的主键返回时,这时有两种方式:

第一种方式:在mapper的insert节点中添加useGeneratedKey和keyProperty属性

第二种方式:在mapper的insert节点下添加selectKey子节点的方式

下面是具体的使用方式举例:


一、MySql中mybatis数据库的users表结构 

+----------+--------------+------+-----+---------+----------------+
| Field    | Type         | Null | Key | Default | Extra          |
+----------+--------------+------+-----+---------+----------------+
| id       | int(11)      | NO   | PRI | NULL    | auto_increment |
| name     | varchar(50)  | YES  |     | NULL    |                |
| password | varchar(255) | YES  |     | NULL    |                |
+----------+--------------+------+-----+---------+----------------+


二、MyBatis配置文件 mybatis-config.xml





	
	
	
		
		
		
		
	
	
	
	
	
	
	
		
	


	
	
	
		
			
			
			
			
			
			
			
				
				
				
				
			
		
	

	
		
	


三、MyBatis的mapper映射文件 UserMapper.xml




	
	
		insert into users(name, password) values(#{name}, #{password})	
	
	
	
	
		insert into users(name, password) values(#{name}, #{password})	
	

	
		
		
		
			select LAST_INSERT_ID()
		
		insert into users(name, password) values(#{name}, #{password})	
	


四、普通的Java实体类User

package cn.kolbe.mybatis.domain;

public class User {

	private Integer id;
	private String name;
	private String password;
	
	public User(String name, String password) {
		this.name = name;
		this.password = password;
	}
	
	public Integer getId() {
		return id;
	}
	public void setId(Integer id) {
		this.id = id;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public String getPassword() {
		return password;
	}
	public void setPassword(String password) {
		this.password = password;
	}
	@Override
	public String toString() {
		return "User [id=" + id + ", name=" + name + ", password=" + password + "]";
	}
	
}


五、测试代码

package cn.kolbe.mybatis;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.InputStream;

import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.Before;
import org.junit.Test;

import cn.kolbe.mybatis.domain.User;

public class MyBatisTest {
	
	/**
	 * 普通的插入操作
	 * 
	 * @throws FileNotFoundException
	 */
	@Test
	public void testGetGeneratedKey1() throws FileNotFoundException {
		SqlSession session = factory.openSession();
		String statement = "cn.kolbe.mybatis.domain.UserMapper.getGeneratedKey1";
		User user = new User("User_01","123456");
		int result = session.update(statement, user);		// result为受影响的行数
		session.commit();
		System.out.println(user);				// 输出 User [id=null, name=User_02, password=123456]
		System.out.println("result:" + result);                 // 输入 result: 1
		
	}
	
	/**
	 * 在insert中添加了 useGeneratedKeys="true" keyProperty="id" 属性的插入操作
	 * 
	 * @throws FileNotFoundException
	 */
	@Test
	public void testGetGeneratedKey2() throws FileNotFoundException {
		SqlSession session = factory.openSession();
		String statement = "cn.kolbe.mybatis.domain.UserMapper.getGeneratedKey2";
		User user = new User("User_02","123456");
		
		// 此时MyBatis不仅返回了受影响的行数,还将自增的id赋值给了传进去的user
		// 这时可以看到user打印出来的id是有值的!!!
		int result = session.update(statement, user);		// 受影响的行数
		session.commit();
		System.out.println(user);                               // 输出 User [id=2, name=User_02, password=123456]
		System.out.println("result:" + result);                 // 输出 result: 1
	}
	
	/**
	 * 在insert中添加子节点子节点的插入操作
	 * 
	 *		select select last_insert_id()
	 * 
	 * 
	 * @throws FileNotFoundException
	 */
	@Test
	public void testGetGeneratedKey3() throws FileNotFoundException {
		SqlSession session = factory.openSession();
		String statement = "cn.kolbe.mybatis.domain.UserMapper.getGeneratedKey3";
		User user = new User("User_03", "123456");
		
		// 此时MyBatis不仅返回了受影响的行数,还将自增的id赋值给了传进去的user
		// 这时可以看到user打印出来的id是有值的!!!
		int result = session.update(statement, user);		// 受影响的行数
		session.commit();
		System.out.println(user);                               // 输出 User [id=3, name=User_03, password=123456]
		System.out.println("result:" + result);                 // 输出 result: 1
	}
	
	/**
	 * 初始化SessionFactory
	 * @throws Exception
	 */
	@Before
	public void init() throws Exception {
		InputStream in = new FileInputStream("src/main/java/mybatis-config.xml");
		factory = new SqlSessionFactoryBuilder().build(in);
	}
	
	private SqlSessionFactory factory;
}


注:

1)使用selectKey的方式获取主键,相当于调用了MySql的LAST_INSERT_ID()函数,意思是查询最后一次插入的ID,相当于执行了两条语句

2)两种方式的主键获取方式都是通过MyBatis为我们自动将id值赋值给传入的javaBean,而不是通过update语句返回,也就是update返回的同jdbc一致(影响的行数),获取id的方式是执行完update后调用user.getId()方法,具体可参看上面的例子。


转载于:https://my.oschina.net/kolbe/blog/512904

你可能感兴趣的:(MyBatis:获取插入记录的自增主键)