1. 程式人生 > >spring boot 學習筆記 (8)MyBatis 註解版

spring boot 學習筆記 (8)MyBatis 註解版

註解版

註解版的使用方式和 XML 版本相同,只有在構建 SQL 方面有所區別,所以本課重點介紹兩者之間的差異部分。

相關配置

註解版在 application.properties 只需要指明實體類的包路徑即可,其他保持不變:

mybatis.type-aliases-package=com.neo.model

spring.datasource.url=jdbc:mysql://localhost:3306/test?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true
spring.datasource.username=root
spring.datasource.password=root
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver

傳參方式

先來介紹一下使用註解版的 MyBatis 如何將引數傳遞到 SQL 中。

直接使用

@Delete("DELETE FROM users WHERE id =#{id}")
void delete(Long id);

在 SQL 中使用 #{id} 來接收同名引數。

使用 @Param

如果你的對映方法的形參有多個,這個註解使用在對映方法的引數上就能為它們取自定義名字。若不給出自定義名字,多引數則先以 "param" 作字首,再加上它們的引數位置作為引數別名。例如,#{param1}、#{param2},這個是預設值。如果註解是 @Param("person"),那麼引數就會被命名為 #{person}。

@Select("SELECT * FROM users WHERE user_sex = #{user_sex}")
List<User> getListByUserSex(@Param("user_sex") String userSex);

使用 Map

需要傳送多個引數時,可以考慮使用 Map:

@Select("SELECT * FROM users WHERE username=#{username} AND user_sex = #{user_sex}")
List<User> getListByNameAndSex(Map<String, Object> map);

使用時將引數依次加入到 Map 中即可:

Map param=  new HashMap();
param.put("username","aa");
param.put("user_sex","MAN");
List<User> users = userMapper.getListByNameAndSex(param);

使用物件

最常用的使用方式是直接使用物件:

@Insert("INSERT INTO users(userName,passWord,user_sex) VALUES(#{userName}, #{passWord}, #{userSex})")
void insert(User user);

在執行時,系統會自動讀取物件的屬性並值賦值到同名的 #{xxx} 中。

註解介紹

註解版最大的特點是具體的 SQL 檔案需要寫在 Mapper 類中,取消了 Mapper 的 XML 配置

上面介紹引數的時候,已經使用了 @Select、@Delete 等標籤,這就是 MyBatis 提供的註解來取代其 XML 檔案配置,下面我們一一介紹。

@Select 註解

@Select 主要在查詢的時候使用,查詢類的註解,所有的查詢均使用這個,具體如下:

@Select("SELECT * FROM users WHERE user_sex = #{user_sex}")
List<User> getListByUserSex(@Param("user_sex") String userSex);

@Insert 註解

@Insert,插入資料庫時使用,直接傳入實體類會自動解析屬性到對應的值,示例如下:

@Insert("INSERT INTO users(userName,passWord,user_sex) VALUES(#{userName}, #{passWord}, #{userSex})")
void insert(User user);

@Update 註解

@Update,所有的更新操作 SQL 都可以使用 @Update。

@Update("UPDATE users SET userName=#{userName},nick_name=#{nickName} WHERE id =#{id}")
void update(UserEntity user);

@Delete 註解

@Delete 處理資料刪除。

@Delete("DELETE FROM users WHERE id =#{id}")
void delete(Long id);

以上就是專案中常用的增、刪、改、查,但有時候我們有一些特殊的場景需要處理,比如查詢的物件返回值屬性名和欄位名不一致,或者物件的屬性中使用了列舉。我們期望查詢的返回結果可以將此欄位自動轉化為對應的型別,MyBatis 提供了另外兩個註解來支援:@Results 和 @Result。

@Results 和 @Result 註解

這兩個註解配合來使用,主要作用是將資料庫中查詢到的數值轉化為具體的欄位,修飾返回的結果集,關聯實體類屬性和資料庫欄位一一對應,如果實體類屬性和資料庫屬性名保持一致,就不需要這個屬性來修飾。示例如下:

@Select("SELECT * FROM users")
@Results({
    @Result(property = "userSex",  column = "user_sex", javaType = UserSexEnum.class),
    @Result(property = "nickName", column = "nick_name")
})
List<UserEntity> getAll();

為了更接近實際專案,特地將 user_sex、nick_name 兩個屬性加了下劃線和實體類屬性名不一致,另外 user_sex 使用了列舉,使用 @Results 和 @Result 即可解決這樣的問題。

瞭解更多註解使用請點選這裡檢視

注意,使用 # 符號和 $ 符號的不同:

// This example creates a prepared statement, something like select * from teacher where name = ?;
@Select("Select * from teacher where name = #{name}")
Teacher selectTeachForGivenName(@Param("name") String name);

// This example creates n inlined statement, something like select * from teacher where name = 'someName';
@Select("Select * from teacher where name = '${name}'")
Teacher selectTeachForGivenName(@Param("name") String name);

同上,上面兩個例子可以發現,使用 # 會對 SQL 進行預處理,使用 $ 時拼接 SQL,建議使用 #,使用 $ 有 SQL 注入的可能性。

動態 SQL

MyBatis 最大的特點是可以靈活的支援動態 SQL,在註解版中提供了兩種方式來支援,第一種是使用註解來實現,另一種是提供 SQL 類來支援。

使用註解來實現

用 script 標籤包圍,然後像 XML 語法一樣書寫:

