Fluent MyBatis實現動態SQL

MyBatis 令人喜歡的一大特性就是動態 SQL。在使用 JDBC 的過程中, 根據條件進行 SQL 的拼接是很麻煩且很容易出錯的,
MyBatis雖然提供瞭動態拼裝的能力,但這些寫xml文件,也確實折磨開發。Fluent MyBatis提供瞭更貼合Java語言特質的,對程序員友好的Fluent拼裝能力。

Fluent MyBatis動態SQL,寫SQL更爽

數據準備

為瞭後面的演示, 創建瞭一個 Maven 項目 fluent-mybatis-dynamic, 創建瞭對應的數據庫和表

DROP TABLE IF EXISTS `student`;

CREATE TABLE `student`
(
    `id`           bigint(21) unsigned NOT NULL AUTO_INCREMENT COMMENT '編號',
    `name`         varchar(20) DEFAULT NULL COMMENT '姓名',
    `phone`        varchar(20) DEFAULT NULL COMMENT '電話',
    `email`        varchar(50) DEFAULT NULL COMMENT '郵箱',
    `gender`       tinyint(2)  DEFAULT NULL COMMENT '性別',
    `locked`       tinyint(2)  DEFAULT NULL COMMENT '狀態(0:正常,1:鎖定)',
    `gmt_created`  datetime    DEFAULT CURRENT_TIMESTAMP COMMENT '存入數據庫的時間',
    `gmt_modified` datetime    DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '修改的時間',
    `is_deleted`   tinyint(2)  DEFAULT 0,
    PRIMARY KEY (`id`)
) ENGINE = InnoDB
  DEFAULT CHARSET = utf8mb4
  COLLATE = utf8mb4_0900_ai_ci COMMENT ='學生表';

代碼生成

使用Fluent Mybatis代碼生成器,生成對應的Entity文件

public class Generator {
    static final String url = "jdbc:mysql://localhost:3306/fluent_mybatis?useSSL=false&useUnicode=true&characterEncoding=utf-8";
    /**
     * 生成代碼的package路徑
     */
    static final String basePackage = "cn.org.fluent.mybatis.dynamic";

    /**
     * 使用 test/resource/init.sql文件自動初始化測試數據庫
     */
    @BeforeAll
    static void runDbScript() {
        DataSourceCreatorFactory.create("dataSource");
    }

    @Test
    void test() {
        FileGenerator.build(Nothing.class);
    }

    @Tables(
        /** 數據庫連接信息 **/
        url = url, username = "root", password = "password",
        /** Entity類parent package路徑 **/
        basePack = basePackage,
        /** Entity代碼源目錄 **/
        srcDir = "src/main/java",
        /** 如果表定義記錄創建,記錄修改,邏輯刪除字段 **/
        gmtCreated = "gmt_created", gmtModified = "gmt_modified", logicDeleted = "is_deleted",
        /** 需要生成文件的表 ( 表名稱:對應的Entity名稱 ) **/
        tables = @Table(value = {"student"})
    )

    public static class Nothing {
    }
}

編譯項目,ok,下面我們開始動態SQL構造旅程

在 WHERE 條件中使用動態條件

在mybatis中,if 標簽是大傢最常使用的。在查詢、刪除、更新的時候結合 test 屬性聯合使用。

示例:根據輸入的學生信息進行條件檢索

  • 當隻輸入用戶名時, 使用用戶名進行模糊檢索;
  • 當隻輸入性別時, 使用性別進行完全匹配
  • 當用戶名和性別都存在時, 用這兩個條件進行查詢匹配查詢

mybatis動態 SQL寫法

