MybatisPlus 3.5.1 代码生成器自定义Service层

码小农

已于 2022-09-26 21:56:34 修改

阅读量1.2k

点赞数

分类专栏： java 文章标签： java mybatis

于 2022-09-26 21:54:56 首次发布

本文链接：https://blog.csdn.net/xwu_09/article/details/127060987

版权

java 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.1</version>
</dependency>

使用Freemarker引擎模板

<dependency>
	<groupId>org.freemarker</groupId>
	<artifactId>freemarker</artifactId>
	<version>2.3.31</version>
</dependency>

OK，按照官方文档，写一个main方法

public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:mysql://192.168.10.1:3306/platform", "root", "root")
                .globalConfig(builder -> {
                    builder.author("wux") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir(System.getProperty("user.dir") + "/src/main/java"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.wux") // 设置父包名
                            .moduleName("platform") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, System.getProperty("user.dir")+"/src/main/resources/mapper")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder
                    		// 表前缀
                    		.likeTable(new LikeTable("wux_"))
                    		// 生成的entity过滤掉前缀
                    		.addTablePrefix("wux_");
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }

然后，service层得到的结构如下图
在这里插入图片描述
但其实我并不想使用 IService 和 ServiceImpl 这样的结构，Service只要有一层就可以了。
所以，我查了源码，发现代码生成器最后依赖模板生成，那只要改了模板结构就可以了。
找到项目依赖的 mybatis-plus-generator 3.5.1 jar包，使用解压缩文件打开，把templates包下的 serviceImpl.java.flt 复制出来，用文本编辑器打开，修改内容后再复制到jar包里面覆盖原来的就可以了。
在这里插入图片描述

在这里插入图片描述
还要把 serviceImpl 文件生成规则改一下，最终代码如下

public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:mysql://192.168.10.1:3306/platform", "root", "root")
                .globalConfig(builder -> {
                    builder.author("wux") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir(System.getProperty("user.dir") + "/src/main/java"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.wux") // 设置父包名
                            .moduleName("platform") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, System.getProperty("user.dir")+"/src/main/resources/mapper")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder
                    		// 表前缀
                    		.likeTable(new LikeTable("wux_"))
                    		.serviceBuilder()
                            .convertServiceImplFileName((entityName) -> entityName + "Service")
                    		// 生成的entity过滤掉前缀
                    		.addTablePrefix("wux_");
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }

最后再执行一次代码，欸，就得到你想要的了
在这里插入图片描述