Mybatis 动态SQL – 使用choose标签动态生成更新语句

之前我们介绍了if,set标签的使用；本篇我们需要在if,set标签的基础上介绍如何使用Mybatis提供的choose标签动态生成更新语句。

如果您对if,set标签动态生成更新语句不太了解，建议您先进行了解后再阅读本篇，可以参考：

Mybatis 动态SQL – 使用if,set标签动态生成更新语句https://blog.csdn.net/m1729339749/article/details/132636964

一、数据准备

这里我们直接使用脚本初始化数据库中的数据

-- 如果数据库不存在则创建数据库
CREATE DATABASE IF NOT EXISTS demo DEFAULT CHARSET utf8;
-- 切换数据库
USE demo;
-- 创建用户表
CREATE TABLE IF NOT EXISTS T_TEACHER(
  ID INT PRIMARY KEY COMMENT '教师编号',
  TEACHER_NAME VARCHAR(64) NOT NULL COMMENT '教师名称',
  DEPARTMENT VARCHAR(16) NOT NULL COMMENT '所属部门',
  BIRTH DATE NOT NULL COMMENT '出生年月',
  DEGREE VARCHAR(16) NOT NULL COMMENT '学历(ZK:专科, BK:本科, YJS:研究生, BS:博士)'
);
-- 插入用户数据
INSERT INTO T_TEACHER(ID, TEACHER_NAME, DEPARTMENT, BIRTH, DEGREE)
VALUES(1, '张三1', '001', '1990-06-12', 'BK'),
      (2, '李四1', '002', '1992-05-10', 'BK'),
      (3, '张三2', '003', '1988-01-15', 'YJS'),
      (4, '李四2', '001', '1979-03-10', 'BK'),
      (5, '李四3', '003', '1995-08-16', 'YJS');

创建了一个名称为demo的数据库；并在库里创建了名称为T_TEACHER的教师表并向表中插入了数据

二、环境搭建

1、创建实体类

在cn.horse.demo下创建TeacherInfo实体类：

TeacherInfo类：

package cn.horse.demo;

import java.time.LocalDate;

public class TeacherInfo {
    private Integer id;
    private String name;
    private String department;
    private LocalDate birth;
    private String degree;

    @Override
    public String toString() {
        StringBuilder stringBuilder = new StringBuilder();
        stringBuilder.append("{ ");
        stringBuilder.append("id: ");
        stringBuilder.append(this.id);
        stringBuilder.append(", ");
        stringBuilder.append("name: ");
        stringBuilder.append(this.name);
        stringBuilder.append(", ");
        stringBuilder.append("department: ");
        stringBuilder.append(this.department);
        stringBuilder.append(", ");
        stringBuilder.append("birth: ");
        stringBuilder.append(this.birth);
        stringBuilder.append(", ");
        stringBuilder.append("degree: ");
        stringBuilder.append(this.degree);
        stringBuilder.append(" }");
        return stringBuilder.toString();
    }
}

2、Mapper配置文件

在resources的目录下新建TeacherInfoMapper.xml配置文件

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "https://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.horse.demo.TeacherInfoMapper">

    <select id="findById" resultType="cn.horse.demo.TeacherInfo">
        SELECT
            ID,
            TEACHER_NAME name,
            DEPARTMENT,
            BIRTH,
            DEGREE
        FROM T_TEACHER
        WHERE ID = #{id}
    </select>
</mapper>

3、引入配置文件

在resources下新建mybatis-config.xml配置文件，并引入TeacherInfoMapper.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>
    <settings>
        <setting name="logImpl" value="JDK_LOGGING"/>
    </settings>

    <environments default="development">
        <environment id="development">
            <transactionManager type="JDBC"/>
            <dataSource type="POOLED">
                <property name="driver" value="org.gjt.mm.mysql.Driver"/>
                <property name="url" value="jdbc:mysql://localhost:3306/demo?useUnicode=true&amp;useSSL=false&amp;characterEncoding=utf8"/>
                <property name="username" value="root"/>
                <property name="password" value="horse"/>
            </dataSource>
        </environment>
    </environments>

    <mappers>
        <mapper resource="demo/TeacherInfoMapper.xml" />
    </mappers>
</configuration>

4、会话工具类

在cn.horse.demo包下新建SqlSessionUtils工具类

package cn.horse.demo;

import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;

import java.io.InputStream;
import java.util.Objects;

public class SqlSessionUtils {

    private static final SqlSessionFactory sqlSessionFactory;
    static {
        // 读取mybatis配置文件
        InputStream inputStream = ClassLoader.getSystemClassLoader().getResourceAsStream("mybatis-config.xml");
        // 根据配置创建SqlSession工厂
        sqlSessionFactory = new SqlSessionFactoryBuilder()
                .build(inputStream);
    }

    /**
     * 开启会话
     * @return
     */
    public static SqlSession openSession() {
        return sqlSessionFactory.openSession();
    }

    /**
     * 关闭会话
     * @param sqlSession
     */
    public static void closeSession(SqlSession sqlSession) {
        if(Objects.nonNull(sqlSession)) {
            sqlSession.close();
        }
    }
}