@Update("<script>
  update users
    <set>
      <if test="userName != null">userName=#{userName},</if>
      <if test="nickName != null">nick_name=#{nickName},</if>
    </set>
  where id=#{id}
</script>")
void update(User user);

這種方式就是註解 + XML 的混合使用方式,既有 XML 靈活又有註解的方便,但也有一個缺點需要在 Java 程式碼中拼接 XML 語法很不方便,因此 MyBatis 又提供了一種更優雅的使用方式來支援動態構建 SQL。

使用 SQL 構建類來支援

以分頁為例進行演示,首先定義一個 UserSql 類,提供方法拼接需要執行的 SQL:

public class UserSql {
    public String getList(UserParam userParam) {
        StringBuffer sql = new StringBuffer("select id, userName, passWord, user_sex as userSex, nick_name as nickName");
        sql.append(" from users where 1=1 ");
        if (userParam != null) {
            if (StringUtils.isNotBlank(userParam.getUserName())) {
                sql.append(" and userName = #{userName}");
            }
            if (StringUtils.isNotBlank(userParam.getUserSex())) {
                sql.append(" and user_sex = #{userSex}");
            }
        }
        sql.append(" order by id desc");
        sql.append(" limit " + userParam.getBeginLine() + "," + userParam.getPageSize());
        log.info("getList sql is :" +sql.toString());
        return sql.toString();
    }
}

可以看出 UserSql 中有一個方法 getList,使用 StringBuffer 對 SQL 進行拼接,通過 if 判斷來動態構建 SQL,最後方法返回需要執行的 SQL 語句。

接下來只需要在 Mapper 中引入這個類和方法即可。

@SelectProvider(type = UserSql.class, method = "getList")
List<UserEntity> getList(UserParam userParam);
  • type:動態生成 SQL 的類
  • method:類中具體的方法名

相對於 @SelectProvider 提供查詢 SQL 方法匯入,還有 @InsertProvider、@UpdateProvider、@DeleteProvider 提供給插入、更新、刪除的時候使用。

結構化 SQL

可能你會覺得這樣拼接 SQL 很麻煩,SQL 語句太複雜也比較亂,彆著急!MyBatis 給我們提供了一種升級的方案:結構化 SQL。

示例如下:

public String getCount(UserParam userParam) {
   String sql= new SQL(){{
        SELECT("count(1)");
        FROM("users");
        if (StringUtils.isNotBlank(userParam.getUserName())) {
            WHERE("userName = #{userName}");
        }
        if (StringUtils.isNotBlank(userParam.getUserSex())) {
            WHERE("user_sex = #{userSex}");
        }
        //從這個 toString 可以看出,其內部使用高效的 StringBuilder 實現 SQL 拼接
    }}.toString();

    log.info("getCount sql is :" +sql);
    return sql;
}
  • SELECT 表示要查詢的欄位,可以寫多行,多行的 SELECT 會智慧地進行合併而不會重複。
  • FROM 和 WHERE 跟 SELECT 一樣,可以寫多個引數,也可以在多行重複使用,最終會智慧合併而不會報錯。這樣語句適用於寫很長的 SQL,且能夠保證 SQL 結構清楚,便於維護、可讀性高。

更多結構化的 SQL 語法請參考 SQL 語句構建器類

具體使用和 XML 版本一致,直接注入到使用的類中即可。

多資料來源使用

註解版的多資料來源使用和 XML 版本的多資料來源基本一致。

首先配置多資料來源:

mybatis.type-aliases-package=com.neo.model

spring.datasource.test1.jdbc-url=jdbc:mysql://localhost:3306/test1?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true
spring.datasource.test1.username=root
spring.datasource.test1.password=root
spring.datasource.test1.driver-class-name=com.mysql.cj.jdbc.Driver

spring.datasource.test2.jdbc-url=jdbc:mysql://localhost:3306/test2?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true
spring.datasource.test2.username=root
spring.datasource.test2.password=root
spring.datasource.test2.driver-class-name=com.mysql.cj.jdbc.Driver

分別構建兩個不同的資料來源。

DataSource1 配置:

@Configuration
@MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef  = "test1SqlSessionTemplate")
public class DataSource1Config {

    @Bean(name = "test1DataSource")
    @ConfigurationProperties(prefix = "spring.datasource.test1")
    @Primary
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "test1SqlSessionFactory")
    @Primary
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        return bean.getObject();
    }

    @Bean(name = "test1TransactionManager")
    @Primary
    public DataSourceTransactionManager testTransactionManager(@Qualifier("test1DataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "test1SqlSessionTemplate")
    @Primary
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

DataSource2 配置和 DataSource1 配置基本相同,只是去掉了 @Primary。

將以前的 Userapper 分別複製到 test1 和 test2 目錄下,分別作為兩個不同資料來源的 Mapper 來使用。

測試

分別注入兩個不同的 Mapper,想操作哪個資料來源就使用哪個資料來源的 Mapper 進行操作處理。

@RunWith(SpringRunner.class)
@SpringBootTest
public class UserMapperTest {
    @Autowired
    private User1Mapper user1Mapper;
    @Autowired
    private User2Mapper user2Mapper;

    @Test
    public void testInsert() throws Exception {
        user1Mapper.insert(new User("aa111", "a123456", UserSexEnum.MAN));
        user1Mapper.insert(new User("bb111", "b123456", UserSexEnum.WOMAN));
        user2Mapper.insert(new User("cc222", "b123456", UserSexEnum.MAN));
    }
}

執行測試用例完成後,檢查 test1 庫中的使用者表有兩條資料,test2 庫中的使用者表有 1 條資料證明測試成功。

如何選擇

兩種模式各有特點,註解版適合簡單快速的模式,在微服務架構中,一般微服務都有自己對應的資料庫,多表連線查詢的需求會大大的降低,會越來越適合註解版。XML 模式比適合大型專案,可以靈活地動態生成 SQL,方便調整 SQL,也有痛痛快快、洋洋灑灑地寫 SQL 的感覺。在具體開發過程中,根據公司業務和團隊技術基礎進行選型即可。