MyBatis之insert——主键自增和自定义主键

比眉伴天荒 2022-06-10 00:35 2344阅读 0赞

一、实现描述：我们在写程序时，时常会遇到这样的问题：插入一条数据，但因有主键的存在，而有时候会报“违反唯一性约束”的错误，下面呢，我们就来利用MyBatis来解决这个问题。

二、此文使用的数据库为MySQL，如需要使用其他数据库，请做适当更改即可。

三、根据主键的自增在MyBatis中进行配置实现，实现如下：

<insert id="insertRole" parameterType="role" useGeneratedKeys="true" keyProperty="id">
       insert into t_role(role_name,note) values (#{roleName},#{note})
    </insert>

解析：

1、我们使用keyProperty属性指定id为主键字段，同时使用useGeneratedKeys属性告诉MyBatis这个主键是否使用数据库的内置规则生成。

2、其中，在数据库中设置id的自增规则，如下所示：

mysql>alter table t_role change id id int primary key auto_increment;

3、在上面配置中，我们指定了id列为自增字段，因此我们建立POJO，这样的话，使用MyBatis的主键回填功能，实现正确的插入操作（无需设置id的值，MyBatis会用数据库的设置进行处理）。  
4、测试：

sqlSession = SqlSessionFactoryUtil.openSqlSession();
    RoleMapper roleMapper = sqlSession.getMapper(RoleMapper.class);
    Role role = new Role();
    role.setRoleName("test4");
    role.setNote("test4Note");
    roleMapper.insertRole(role);
    System.err.println(role.getId());

四、自定义主键生成规则实现

实例如下：

<insert id="insertRole" parameterType="role" useGeneratedKeys="true" keyProperty="id">
        <selectKey keyProperty="id" resultTyle="int" order="BEFORE">
                 select if(max(id) is null, 1, max(id) + 2) as newId from t_role
         </selectKey>
         insert into t_role(id,role_name,note) values(#{id},#{roleName},#{note});
    </insert>

解析：上面显示的为MyBatis定义自增规则的配置，其实上面的selectKey配置类似于数据库中的触发器，order=“BEFORE”规定了在执行语句之前执行newId的查找，然后在执行插入语句。

这样我们就利用自定义的主键规则实现了数据的新增。