这篇文章将为大家详细讲解有关使用mybatis-plus如何实现生成mapper扩展文件，文章内容质量较高，因此小编分享给大家做个参考，希望大家阅读完这篇文章后对相关知识有一定的了解。
我们都知道，mybatis-plus是一个mybatis的增强工具，为简化开发、提高效率而生，我们经常使用mybatis-plus生成controller、service、mapper等文件，对于简单的curd，可以直接使用mybatis-plus封装好的方法。
然而，我们经常有这样那样的需求，需要额外编写sql实现，如果直接在mapper.xml文件中编写，一旦数据库表结构改动需要重新生成文件就悲催了，不得不花大量精力修改代码。所以，这里介绍一种方式，自动生成mapper扩展文件，我们自定义编写的程序存放在扩展文件中，这样在数据库表结构改动时，不用担心程序会被覆盖，也不用修改代码。
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扩展文件
  熟悉mybatis-plus的朋友都知道，mybatis-plus提供了一款代码生成器，可以自动生成代码，我们就从这款代码生成器入手。
  代码生成器配置完毕后，运行时会执行 AutoGenerator.execute() 方法，我们先看看这个东东
熟悉的GlobalConfig、DataSourceConfig等等就不介绍了，我们关注的是InjectionConfig、TemplateConfig和ConfigBuilder。这里先讲述一下我们的思路：把ext文件通过配置直接生成，并保留mybatis-plus为service扩展的批量操作，我们需要三个文件，第一个文件生成ext.java，第二个文件生成ext.xml，第三个覆盖serviceImpl文件（保留mybatis-plus为service扩展的批量操作）
**************************干货来咯**************************
// 生成目录
public static final String OUTPUT_DIR = "项目路径/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 文件，并且已存在则不创建
if (filePath.contains(MAPPER_EXT) || fileType == FileType.CONTROLLER || fileType == FileType.SERVICE || fileType == FileType.SERVICE_IMPL) {
if (new File(filePath).exists()) {
return false;
}
}
// 判断文件夹是否需要创建
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} 服务实现类
* </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补充）
  我们概述中所述问题真的存在吗？mybatis-plus如果需要扩展文件那么他为什么不提供呢？当然是问题不存在，根本不需要扩展文件，如果你存在这样的问题，你总是覆盖文件说明你的用法有问题。
  这里直接说应该怎么使用，我们把本文所述的扩展的.vm文件通通删掉，InjectionConfig的initMap方法清空，focList相关全部删除，修改cfg.setFileCreate(...)如下所示
cfg.setFileCreate(new IFileCreate() {
@Override
public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
// 如果已存在并且不是实体类则不创建
if (new File(filePath).exists() && fileType != FileType.ENTITY) {
return false;
}
// 判断文件夹是否需要创建
checkDir(filePath);
return true;
}
});我们分析下：文件不存在一般是我们刚刚新建数据库表或者新增了一个表，此时肯定是要创建文件的。文件存在的时候分两种情况，一是文件是实体类，为了应对将来可能的新增修改删除字段，必须重写，那么使用时不能对该实体做任何增删改操作；二是文件非实体类，也就是service、map、xml等，因为代码生成和第一次生成没有什么区，也就没有重写的必要，而且很多时候也已经编写了代码。
关于使用mybatis-plus如何实现生成mapper扩展文件就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
