MybatisPlus關(guān)聯(lián)查詢的完美實(shí)現(xiàn)方案

更新時(shí)間：2021年12月15日 09:38:06 作者：97牛肉面

我們?cè)陧?xiàng)目開(kāi)發(fā)的時(shí)候,難免會(huì)遇到連表查詢的操作,所以下面這篇文章主要給大家介紹了關(guān)于MybatisPlus關(guān)聯(lián)查詢的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下

Mybatis-Plus 簡(jiǎn)介

什么是 MyBatis-Plus？Mybatis-Plus：為簡(jiǎn)化開(kāi)發(fā)而生

MyBatis-Plus（簡(jiǎn)稱 MP）是一個(gè)基于 MyBatis 的增強(qiáng)工具，它對(duì) Mybatis 的基礎(chǔ)功能進(jìn)行了增強(qiáng)，但未做任何改變。使得我們可以可以在 Mybatis 開(kāi)發(fā)的項(xiàng)目上直接進(jìn)行升級(jí)為 Mybatis-plus，正如它對(duì)自己的定位，它能夠幫助我們進(jìn)一步簡(jiǎn)化開(kāi)發(fā)過(guò)程，提高開(kāi)發(fā)效率。

Mybatis-Plus 其實(shí)可以看作是對(duì) Mybatis 的再一次封裝，升級(jí)之后，對(duì)于單表的 CRUD 操作，調(diào)用 Mybatis-Plus 所提供的 API 就能夠輕松實(shí)現(xiàn)，此外還提供了各種查詢方式、分頁(yè)等行為。最最重要的，開(kāi)發(fā)人員還不用去編寫(xiě) XML，這就大大降低了開(kāi)發(fā)難度

官網(wǎng)：https://mp.baomidou.com

連表？Left Join？Inner Join？

MybtaisPlus 極大簡(jiǎn)便了單表的操作，但對(duì)應(yīng)連表查詢更多的還是在xml中實(shí)現(xiàn)，本人是一個(gè)單表主義者（能單表，盡量不連表），阿里的相關(guān)規(guī)范中也提到：盡量使用單表，連表盡量不要超過(guò)3張，于是更多的時(shí)候，是把主表查詢處理，然后使用流處理把關(guān)聯(lián)的表Id集合起來(lái)，再通過(guò)listByIds轉(zhuǎn)toMap，得到Id對(duì)應(yīng)的對(duì)象，再進(jìn)行g(shù)et相對(duì)應(yīng)賦值，最初也沒(méi)啥，當(dāng)寫(xiě)的越來(lái)越多了之后，問(wèn)題就開(kāi)始暴露了，沒(méi)錯(cuò)，代碼實(shí)在是太長(zhǎng)，太臭啦?。?！

舊版代碼

Set<Long> systemProjectFileSecurityIdSet = records.stream().map(SystemProjectFileSecurityGroupSetting::getSystemProjectFileSecurityId).collect(Collectors.toSet());
Map<Long, SystemProjectFileSecurity> systemProjectFileSecurityMap = new HashMap<>();
if (!systemProjectFileSecurityIdSet.isEmpty()) {
            systemProjectFileSecurityMap.putAll(systemProjectFileSecurityService.listByIds(systemProjectFileSecurityIdSet)
                    .stream()
                    .collect(Collectors.toMap(SystemProjectFileSecurity::getSystemProjectFileSecurityId, systemProjectFileSecurity -> systemProjectFileSecurity)));
}

舊版代碼太長(zhǎng)了啦，每次關(guān)聯(lián)，都需要5-7行，一張業(yè)務(wù)表少說(shuō)也有3到5個(gè)表關(guān)聯(lián)，這樣編寫(xiě)的話，顯得代碼太過(guò)累贅，以及冗余，整體看的實(shí)在太難受啦

新版優(yōu)化

Map<Long, SystemProjectFileSecurity> systemProjectFileSecurityMap = 
    linkTableUtils.linkSystemProjectFileSecurity(records, SystemProjectFileSecurityGroupSetting::getSystemProjectFileSecurityId);

我們對(duì)冗余的代碼進(jìn)行的抽取，目前1-2行就可以實(shí)現(xiàn)5-7行的功能，就算關(guān)聯(lián)查詢更多的表，代碼看起來(lái)也更整潔，嗯，真香

相關(guān)工具類

import com.baomidou.mybatisplus.extension.service.IService;
import org.springframework.stereotype.Component;
 
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;
 
/**
 * 鏈接其他表的工具類
 *
 * @author zengqinghua
 */
@Component
public class LinkTableUtils {
 
    /**
     * 鏈接 XXX表
     */
    public <T> Map<Long, SystemProjectFileSecurity> linkSystemProjectFileSecurity(List<T> records, Function<T, Long> getKey) {
        return this.linkTable(
                TableEnum.SystemProjectFileSecurity,
                records, getKey,
                SystemProjectFileSecurity::getSystemProjectFileSecurityId
        );
    }
 
    /**
     * 鏈接 表（通用方法）
     *
     * @param tableEnum 表名枚舉
     * @param records   數(shù)據(jù)源列表
     * @param getKey    查詢的Id
     * @param getRKey   map的key
     * @param <T>       數(shù)據(jù)源
     * @param <R>       結(jié)果
     * @return
     */
    public <T, R> Map<Long, R> linkTable(
            TableEnum tableEnum,
            List<T> records, Function<T, Long> getKey,
            Function<R, Long> getRKey
    ) {
        if (records.isEmpty()) {
            return new HashMap<>();
        }
 
        IService<R> iService = (IService) ApplicationContextUtil.getBean(tableEnum.tClass);
 
        // 得到對(duì)應(yīng)的id
        Set<Long> idSet = records.stream().map(getKey).collect(Collectors.toSet());
        Map<Long, R> map = new HashMap<>();
        if (!idSet.isEmpty()) {
            map.putAll(iService.listByIds(idSet)
                    .stream()
                    .collect(Collectors.toMap(getRKey, data -> data)));
        }
        return map;
    }
 
    public enum TableEnum {
        SystemProjectFileSecurity("xxxx", ISystemProjectFileSecurityService.class);
 
        private final String tableRemark;
        private final Class tClass;
 
        TableEnum(String tableRemark, Class tClass) {
            this.tableRemark = tableRemark;
            this.tClass = tClass;
        }
    }
}