MyBatis-Plus中如何使用ResultMap的方法示例
MyBatis-Plus (簡稱MP)是一個(gè)MyBatis的增強(qiáng)工具,在MyBatis的基礎(chǔ)上只做增強(qiáng)不做改變,為簡化開發(fā)、提高效率而生。
MyBatis-Plus對(duì)MyBatis基本零侵入,完全可以與MyBatis混合使用,這點(diǎn)很贊。
在涉及到關(guān)系型數(shù)據(jù)庫增刪查改的業(yè)務(wù)時(shí),我比較喜歡用MyBatis-Plus,開發(fā)效率極高。具體的使用可以參考官網(wǎng),或者自己上手摸索感受一下。
下面簡單總結(jié)一下在MyBatis-Plus中如何使用ResultMap。
問題說明
先看個(gè)例子:
有如下兩張表:
create table tb_book
(
id bigint primary key,
name varchar(32),
author varchar(20)
);
create table tb_hero
(
id bigint primary key,
name varchar(32),
age int,
skill varchar(32),
bid bigint
);
其中,tb_hero中的bid關(guān)聯(lián)tb_book表的id。
下面先看Hero實(shí)體類的代碼,如下:
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.annotation.JsonInclude;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;
@Getter
@Setter
@NoArgsConstructor
@TableName("tb_hero")
@JsonInclude(JsonInclude.Include.NON_NULL)
public class Hero {
@TableId("id")
private Long id;
@TableField(value = "name", keepGlobalFormat = true)
private String name;
@TableField(value = "age", keepGlobalFormat = true)
private Integer age;
@TableField(value = "skill", keepGlobalFormat = true)
private String skill;
@TableField(value = "bid", keepGlobalFormat = true)
private Long bookId;
// *********************************
// 數(shù)據(jù)庫表中不存在以下字段(表join時(shí)會(huì)用到)
// *********************************
@TableField(value = "book_name", exist = false)
private String bookName;
@TableField(value = "author", exist = false)
private String author;
}
注意了,我特地把tb_hero表中的bid字段映射成實(shí)體類Hero中的bookId屬性。
測(cè)試BaseMapper中內(nèi)置的insert()方法或者IService中的save()方法
MyBatis-Plus打印出的SQL為:
==> Preparing: INSERT INTO tb_hero ( id, "name", "age", "skill", "bid" ) VALUES ( ?, ?, ?, ?, ? )
==> Parameters: 1589788935356416(Long), 阿飛(String), 18(Integer), 天下第一快劍(String), 1(Long)
沒毛病, MyBatis-Plus會(huì)根據(jù)@TableField指定的映射關(guān)系,生成對(duì)應(yīng)的SQL。
測(cè)試BaseMapper中內(nèi)置的selectById()方法或者IService中的getById()方法
MyBatis-Plus打印出的SQL為:
==> Preparing: SELECT id,"name","age","skill","bid" AS bookId FROM tb_hero WHERE id=?
==> Parameters: 1(Long)
也沒毛病,可以看到生成的SELECT中把bid做了別名bookId。
測(cè)試自己寫的SQL
比如現(xiàn)在我想連接tb_hero與tb_book這兩張表,如下:
@Mapper
@Repository
public interface HeroMapper extends BaseMapper<Hero> {
@Select({"SELECT tb_hero.*, tb_book.name as book_name, tb_book.author" +
" FROM tb_hero" +
" LEFT JOIN tb_book" +
" ON tb_hero.bid = tb_book.id" +
" ${ew.customSqlSegment}"})
IPage<Hero> pageQueryHero(@Param(Constants.WRAPPER) Wrapper<Hero> queryWrapper,
Page<Hero> page);
}
查詢MyBatis-Plus打印出的SQL為:
==> Preparing: SELECT tb_hero.*, tb_book.name AS book_name, tb_book.author FROM tb_hero LEFT JOIN tb_book ON tb_hero.bid = tb_book.id WHERE ("bid" = ?) ORDER BY id ASC LIMIT ? OFFSET ?
==> Parameters: 2(Long), 1(Long), 1(Long)
SQL沒啥問題,過濾與分頁也都正常,但是此時(shí)你會(huì)發(fā)現(xiàn)bookId屬性為null,如下:

