Mybatis实现InsertOrUpdate功能

需求

最近在项目开发中遇到这样一个需求：每天需要对相同的数据（也有可能是不同的）进行两次入库操作，数据不存在便insert，存在则update。于是就用到了Mybatis的InsertOrUpdate功能。

实现

每次操作数据库之前，先根据id查询有没有记录，有则进行update操作，没有则进行insert操作。

model类代码如下。其中count为非业务字段（也不是表sheet中的字段），只是方便Mybatis进行insertOrUpdate操作的附加字段。

import lombok.Data;

@Data
public class Sheet {

    /**
     * 主键
     */
    private String id;
    /**
     * 客户姓名
     */
    private String customerName;
    /**
     * 。。。省略其他字段
     */

    /**
     * 该字段为非业务字段。Mybatis配置文件需要要到该字段，方便进行insertOrUpdate操作
     */
    private int count;
}

Mybatis的mapper.xml配置文件代码如下。代码含义：先执行selectKey语句，把结果赋值给Sheet类的count属性。如果count大于0，表示记录已存在，则进行update操作。如果count等于0，表示没有记录，则进行insert操作。

<update id="insertOrUpdate" parameterType="Sheet" >
        <selectKey keyProperty="count" resultType="int" order="BEFORE">
            select count(1) from sheet where ID= #{id}
        </selectKey>
        <if test="count > 0">
            update sheet 
            <set>
                <if test="customerName != null and customerName != ''">
                    CUSTOMER_NAME= #{customerName},
                </if>
            </set>
            where ID = #{id}
        </if>
        <if test="count==0">
            insert into sheet
            <trim prefix="(" suffix=")" suffixOverrides=",">
                ID,
                <if test="customerName != null and customerName != ''">
                    CUSTOMER_NAME,
                </if>
            </trim>
            <trim prefix="values (" suffix=")" suffixOverrides=",">
                #{id},
                <if test="customerName != null and customerName != ''">
                    #{customerName},
                </if>
            </trim>
        </if>
    </update>

selectKey标签可以给update标签中的parameterType属性（model类）对应的对象设置属性值。selectKey标签的属性描述：

keyProperty：selectKey 语句结果应该被设置的目标属性。此处对应的就是Sheet类的count属性。

resultType：结果的类型，此处为属性count的类型。

order：可以被设置为 BEFORE 或 AFTER。BEFORE表示先执行selectKey语句，后执行update语句；AFTER表示先执行update语句，后执行selectKey语句。