MybatisPlus 生成代码的测试方法

pom.xml

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>${mybatisplus}</version>
</dependency>

JAVA方法 

private void generateByTables(boolean serviceNameStartWithI, String packageName,String tablePrefix, String... tableNames) {
    GlobalConfig config = new GlobalConfig();
    String dbUrl = "jdbc:mysql://10.3.8.26:3306/test";
    DataSourceConfig dataSourceConfig = new DataSourceConfig();
    dataSourceConfig.setDbType(DbType.MYSQL)
            .setUrl(dbUrl)
            .setUsername("root")
            .setPassword("123456")
            .setDriverName("com.mysql.jdbc.Driver");
    StrategyConfig strategyConfig = new StrategyConfig();
    strategyConfig
            .setCapitalMode(false)
            .setTablePrefix(tablePrefix)
            .setEntityLombokModel(false)
            .setRestControllerStyle(true)
            .setDbColumnUnderline(true)
            //.setExclude()
            .setNaming(NamingStrategy.underline_to_camel)
            .setInclude(tableNames);//修改替换成你需要的表名，多个表名传数组
    config.setActiveRecord(false)
            .setAuthor("XXX")
            .setOutputDir("D:\\workspace\\gpmp\\src\\main\\java")
            .setFileOverride(true)
            .setEnableCache(false)
            .setBaseResultMap(true)
            .setBaseColumnList(true);
    if (!serviceNameStartWithI) {
        config.setServiceName("%sService");
    }
    new AutoGenerator().setGlobalConfig(config)
            .setDataSource(dataSourceConfig)
            .setStrategy(strategyConfig)
            .setPackageInfo(
                    new PackageConfig()
                            .setParent(packageName)
                            .setController("controller")
                            .setEntity("entity")
            ).execute();
}

测试类： 

@Test
public void generateCode() {
    String packageName = "com.example.mybatisplus";
    boolean serviceNameStartWithI = true;//user -> UserService, 设置成true: user -> IUserService
    String tableprefix = "t_";
    
    String[] tableNames = new String[]{"t_student"};
    
    generateByTables(serviceNameStartWithI, packageName, tableprefix, tableNames);
}