5、JDK 日志系统配置

在resources的目录下新建logging.properties配置文件

handlers=java.util.logging.ConsoleHandler
.level=INFO

cn.horse.demo.TeacherInfoMapper.level=FINER
java.util.logging.ConsoleHandler.level=ALL
java.util.logging.ConsoleHandler.formatter=java.util.logging.SimpleFormatter
java.util.logging.SimpleFormatter.format=%1$tY-%1$tm-%1$td %1$tT.%1$tL %4$s %3$s - %5$s%6$s%n

在cn.horse.demo下新建JdkLogConfig类：

JdkLogConfig类：

package cn.horse.demo;

import java.io.IOException;
import java.io.InputStream;
import java.util.logging.LogManager;

public class JdkLogConfig {

    public JdkLogConfig() {
        try {
            InputStream inputStream = ClassLoader.getSystemClassLoader().getResourceAsStream("logging.properties");
            LogManager.getLogManager().readConfiguration(inputStream);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
}

6、启动程序

package cn.horse.demo;

import org.apache.ibatis.session.SqlSession;

import java.util.List;

public class Main {
    public static void main(String[] args) {
        // 引入JDK日志配置
        System.setProperty("java.util.logging.config.class", "cn.horse.demo.JdkLogConfig");

        findById("cn.horse.demo.TeacherInfoMapper.findById", 1);
    }

    private static void findById(String statement, Integer id) {
        SqlSession sqlSession = null;
        try {
            sqlSession = SqlSessionUtils.openSession();
            TeacherInfo teacherInfo = sqlSession.selectOne(statement, id);
            System.out.println(teacherInfo);
        } finally {
            SqlSessionUtils.closeSession(sqlSession);
        }
    }
}

执行后的结果如下：

三、更新教师的学历

1、创建更新实体

在cn.horse.demo下创建更新实体，这里学历字段使用到了枚举类型：

TeacherInfoUpdate类：

package cn.horse.demo;

public class TeacherInfoUpdate {

    private Integer id;
    private Degree degree;

    public void setId(Integer id) {
        this.id = id;
    }

    public void setDegree(Degree degree) {
        this.degree = degree;
    }
}

Degree枚举类：

package cn.horse.demo;

public enum Degree {

    // 专科
    COLLEGE,
    // 本科
    BACHELOR,
    // 研究生
    GRADUATE,
    // 博士
    DOCTORAL
}

2、配置更新标签

在TeacherInfoMapper.xml配置文件中添加更新标签

<update id="update" parameterType="cn.horse.demo.TeacherInfoUpdate">
    UPDATE T_TEACHER
    <set>
        <choose>
            <when test="degree == @cn.horse.demo.Degree@COLLEGE">
                DEGREE = 'ZK',
            </when>
            <when test="degree == @cn.horse.demo.Degree@BACHELOR">
                DEGREE = 'BK',
            </when>
            <when test="degree == @cn.horse.demo.Degree@GRADUATE">
                DEGREE = 'YJS',
            </when>
            <when test="degree == @cn.horse.demo.Degree@DOCTORAL">
                DEGREE = 'BS',
            </when>
            <otherwise>
                DEGREE = 'ZK',
            </otherwise>
        </choose>
    </set>
    WHERE ID = #{id}
</update>

choose标签：标签中可以包含多个when标签、一个otherwise标签

when标签：test属性值用作条件判断，与if标签的test属性一样；when标签可以多次使用

otherwise标签：otherwise标签最多只能使用一次，并且只能作为choose标签内的最后一个标签。

choose标签类似于Java中的switch语句，会依次判断when标签中的条件是否满足，直到找到满足条件的when标签，并将when标签中的条件语句拼装到SQL语句中并跳出choose标签；如果找不到满足条件的when标签，则将otherwise标签中的条件语句拼装到SQL语句中并跳出choose标签；

分析：

在配置中，我们使用到了choose标签，通过when标签判断学历枚举数据来设置不同的值，将学历枚举数据转换成数据库表中存储的学历数据。

注意：

枚举类型在条件中使用时，格式如下：

      @枚举类型@枚举值

枚举类型需要全限类名（包名.类名）

3、更新教师的学历

新增updateTeacherInfo静态方法

private static void updateTeacherInfo(String statement, TeacherInfoUpdate update) {
    SqlSession sqlSession = null;
    try {
        sqlSession = SqlSessionUtils.openSession();
        sqlSession.update(statement, update);
        sqlSession.commit();
    } finally {
        SqlSessionUtils.closeSession(sqlSession);
    }
}

执行更新并查询

TeacherInfoUpdate update = new TeacherInfoUpdate();
update.setId(1);
update.setDegree(Degree.GRADUATE);
updateTeacherInfo("cn.horse.demo.TeacherInfoMapper.update", update);

findById("cn.horse.demo.TeacherInfoMapper.findById", 1);

执行后的结果如下：

结果分析：

choose标签中，依次判断条件是否满足，因为degree为GRADUATE，直到找到第三个条件满足，则将条件DEGREE=’YJS’,拼装到SQL语句中并跳出choose标签；所以执行的更新语句中包含DEGREE = ?