為什么呢?
調(diào)用BaseMapper中內(nèi)置的selectById()方法并沒有出現(xiàn)這種情況?????
回過頭來再對(duì)比一下在HeroMapper中自己定義的查詢與MyBatis-Plus自帶的selectById()有啥不同,還記得上面的剛剛的測(cè)試嗎,生成的SQL有啥不同?
原來,MyBatis-Plus為BaseMapper中內(nèi)置的方法生成SQL時(shí),會(huì)把SELECT子句中bid做別名bookId,而自己寫的查詢MyBatis-Plus并不會(huì)幫你修改SELECT子句,也就導(dǎo)致bookId屬性為null。
解決方法
方案一:表中的字段與實(shí)體類的屬性嚴(yán)格保持一致(字段有下劃線則屬性用駝峰表示)
在這里就是tb_hero表中的bid字段映射成實(shí)體類Hero中的bid屬性。這樣當(dāng)然可以解決問題,但不是本篇講的重點(diǎn)。
方案二:把自己寫的SQL中bid做別名bookId
方案三:使用@ResultMap,這是此篇的重點(diǎn)
在@TableName設(shè)置autoResultMap = true
@TableName(value = "tb_hero", autoResultMap = true)
public class Hero {
}
然后在自定義查詢中添加@ResultMap注解,如下:
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.ResultMap;
import org.apache.ibatis.annotations.Select;
import org.springframework.stereotype.Repository;
@Mapper
@Repository
public interface HeroMapper extends BaseMapper<Hero> {
@ResultMap("mybatis-plus_Hero")
@Select({"SELECT tb_hero.*, tb_book.name as book_name, tb_book.author" +
" FROM tb_hero" +
" LEFT JOIN tb_book" +
" ON tb_hero.bid = tb_book.id" +
" ${ew.customSqlSegment}"})
IPage<Hero> pageQueryHero(@Param(Constants.WRAPPER) Wrapper<Hero> queryWrapper,
Page<Hero> page);
}
這樣,也能解決問題。
下面簡單看下源碼,@ResultMap("mybatis-plus_實(shí)體類名")怎么來的。
詳情見: com.baomidou.mybatisplus.core.metadata.TableInfo#initResultMapIfNeed()
/**
* 自動(dòng)構(gòu)建 resultMap 并注入(如果條件符合的話)
*/
void initResultMapIfNeed() {
if (autoInitResultMap && null == resultMap) {
String id = currentNamespace + DOT + MYBATIS_PLUS + UNDERSCORE + entityType.getSimpleName();
List<ResultMapping> resultMappings = new ArrayList<>();
if (havePK()) {
ResultMapping idMapping = new ResultMapping.Builder(configuration, keyProperty, StringUtils.getTargetColumn(keyColumn), keyType)
.flags(Collections.singletonList(ResultFlag.ID)).build();
resultMappings.add(idMapping);
}
if (CollectionUtils.isNotEmpty(fieldList)) {
fieldList.forEach(i -> resultMappings.add(i.getResultMapping(configuration)));
}
ResultMap resultMap = new ResultMap.Builder(configuration, id, entityType, resultMappings).build();
configuration.addResultMap(resultMap);
this.resultMap = id;
}
}
注意看上面的字符串id的構(gòu)成,你應(yīng)該可以明白。
思考: 這種方式的ResultMap默認(rèn)是強(qiáng)綁在一個(gè)@TableName上的,如果是某個(gè)聚合查詢或者查詢的結(jié)果并非對(duì)應(yīng)一個(gè)真實(shí)的表怎么辦呢?有沒有更優(yōu)雅的方式?
自定義@AutoResultMap注解
基于上面的思考,我做了下面簡單的實(shí)現(xiàn):
自定義@AutoResultMap注解
import java.lang.annotation.*;
/**
* 使用@AutoResultMap注解的實(shí)體類
* 自動(dòng)生成{auto.mybatis-plus_類名}為id的resultMap
* {@link MybatisPlusConfig#initAutoResultMap()}
*/
@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface AutoResultMap {
}
動(dòng)時(shí)掃描@AutoResultMap注解的實(shí)體類
package com.bytesfly.mybatis.config;
import cn.hutool.core.util.ClassUtil;
import cn.hutool.core.util.ReflectUtil;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.baomidou.mybatisplus.extension.toolkit.JdbcUtils;
import com.bytesfly.mybatis.annotation.AutoResultMap;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import javax.annotation.PostConstruct;
import java.util.Set;
/**
* 可添加一些插件
*/
@Configuration
@EnableTransactionManagement(proxyTargetClass = true)
@MapperScan(basePackages = "com.bytesfly.mybatis.mapper")
@Slf4j
public class MybatisPlusConfig {
@Autowired
private SqlSessionTemplate sqlSessionTemplate;
/**
* 分頁插件(根據(jù)jdbcUrl識(shí)別出數(shù)據(jù)庫類型, 自動(dòng)選擇適合該方言的分頁插件)
* 相關(guān)使用說明: https://baomidou.com/guide/page.html
*/
@Bean
public MybatisPlusInterceptor mybatisPlusInterceptor(DataSourceProperties dataSourceProperties) {
String jdbcUrl = dataSourceProperties.getUrl();
DbType dbType = JdbcUtils.getDbType(jdbcUrl);
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(dbType));
return interceptor;
}
/**
* @AutoResultMap注解的實(shí)體類自動(dòng)構(gòu)建resultMap并注入
*/
@PostConstruct
public void initAutoResultMap() {
try {
log.info("--- start register @AutoResultMap ---");
String namespace = "auto";
String packageName = "com.bytesfly.mybatis.model.db.resultmap";
Set<Class<?>> classes = ClassUtil.scanPackageByAnnotation(packageName, AutoResultMap.class);
org.apache.ibatis.session.Configuration configuration = sqlSessionTemplate.getConfiguration();
for (Class clazz : classes) {
MapperBuilderAssistant assistant = new MapperBuilderAssistant(configuration, "");
assistant.setCurrentNamespace(namespace);
TableInfo tableInfo = TableInfoHelper.initTableInfo(assistant, clazz);
if (!tableInfo.isAutoInitResultMap()) {
// 設(shè)置 tableInfo的autoInitResultMap屬性 為 true
ReflectUtil.setFieldValue(tableInfo, "autoInitResultMap", true);
// 調(diào)用 tableInfo#initResultMapIfNeed() 方法,自動(dòng)構(gòu)建 resultMap 并注入
ReflectUtil.invoke(tableInfo, "initResultMapIfNeed");
}
}
log.info("--- finish register @AutoResultMap ---");
} catch (Throwable e) {
log.error("initAutoResultMap error", e);
System.exit(1);
}
}
}
關(guān)鍵代碼其實(shí)沒有幾行,耐心看下應(yīng)該不難懂。
使用@AutoResultMap注解
還是用例子來說明更直觀。
下面是一個(gè)聚合查詢:
@Mapper
@Repository
public interface BookMapper extends BaseMapper<Book> {
@ResultMap("auto.mybatis-plus_BookAgg")
@Select({"SELECT tb_book.id, max(tb_book.name) as name, array_agg(distinct tb_hero.id order by tb_hero.id asc) as hero_ids" +
" FROM tb_hero" +
" INNER JOIN tb_book" +
" ON tb_hero.bid = tb_book.id" +
" GROUP BY tb_book.id"})
List<BookAgg> agg();
}
其中BookAgg的定義如下,在實(shí)體類上使用了@AutoResultMap注解:
@Getter
@Setter
@NoArgsConstructor
@AutoResultMap
public class BookAgg {
@TableId("id")
private Long bookId;
@TableField("name")
private String bookName;
@TableField("hero_ids")
private Object heroIds;
}
完整代碼見: https://github.com/bytesfly/springboot-demo/tree/master/springboot-mybatis-plus
到此這篇關(guān)于MyBatis-Plus中如何使用ResultMap的方法示例的文章就介紹到這了,更多相關(guān)MyBatis-Plus使用ResultMap內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
SpringBoot整合WebSocket實(shí)現(xiàn)聊天室流程全解
WebSocket協(xié)議是基于TCP的一種新的網(wǎng)絡(luò)協(xié)議。本文將通過SpringBoot集成WebSocket實(shí)現(xiàn)簡易聊天室,對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,感興趣的可以了解一下2023-01-01
使用springmvc運(yùn)行流程分析,手寫spring框架嘗試
這篇文章主要介紹了使用springmvc運(yùn)行流程分析,手寫spring框架嘗試,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2021-10-10
Security 登錄認(rèn)證流程詳細(xì)分析詳解
本文Security登錄認(rèn)證流程詳細(xì)分析詳解,文中通過示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2022-01-01
SpringCloud Feign配置應(yīng)用詳細(xì)介紹
這篇文章主要介紹了SpringCloud Feign配置應(yīng)用,feign是netflix提供的服務(wù)間基于http的rpc調(diào)用框架,在spring cloud得到廣泛應(yīng)用2022-09-09

