springboot框架+mybatisPlus代码生成器

首先导入的pom文件：

 

<!-- freemarker 模板引擎 -->
<dependency>
   <groupId>org.freemarker</groupId>
   <artifactId>freemarker</artifactId>
   <version>2.3.28</version>
</dependency>

       <!--myBatisPlus开始-->
      <dependency>
         <groupId>org.projectlombok</groupId>
         <artifactId>lombok</artifactId>
         <optional>true</optional>
      </dependency>
  

<!-- 代码自动生成-->

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>2.2.0</version>
</dependency>

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

<dependency>
   <groupId>com.baomidou</groupId>
   <artifactId>mybatis-plus-generate</artifactId>
   <version>2.3</version>
</dependency>

<!-- velocity 模板引擎, 默认 -->
<dependency>
   <groupId>org.apache.velocity</groupId>
   <artifactId>velocity-engine-core</artifactId>
   <version>2.0</version>
</dependency>
<!-- beetl 模板引擎 -->
<dependency>
   <groupId>com.ibeetl</groupId>
   <artifactId>beetl</artifactId>
   <version>2.2.5</version>
</dependency>

<!-- Hutool工具包 -->
<dependency>
   <groupId>cn.hutool</groupId>
   <artifactId>hutool-all</artifactId>
   <version>5.0.3</version>
</dependency>

 

生成代码的MAIN方法：

package com.example.mybatisPlus.common;

import cn.hutool.core.util.ClassUtil;
import com.baomidou.mybatisplus.enums.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DbType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class MyGenerator {

    public static void main(String[] args) {
        MyGenerator generator = new MyGenerator();

        String packageName = "com.example.mybatisPlus";   //生成文件所在的包
        boolean serviceNameStartWithI = true;//user -> UserService, 设置成true: user -> IUserService

        generator.generateByTables(serviceNameStartWithI, packageName,"keying"
                ,"music"

        );
    }

    public static String getOutPath(){
        return ClassUtil.getClassPath().replace("target/classes/","src/main/java");
    }

    private void generateByTables(boolean serviceNameStartWithI, String packageName, String auth,String... tableNames) {
        GlobalConfig config = new GlobalConfig();
        String dbUrl = "jdbc:mysql://********:3306";
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        dataSourceConfig.setDbType(DbType.MYSQL)
                .setUrl(dbUrl)
                .setUsername("name")
                .setPassword("password")
                .setDriverName("com.mysql.jdbc.Driver");


        TemplateConfig tc = new TemplateConfig();
        tc.setController("controller.java");
        tc.setEntity("entity.java");
        tc.setMapper("mapper.java");
        tc.setXml("mapper.xml");
        tc.setService("service.java");
        tc.setServiceImpl("serviceImpl.java");

        StrategyConfig strategyConfig = new StrategyConfig();
        strategyConfig
//                .setCapitalMode(true)
                .setEntityLombokModel(true)
                .setDbColumnUnderline(true)
                .setNaming(NamingStrategy.underline_to_camel)
                .setRestControllerStyle(true) // rest风格的controller
                .setInclude(tableNames);//修改替换成你需要的表名，多个表名传数组
        config.setActiveRecord(true).setEnableCache(false)
                .setAuthor(auth).setIdType(IdType.UUID)
                .setFileOverride(true).setBaseColumnList(true).setBaseResultMap(true)
                .setOutputDir(getOutPath())
                .setFileOverride(true);
        if (!serviceNameStartWithI) {
            config.setServiceName("%sService");
        }
        new AutoGenerator().setGlobalConfig(config)
//                .setCfg(customerConfig(getOutPath() + "/" +packageName))  //自定义模板生成
                .setDataSource(dataSourceConfig)
                .setStrategy(strategyConfig)
                .setTemplateEngine(new FreemarkerTemplateEngine())
                .setPackageInfo(
                        new PackageConfig()
                                .setParent(packageName)
                                .setController("controller")
                                .setEntity("model")
                                .setMapper("dao")
                                .setXml("mapping")
                ).execute();

    }
    /**
     * 自定义生成文件
     */
    private static InjectionConfig customerConfig(String path) {
        InjectionConfig config = new InjectionConfig() {
            @Override
            public void initMap() {

            }
        };

        List<FileOutConfig> files = new ArrayList<FileOutConfig>();
        files.add(new FileOutConfig("/templates/entity.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                String entityFile = String.format((path + File.separator + "%s" + ".java"), tableInfo.getEntityName());
                return entityFile;
            }
        });
        files.add(new FileOutConfig("/templates/mapper.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                String entityFile = String.format((path + File.separator + "%s" + ".java"), tableInfo.getMapperName());
                return entityFile;
            }
        });
        files.add(new FileOutConfig("/templates/mapper.xml.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                String entityFile = String.format((path + File.separator + "%s" + ".java"), tableInfo.getXmlName());
                return entityFile;
            }
        });
        files.add(new FileOutConfig("/templates/service.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                String entityFile = String.format((path + File.separator + "%s" + ".java"), tableInfo.getServiceName());
                return entityFile;
            }
        });
        files.add(new FileOutConfig("/templates/serviceImpl.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                String entityFile = String.format((path + File.separator + "%s" + ".java"), tableInfo.getServiceImplName());
                return entityFile;
            }
        });
        files.add(new FileOutConfig("/templates/controller.java.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                String entityFile = String.format((path + File.separator + "%s" + ".java"), tableInfo.getControllerName());
                return entityFile;
            }
        });
        config.setFileOutConfigList(files);
        return config;
    }

}