53 lines
1.7 KiB
Java
53 lines
1.7 KiB
Java
|
package com.example.easy.retry.util;
|
||
|
|
|
||
|
|
import org.springframework.beans.factory.annotation.Value;
|
||
|
|
|
||
|
|
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 {
|
||
|
|
|
||
|
|
@Value("${spring.datasource.url}")
|
||
|
|
private static String dataSourceUrl;
|
||
|
|
|
||
|
|
@Value("${spring.datasource.username}")
|
||
|
|
private static String userName;
|
||
|
|
|
||
|
|
@Value("${spring.datasource.password}")
|
||
|
|
private static String password;
|
||
|
|
|
||
|
|
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("maluxinyu")
|
||
|
|
.build();
|
||
|
|
|
||
|
|
// 策略配置
|
||
|
|
StrategyConfig strategyConfig = new StrategyConfig.Builder()
|
||
|
|
.addInclude("fail_order") // 需要生成的表名
|
||
|
|
.build();
|
||
|
|
|
||
|
|
// 包配置
|
||
|
|
PackageConfig packageConfig = new PackageConfig.Builder()
|
||
|
|
.parent("com.maluxinyu.easyretry")
|
||
|
|
.moduleName("easy-retry-springboot")
|
||
|
|
.build();
|
||
|
|
|
||
|
|
// 代码生成器
|
||
|
|
AutoGenerator generator = new AutoGenerator(dataSourceConfig)
|
||
|
|
.global(globalConfig)
|
||
|
|
.strategy(strategyConfig)
|
||
|
|
.packageInfo(packageConfig);
|
||
|
|
|
||
|
|
// 执行生成代码
|
||
|
|
generator.execute();
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|