48 lines
1.6 KiB
Java
48 lines
1.6 KiB
Java
package com.example.snailjob.util;
|
|
|
|
import com.baomidou.mybatisplus.generator.AutoGenerator;
|
|
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
|
|
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
|
|
import com.baomidou.mybatisplus.generator.config.PackageConfig;
|
|
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
|
|
|
|
public class CodeGen {
|
|
|
|
private static String dataSourceUrl = "jdbc:mysql://localhost:3306/snail_job_260?useSSL=false&characterEncoding=utf8&useUnicode=true";
|
|
|
|
private static String userName = "root";
|
|
|
|
private static String password = "root";
|
|
|
|
public static void main(String[] args) {
|
|
DataSourceConfig dataSourceConfig = new DataSourceConfig.Builder(dataSourceUrl, userName, password).build();
|
|
|
|
// 全局配置
|
|
GlobalConfig globalConfig = new GlobalConfig.Builder()
|
|
.outputDir("src/main/java")
|
|
.author("xiaowoniu")
|
|
.build();
|
|
|
|
// 策略配置
|
|
StrategyConfig strategyConfig = new StrategyConfig.Builder()
|
|
.addInclude("workflow") // 需要生成的表名
|
|
.build();
|
|
|
|
// 包配置
|
|
PackageConfig packageConfig = new PackageConfig.Builder()
|
|
.parent("com.aizuda.snailjob.template.datasource.persistence.po")
|
|
.moduleName("snail-job-springboot")
|
|
.build();
|
|
|
|
// 代码生成器
|
|
AutoGenerator generator = new AutoGenerator(dataSourceConfig)
|
|
.global(globalConfig)
|
|
.strategy(strategyConfig)
|
|
.packageInfo(packageConfig);
|
|
|
|
// 执行生成代码
|
|
generator.execute();
|
|
|
|
}
|
|
}
|