mybatis-plus与springboot整合

      <!-- 代码生成器 -->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.4.1</version>
        </dependency>
        <!-- 模板引擎 -->
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
            <version>2.0</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.4.0</version>
        </dependency>

public class GeneratorUtils {
//    public static String  PROJECT_PATH = "";


    public static void main(String[] args) {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();
        mpg.setGlobalConfig(GeneratorUtils.globalConfig());
        mpg.setDataSource(GeneratorUtils.dataSourceConfig());
        mpg.setPackageInfo(GeneratorUtils.packageConfig("com.baomidou.ant"));
        mpg.setCfg(GeneratorUtils.injectionConfig());
        mpg.setTemplate(GeneratorUtils.templateConfig());
        //参数是表名称，也可以是表名称数组
        mpg.setStrategy(GeneratorUtils.strategyConfig("test_user"));
        mpg.setTemplateEngine(new VelocityTemplateEngine());
        mpg.execute();
    }
    
    public static Scanner scanner = new Scanner(System.in);
    // 获得当前项目的路径，如上图就是 F:stady_program/Mybatis_plus逆向工程
    public final static String PROJECT_PATH = System.getProperty("user.dir");
    public static String partentName = "com.baomidou.ant";

    // 最简单的配置
    public static TemplateConfig templateConfig() {
        TemplateConfig templateConfig = new TemplateConfig();


        return templateConfig.setXml(null);
    }





    // 判断输入的父项目是否存在
    public static boolean parentPackageExits(String project, String parentPackageName){
        StringBuilder path = new StringBuilder(PROJECT_PATH);
        if(!"".equals(project) && project != null){
            path.append("/" + project);
        }
        path.append(("/src/main/java/" + parentPackageName).replace(".","/"));
        File file = new File(path.toString());
        return file.exists();
    }



    public static InjectionConfig injectionConfig(){
        InjectionConfig injectionConfig = new InjectionConfig() {
            @Override
            public void initMap() {
                this.setMap(new HashMap<>()); // 实现InjectionConfig抽象类就需要初始化一个Map集合
            }
        };
        List<FileOutConfig> fileOutConfigList = new ArrayList<>();
        // 根据/templates/mapper.xml.ftl规则在指定的位置生成Mapper文件，可以在多个地方生成。
        String templatePath = "/templates/mapper.xml.vm";
//        String templatePath = "/templates/mapper.xml.ftl";
        fileOutConfigList.add(new FileOutConfig(templatePath) {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 返回Mapper文件的绝对路径
                String path = PROJECT_PATH + "/src/main/resources/mapper/" +
                        tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
                return path;
            }
        });
        // 将对Mapper文件的配置添加到文件输出对象中
        injectionConfig.setFileOutConfigList(fileOutConfigList);
        return injectionConfig;
    }


    public static StrategyConfig strategyConfig (String tableName){
        StrategyConfig strategyConfig = new StrategyConfig();
        strategyConfig.setCapitalMode(true)// 开启全局大写命名
                .setInclude(tableName)// 设置要映射的表
                .setNaming(NamingStrategy.underline_to_camel)// 下划线到驼峰的命名方式
                .setColumnNaming(NamingStrategy.underline_to_camel)// 下划线到驼峰的命名方式
                .setEntityColumnConstant(true)
                .setEntityTableFieldAnnotationEnable(true) //生成的实体有 对应的注解对应
                .setEntityLombokModel(true)// 是否使用lombok
                .setRestControllerStyle(true)// 是否开启rest风格
                .setTablePrefix("t_") // 去除前缀
                .setControllerMappingHyphenStyle(true); // localhost:8080/hello_a_2

        return strategyConfig;
    }


    public static PackageConfig packageConfig(String fatherPackageName){
        fatherPackageName = partentName;
        PackageConfig packageConfig = new PackageConfig();
        packageConfig.setParent(fatherPackageName) // 配置指定项目中各层的名称
                .setController("controller") // Controller层
                .setEntity("entity") // 实体层（pojo层)
                .setMapper("dao") // Dao 层
                .setService("service") // service层
                .setServiceImpl("service.serviceImpl"); // ServiceImp层

        return packageConfig;
    }


    public static GlobalConfig globalConfig(){
        GlobalConfig globalConfig = new GlobalConfig();
//        GlobalConfig gc = new GlobalConfig();

        // 输出文件路径
        globalConfig.setOutputDir(PROJECT_PATH + "/src/main/java")
                .setAuthor("Time Travel")// 设置作者名字
                .setOpen(false)// 是否打开资源管理器
                .setFileOverride(true)// 是否覆盖原来生成的
                .setIdType(IdType.AUTO)// 主键策略
                .setBaseResultMap(true)// 生成resultMap
                .setDateType(DateType.ONLY_DATE) // 设置时间格式，采用Date
                .setServiceName("%sService");// 生成的service接口名字首字母是否为I，这样设置就没有I
        //setBaseColumnList(true)  XML中生成基础列
        return globalConfig;
    }


    public static DataSourceConfig dataSourceConfig(){
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        dataSourceConfig.setUrl("jdbc:mysql://localhost:3306/test11?useUnicode=true&useSSL=false&serverTimezone=Asia/Shanghai")
                .setUsername("root1")
                .setPassword("root1")
                .setDriverName("com.mysql.jdbc.Driver").setDbType(DbType.MYSQL);

        return dataSourceConfig;
    }
 
}