MyBatis新增数据并返回主键值方式

2023-03-06 09:02

短信预约 -IT技能 免费直播动态提醒

MyBatis新增数据并返回主键值

虽然这个功能比较简单，网上帖子也很多，但是有一个地方有点坑，这里做一个对比，作为脱坑的标记。

为了测试，写一个简单的添加功能验证一下，直接看效果。

entity

@Component("user")
public class User {
	private int id;
	private String usercode;
	private String password;
	private String name;
	private String phone;
	private int status;

controller层

@ResponseBody
@RequestMapping(method = RequestMethod.POST,value = "/register")
public int register(User user){
		System.out.println("参数:"+user.toString());
		int id = userSer.register(user);
		System.out.println("id:"+id);
		System.out.println(""+user.getId());
		return id;
}

UserDao

//用户注册
int register(User user);

UserMapper.xml

<insert id="register" parameterType="com.hans.entity.User" useGeneratedKeys="true" keyProperty="id">
    	INSERT INTO USER (
			usercode,
			PASSWORD,
			phone,
			NAME
		)
		VALUES
			(#{phone},#{password},#{phone},#{name})
</insert>

程序运行添加数据，看到数据库数据如下：

再看下，控制台输出情况

可以很清晰的看出，虽然sql语句中并没有声明resultType，但影响的行数(0/1)其实还是赋值给了sql的运行结果，所以控制台的id：1就不难理解了，真正插入操作的主键其实是赋值给实体类User的字段id中。

结论很明显，添加方法返回的是影响的行数(0/1)。

插入数据的主键值其实是赋值给你指定的entity的某个字段中。

核心代码：

useGeneratedKeys="true" keyProperty="id"

MyBatis新增更新返回主键

在往sqlserver数据库新增、修改数据后，可能需要紧接着进行其他操作，比如将数据写入redis，此时需要该条数据的id作为hash结构中的key。

而当sql设置主键自增时，传来的参数是不带id的，所有需要在dao层执行新增更新操作后返回主键。

dao层

新增操作，使用useGeneratedKeys=“true” keyProperty=“id”

   <insert id="addFormula" parameterType="testmaven06calculation.com.cal.res.pojo.CalData" useGeneratedKeys="true" keyProperty="id">
        insert into cal_data(formula,standard,msg)values(#{formula},#{standard},#{msg});
    </insert>

更新操作

    <update id="updateFormula" parameterType="testmaven06calculation.com.cal.res.pojo.CalData" >
    <selectKey keyProperty="id" resultType="int" order="AFTER">
        select id from cal_data where formula=#{formula};
    </selectKey>
        update cal_data set standard=#{standard},msg=#{msg} where formula=#{formula};
    </update>

service层

    @Transactional(rollbackFor = Exception.class)
    public AjaxMessage addUpdateFormula(CalData calData) {
        redisTemplate.setEnableTransactionSupport(true);
        // 检查公式是否存在
        if (existFormula(calData.getFormula())) { // 公式存在
            // 执行更新操作
            try {
                redisTemplate.multi();
                calDataDao.updateFormula(calData);
                redisTemplate.opsForHash().put("calDataHash", String.valueOf(calData.getId()), calData);
                redisTemplate.exec();
                return new AjaxMessage(true, "公式已存在，更新成功", null);
            } catch (Exception e) {
                redisTemplate.discard();
                TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
                return new AjaxMessage(false, "公式已存在，更新失败", null);
            }
        } else { // 公式不存在
            // 执行新增操作
            try {
                redisTemplate.multi();
                calDataDao.addFormula(calData);
                redisTemplate.opsForSet().add("formulaSet", calData.getFormula());
                redisTemplate.opsForHash().put("calDataHash", String.valueOf(calData.getId()), calData);
                redisTemplate.exec();
                return new AjaxMessage(true, "公式不存在，新增成功", null);
            } catch (Exception e) {
                redisTemplate.discard();
                TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
                return new AjaxMessage(false, "公式不存在，新增失败", null);
            }
        }
    }