<select id="selectByStudentSelective" resultMap="BaseResultMap" parameterType="com.homejim.mybatis.entity.Student">
    select
    <include refid="Base_Column_List" />
    from student
    <where>
        <if test="name != null and name !=''">
          and name like concat('%', #{name}, '%')
        </if>
        <if test="sex != null">
          and sex=#{sex}
        </if>
    </where>
</select>

fluent mybatis動態寫法

@Repository
public class StudentDaoImpl extends StudentBaseDao implements StudentDao {
   /**
    * 根據輸入的學生信息進行條件檢索
    * 1. 當隻輸入用戶名時, 使用用戶名進行模糊檢索;
    * 2. 當隻輸入性別時, 使用性別進行完全匹配
    * 3. 當用戶名和性別都存在時, 用這兩個條件進行查詢匹配的用
    *
    * @param name   姓名,模糊匹配
    * @param isMale 性別
    * @return
    */
    @Override
    public List<StudentEntity> selectByNameOrEmail(String name, Boolean isMale) {
        return super.defaultQuery()
            .where.name().like(name, If::notBlank)
            .and.gender().eq(isMale, If::notNull).end()
            .execute(super::listEntity);
    }
}

FluentMyBatis的實現方式至少有下面的好處

  • 邏輯就在方法實現上,不需要額外維護xml,割裂開來
  • 所有的編碼通過IDE智能提示,沒有字符串魔法值編碼
  • 編譯檢查,拼寫錯誤能立即發現

測試

@SpringBootTest(classes = AppMain.class)
public class StudentDaoImplTest extends Test4J {
    @Autowired
    StudentDao studentDao;

    @DisplayName("隻有名字時的查詢")
    @Test
    void selectByNameOrEmail_onlyName() {
        studentDao.selectByNameOrEmail("明", null);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "SELECT id, gmt_created, gmt_modified, is_deleted, email, gender, locked, name, phone " +
                "FROM student " +
                "WHERE name LIKE ?",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{"%明%"});
    }

    @DisplayName("隻有性別時的查詢")
    @Test
    void selectByNameOrEmail_onlyGender() {
        studentDao.selectByNameOrEmail(null, false);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "SELECT id, gmt_created, gmt_modified, is_deleted, email, gender, locked, name, phone " +
                "FROM student " +
                "WHERE gender = ?",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{false});
    }

    @DisplayName("姓名和性別同時存在的查詢")
    @Test
    void selectByNameOrEmail_both() {
        studentDao.selectByNameOrEmail("明", false);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "SELECT id, gmt_created, gmt_modified, is_deleted, email, gender, locked, name, phone " +
                "FROM student " +
                "WHERE name LIKE ? " +
                "AND gender = ?",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{"%明%", false});
    }
}

在 UPDATE 使用動態更新

隻更新有變化的字段, 空值不更新

mybatis xml寫法

<update id="updateByPrimaryKeySelective" parameterType="...">
    update student
    <set>
      <if test="name != null">
        `name` = #{name,jdbcType=VARCHAR},
      </if>
      <if test="phone != null">
        phone = #{phone,jdbcType=VARCHAR},
      </if>
      <if test="email != null">
        email = #{email,jdbcType=VARCHAR},
      </if>
      <if test="gender != null">
        gender = #{gender,jdbcType=TINYINT},
      </if>
      <if test="gmtModified != null">
        gmt_modified = #{gmtModified,jdbcType=TIMESTAMP},
      </if>
    </set>
    where id = #{id,jdbcType=INTEGER}
</update>

fluent mybatis實現

@Repository
public class StudentDaoImpl extends StudentBaseDao implements StudentDao {
    /**
     * 根據主鍵更新非空屬性
     *
     * @param student
     * @return
     */
    @Override
    public int updateByPrimaryKeySelective(StudentEntity student) {
        return super.defaultUpdater()
            .update.name().is(student.getName(), If::notBlank)
            .set.phone().is(student.getPhone(), If::notBlank)
            .set.email().is(student.getEmail(), If::notBlank)
            .set.gender().is(student.getGender(), If::notNull)
            .end()
            .where.id().eq(student.getId()).end()
            .execute(super::updateBy);
    }    
}

測試

@SpringBootTest(classes = AppMain.class)
public class StudentDaoImplTest extends Test4J {
    @Autowired
    StudentDao studentDao;

