mybaitis-plus代码生成类

package com.niu.iam.gen;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Scanner;

/**
 * mybatis plus 代码生成器17  */
public class MybatisGenerator {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        System.out.println("请输入数据库名称？");
        String dbName = scanner.nextLine();
        System.out.println("请输入作者名称？");
        String author = scanner.nextLine();
        System.out.println("请输入所属父包名称？");
        String rootPakName = scanner.nextLine();
        System.out.println("请输入生成的表明（注：多个表之间使用英文逗号分隔）？");
        String tableNames = scanner.nextLine();

        FastAutoGenerator
                .create(new DataSourceConfig.Builder("jdbc:mysql://172.16.1.67/" + dbName, "root", "if^Ve$yPRfRx#x48"))
                .globalConfig(builder -> builder
                        .disableOpenDir()
                        .outputDir("D:\iam-system\src\main\java\com\mybatis\gen\tmp")    // 输出路径
                        .author(author)                                 // 作者
                        .enableSwagger()                                // 开启swagger
                        .dateType(DateType.ONLY_DATE)                   // 日期类型仅使用 Date类型
                        .commentDate("yyyy-MM-dd"))
                .packageConfig(builder -> builder
                        .parent(rootPakName)                              // 父包名
                        .entity("entity")
                        .service("service")
                        .serviceImpl("service.impl")
                        .mapper("mapper")
                        .xml("mapper.xml")
                        .controller("controller"))
                .strategyConfig(builder -> builder
                        .addInclude(getTables(tableNames))
                        .entityBuilder().enableChainModel()
                        .enableLombok()
                        .enableFileOverride()
                        .enableTableFieldAnnotation()
                        .enableActiveRecord()
                        .controllerBuilder().enableFileOverride()
//                        .superClass(BaseController.class)
                        .enableRestStyle()
                        .mapperBuilder().enableFileOverride()
                        .enableBaseColumnList()
                        .enableBaseResultMap()
                        .serviceBuilder().enableFileOverride()
                        .build())
                .execute();
    }

    /**
     * 获取表名称
     */
    private static List<String> getTables(String tableNames) {
        return "all".equals(tableNames) ? Collections.singletonList("*") : Arrays.asList(tableNames.split(","));
    }
}