MyBatis 实现主键回写

当我们向数据库插入数据时，主键是设置的自增的，但是我们插入好数据库后，我们希望得到刚刚插入的数据的主键，mybatis可以实现这样的功能 应用场景： 员工与部门，当新员工进公司，该员工肯定属于某个部门，插入数据的时候，要告诉部门表，这个员工属于哪个部门

数据库表

mapper.xml文件代码

<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> <mapper namespace="UserDaoMapper"> <!-- 插入statement useGeneratedKeys：开启主键回写 keyColumn：主键列名(既数据库表中的列名) keyProperty：主键对应的属性名（实体中的属性名） 插入数据时，传过来的参数是一个对象，#{***}中的名字必须与对象中属性名一致 --> <insert id="saveUser" useGeneratedKeys="true" keyColumn="id" keyProperty="id"> INSERT INTO tb_user ( user_name, password, name, age, sex, birthday, created, updated ) VALUES( #{userName}, #{password}, #{name}, #{age}, #{sex}, #{birthday}, NOW(), NOW() ) </insert> </mapper>

mybatis-config.xml文件代码

<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN" "http://mybatis.org/dtd/mybatis-3-config.dtd"> <configuration> <!-- 引入外部属性资源 --> <properties resource="jdbc.properties" /> <settings> <!-- 设置驼峰匹配 --> <setting name="mapUnderscoreToCamelCase" value="true" /> </settings> <typeAliases> <!-- package 扫描该包下的类，别名就是类名--> <package name="cn.itcast.domain"/> </typeAliases> <environments default="development" > <environment id="development"> <transactionManager type="JDBC" /> <dataSource type="POOLED"> <!-- 指定驱动类 --> <property name="driver" value="${jdbc.driverClass}" /> <!-- 指定连接地址 --> <property name="url" value="${jdbc.url}" /> <!-- 指定数据库登录用户 --> <property name="username" value="${jdbc.username}" /> <!-- 数据库密码 --> <property name="password" value="${jdbc.password}" /> </dataSource> </environment> </environments> <!-- 引入map文件 --> <mappers> <mapper resource="UserDaoMapper.xml"/> </mappers> </configuration>

jdbc.properties文件

jdbc.driverClass=com.mysql.jdbc.Driver jdbc.url=jdbc:mysql://localhost:3306/mybatis jdbc.username=root jdbc.password=123

Test测试类

public class UserDaoTest { private SqlSession sqlSession; @Before public void setUp() throws IOException { // 加载mybatis-config.xml文件 InputStream is = Resources.getResourceAsStream("mybatis-config.xml"); // 创建sqlSessionFactory的构造器 SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder(); // 创建sqlSession工厂 SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBuilder.build(is); // 通过工厂生成sqlSession sqlSession = sqlSessionFactory.openSession(); } @Test /** * 插入用户信息 * @throws IOException */ public void inserUser() throws IOException { UserDao dao = new UserDaoImpl(sqlSession); User user =new User(); user.setUserName("jiumozhi"); user.setPassword("123456"); user.setName("鸠摩智"); user.setAge(30); user.setSex(1); user.setBirthday(new Date()); dao.saveUser(user); System.out.println(user); } }

dao层方法

@Override public void saveUser(User user) { sqlSession.insert("UserDaoMapper.saveUser", user); sqlSession.commit(); sqlSession.close(); }

展示结果

image.png

---

源作者：顾国玉 原文：https://blog.csdn.net/u010452388/article/details/80822657