    @Test
    void updateByPrimaryKeySelective() {
        StudentEntity student = new StudentEntity()
            .setId(1L)
            .setName("test")
            .setPhone("13866668888");
        studentDao.updateByPrimaryKeySelective(student);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "UPDATE student " +
                "SET gmt_modified = now(), " +
                "name = ?, " +
                "phone = ? " +
                "WHERE id = ?",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{"test", "13866668888", 1L});
    }
}

choose 標簽

在mybatis中choose when otherwise 標簽可以幫我們實現 if else 的邏輯。

查詢條件,假設 name 具有唯一性, 查詢一個學生

  • 當 id 有值時, 使用 id 進行查詢;
  • 當 id 沒有值時, 使用 name 進行查詢;
  • 否則返回空

mybatis xml實現

<select id="selectByIdOrName" resultMap="BaseResultMap" parameterType="...">
    select
    <include refid="Base_Column_List" />
    from student
    <where>
        <choose>
          <when test="id != null">
            and id=#{id}
          </when>
          <when test="name != null and name != ''">
            and name=#{name}
          </when>
          <otherwise>
            and 1=2
          </otherwise>
        </choose>
    </where>
</select>

fluent mybatis實現方式

@Repository
public class StudentDaoImpl extends StudentBaseDao implements StudentDao {

    /**
     * 1. 當 id 有值時, 使用 id 進行查詢;
     * 2. 當 id 沒有值時, 使用 name 進行查詢;
     * 3. 否則返回空
     */ 
    @Override
    public StudentEntity selectByIdOrName(StudentEntity student) {
       return super.defaultQuery()
           .where.id().eq(student.getId(), If::notNull)
           .and.name().eq(student.getName(), name -> isNull(student.getId()) && notBlank(name))
           .and.apply("1=2", () -> isNull(student.getId()) && isBlank(student.getName()))
           .end()
           .execute(super::findOne).orElse(null);
    }
}

測試

@SpringBootTest(classes = AppMain.class)
public class StudentDaoImplTest extends Test4J {
    @Autowired
    StudentDao studentDao;

    @DisplayName("有 ID 則根據 ID 獲取")
    @Test
    void selectByIdOrName_byId() {
        StudentEntity student = new StudentEntity();
        student.setName("小飛機");
        student.setId(1L);

        StudentEntity result = studentDao.selectByIdOrName(student);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "SELECT id, gmt_created, gmt_modified, is_deleted, email, gender, locked, name, phone " +
                "FROM student " +
                "WHERE id = ?",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{1L});
    }

    @DisplayName("沒有 ID 則根據 name 獲取")
    @Test
    void selectByIdOrName_byName() {
        StudentEntity student = new StudentEntity();
        student.setName("小飛機");
        student.setId(null);

        StudentEntity result = studentDao.selectByIdOrName(student);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "SELECT id, gmt_created, gmt_modified, is_deleted, email, gender, locked, name, phone " +
                "FROM student " +
                "WHERE name = ?",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{"小飛機"});
    }

    @DisplayName("沒有 ID 和 name, 返回 null")
    @Test
    void selectByIdOrName_null() {
        StudentEntity student = new StudentEntity();
        StudentEntity result = studentDao.selectByIdOrName(student);
        // 驗證執行的sql語句
        db.sqlList().wantFirstSql().eq("" +
                "SELECT id, gmt_created, gmt_modified, is_deleted, email, gender, locked, name, phone " +
                "FROM student " +
                "WHERE 1=2",
            StringMode.SameAsSpace);
        // 驗證sql參數
        db.sqlList().wantFirstPara().eqReflect(new Object[]{});
    }
}

參考

示例代碼地址
Fluent MyBatis地址
Fluent MyBatis文檔
Test4J框架

到此這篇關於Fluent MyBatis實現動態SQL的文章就介紹到這瞭,更多相關Fluent MyBatis 動態SQL內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: