import com.baomidou.mybatisplus.generator.AutoGenerator; //导入依赖的package包/类
@Test
public void generator() {
    AutoGenerator mpg = new AutoGenerator();

    /** 全局配置 **/
    mpg.setGlobalConfig(setGlobalConfig());
    /**  数据源配置 **/
    mpg.setDataSource(setDataSourceConfig());
    /** 生成策略配置 **/
    mpg.setStrategy(setStrategyConfig());
    /** 生成包配置 **/
    mpg.setPackageInfo(setPackageConfig());
    /** 自定义模板配置 **/
    mpg.setTemplate(setTemplateConfig());


    /** 注入自定义配置，可以在 VM 中使用 MAP集合 设置的值 **/
    InjectionConfig cfg = new InjectionConfig() {
        @Override
        public void initMap() {
            Map<String, Object> map = new HashMap<String, Object>();
            map.put("Name", this.getConfig().getGlobalConfig().getAuthor() + "-MP");
            this.setMap(map);
        }
    };
    mpg.setCfg(cfg);

    /** 执行生成 **/
    mpg.execute();

    /** 打印注入设置 **/
    System.err.println(mpg.getCfg().getMap().get("Name"));
}
 

import com.baomidou.mybatisplus.generator.AutoGenerator; //导入依赖的package包/类
private void generateByTables(String packageName, String... tableNames) {
    GlobalConfig config = new GlobalConfig();
    String dbUrl = "jdbc:mysql://localhost:3306/mybatis-plus";
    DataSourceConfig dataSourceConfig = new DataSourceConfig();
    dataSourceConfig.setDbType(DbType.MYSQL)
        .setUrl(dbUrl)
        .setUsername("root")
        .setPassword("")
        .setDriverName("com.mysql.jdbc.Driver");
    StrategyConfig strategyConfig = new StrategyConfig();
    strategyConfig
        .setCapitalMode(true)
        .setEntityLombokModel(false)
        .setDbColumnUnderline(true)
        .setNaming(NamingStrategy.underline_to_camel)
        .entityTableFieldAnnotationEnable(enableTableFieldAnnotation)
        .fieldPrefix(fieldPrefix)//test_id -> id, test_type -> type
        .setInclude(tableNames);//修改替换成你需要的表名，多个表名传数组
    config.setActiveRecord(false)
        .setIdType(tableIdType)
        .setAuthor("K神带你飞")
        .setOutputDir("d:\\codeGen")
        .setFileOverride(true);
    if (!serviceClassNameStartWithI) {
        config.setServiceName("%sService");
    }
    new AutoGenerator().setGlobalConfig(config)
        .setDataSource(dataSourceConfig)
        .setStrategy(strategyConfig)
        .setPackageInfo(
            new PackageConfig()
                .setParent(packageName)
                .setController("controller")
                .setEntity("entity")
        ).execute();
}