溫馨提示×
您好，登錄后才能下訂單哦！
點(diǎn)擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》
您好，登錄后才能下訂單哦！
這篇文章將為大家詳細(xì)講解有關(guān)使用mybatis-plus如何實(shí)現(xiàn)生成mapper擴(kuò)展文件，文章內(nèi)容質(zhì)量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關(guān)知識有一定的了解。
我們都知道，mybatis-plus是一個mybatis的增強(qiáng)工具，為簡化開發(fā)、提高效率而生，我們經(jīng)常使用mybatis-plus生成controller、service、mapper等文件，對于簡單的curd，可以直接使用mybatis-plus封裝好的方法。
然而，我們經(jīng)常有這樣那樣的需求，需要額外編寫sql實(shí)現(xiàn)，如果直接在mapper.xml文件中編寫，一旦數(shù)據(jù)庫表結(jié)構(gòu)改動需要重新生成文件就悲催了，不得不花大量精力修改代碼。所以，這里介紹一種方式，自動生成mapper擴(kuò)展文件，我們自定義編寫的程序存放在擴(kuò)展文件中，這樣在數(shù)據(jù)庫表結(jié)構(gòu)改動時，不用擔(dān)心程序會被覆蓋，也不用修改代碼。
mybatis-plus版本
<dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>3.2.0</version> </dependency> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-generator</artifactId> <version>3.2.0</version> </dependency> <dependency> <groupId>org.apache.velocity</groupId> <artifactId>velocity-engine-core</artifactId> <version>2.1</version> </dependency>
mybatis-plus生成mapper擴(kuò)展文件
  熟悉mybatis-plus的朋友都知道，mybatis-plus提供了一款代碼生成器，可以自動生成代碼，我們就從這款代碼生成器入手。
  代碼生成器配置完畢后，運(yùn)行時會執(zhí)行 AutoGenerator.execute() 方法，我們先看看這個東東
熟悉的GlobalConfig、DataSourceConfig等等就不介紹了，我們關(guān)注的是InjectionConfig、TemplateConfig和ConfigBuilder。這里先講述一下我們的思路：把ext文件通過配置直接生成，并保留mybatis-plus為service擴(kuò)展的批量操作，我們需要三個文件，第一個文件生成ext.java，第二個文件生成ext.xml，第三個覆蓋serviceImpl文件（保留mybatis-plus為service擴(kuò)展的批量操作）
**************************干貨來咯**************************
// 生成目錄
public static final String OUTPUT_DIR = "項(xiàng)目路徑/src/main/java";
// mapperExt目錄
public static final String MAPPER_EXT = "ext目錄";
// 模板配置，這里可以自定義模板路徑，如果路徑如下所示，則該部分可以省略
TemplateConfig tc = new TemplateConfig();
tc.setServiceImpl("templates/serviceImpl.java");
ag.setTemplate(tc);
// 自定義配置
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
Map<String, Object> map = new HashMap<>(10);
/// 這里可以在 VM 文件中用 ${cfg.MapperExt} 引用該值
map.put("MapperExt", MAPPER_EXT.replace('/', '.'));
this.setMap(map);
}
};
// 自定義輸出配置
List<FileOutConfig> focList = new ArrayList<>();
focList.add(new FileOutConfig("templates/mapperExt.xml.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
return String.format("%s/%s/%sMapperExt%s", OUTPUT_DIR, MAPPER_EXT, tableInfo.getEntityName(), StringPool.DOT_XML);
}
});
focList.add(new FileOutConfig("templates/mapperExt.java.vm") {
@Override
public String outputFile(TableInfo tableInfo) {
return String.format("%s/%s/%sMapperExt%s", OUTPUT_DIR, MAPPER_EXT, tableInfo.getEntityName(), StringPool.DOT_JAVA);
}
});
cfg.setFileCreate(new IFileCreate() {
@Override
public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
// 如果是 mapperExt、service、controller 文件，并且已存在則不創(chuàng)建
if (filePath.contains(MAPPER_EXT) || fileType == FileType.CONTROLLER || fileType == FileType.SERVICE || fileType == FileType.SERVICE_IMPL) {
if (new File(filePath).exists()) {
return false;
}
}
// 判斷文件夾是否需要創(chuàng)建
checkDir(filePath);
return true;
}
});
cfg.setFileOutConfigList(focList);
ag.setCfg(cfg);mapperExt.java.vm配置
package ${cfg.MapperExt};
import ${package.Mapper}.${table.mapperName};
/**
* <p>
* $!{table.comment} MapperExt 接口
* </p>
*
* @author ${author}
* @since ${date}
*/
#if(${kotlin})
interface ${table.mapperName}Ext : ${table.mapperName}
#else
public interface ${table.mapperName}Ext extends ${table.mapperName} {
}
#endmapperExt.xml.vm配置
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="${cfg.MapperExt}.${table.mapperName}Ext">
</mapper>serviceImpl.java.vm配置
package ${package.ServiceImpl};
import ${package.Entity}.${entity};
import ${cfg.MapperExt}.${table.mapperName}Ext;
import ${package.Service}.${table.serviceName};
import ${superServiceImplClassPackage};
import org.springframework.stereotype.Service;
/**
* <p>
* $!{table.comment} 服務(wù)實(shí)現(xiàn)類
* </p>
*
* @author ${author}
* @since ${date}
*/
@Service
#if(${kotlin})
open class ${table.serviceImplName} : ${superServiceImplClass}<${table.mapperName}Ext, ${entity}>(), ${table.serviceName} {
}
#else
public class ${table.serviceImplName} extends ${superServiceImplClass}<${table.mapperName}Ext, ${entity}> implements ${table.serviceName} {
}
#end新思路（2020-04-17補(bǔ)充）
  我們概述中所述問題真的存在嗎？mybatis-plus如果需要擴(kuò)展文件那么他為什么不提供呢？當(dāng)然是問題不存在，根本不需要擴(kuò)展文件，如果你存在這樣的問題，你總是覆蓋文件說明你的用法有問題。
  這里直接說應(yīng)該怎么使用，我們把本文所述的擴(kuò)展的.vm文件通通刪掉，InjectionConfig的initMap方法清空，focList相關(guān)全部刪除，修改cfg.setFileCreate(...)如下所示
cfg.setFileCreate(new IFileCreate() {
@Override
public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
// 如果已存在并且不是實(shí)體類則不創(chuàng)建
if (new File(filePath).exists() && fileType != FileType.ENTITY) {
return false;
}
// 判斷文件夾是否需要創(chuàng)建
checkDir(filePath);
return true;
}
});我們分析下：文件不存在一般是我們剛剛新建數(shù)據(jù)庫表或者新增了一個表，此時肯定是要創(chuàng)建文件的。文件存在的時候分兩種情況，一是文件是實(shí)體類，為了應(yīng)對將來可能的新增修改刪除字段，必須重寫，那么使用時不能對該實(shí)體做任何增刪改操作；二是文件非實(shí)體類，也就是service、map、xml等，因?yàn)榇a生成和第一次生成沒有什么區(qū)，也就沒有重寫的必要，而且很多時候也已經(jīng)編寫了代碼。
關(guān)于使用mybatis-plus如何實(shí)現(xiàn)生成mapper擴(kuò)展文件就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學(xué)到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。
免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進(jìn)行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。
億速云公眾號
手機(jī)網(wǎng)站二維碼