This commit is contained in:
shikong 2024-04-17 18:14:48 +08:00
parent 2999eecbfb
commit b27db9d51d
2 changed files with 8 additions and 7 deletions

View File

@ -16,7 +16,7 @@ import java.io.Serial;
* 设备通道信息 实体类
*
* @author shikong
* @since 2024-04-15
* @since 2024-04-17
*/
@Accessors(chain = true)
@Data(staticConstructor = "create")

View File

@ -38,19 +38,20 @@ public class OrmGenerator {
FlexDataSource flexDataSource = FlexGlobalConfig.getDefaultConfig().getDataSource();
Map<String, DataSource> dataSourceMap = flexDataSource.getDataSourceMap();
generateByDataSource(dataSourceMap);
String mysql = "mysql";
generateByDataSource(dataSourceMap, mysql);
}
private static void generateByDataSource(Map<String, DataSource> dataSourceMap) {
DataSource mysql = dataSourceMap.get("mysql");
private static void generateByDataSource(Map<String, DataSource> dataSourceMap, String dataSourceName) {
DataSource dataSource = dataSourceMap.get(dataSourceName);
//创建配置内容
GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setBasePackage("cn.skcks.wx.official.orm");
globalConfig.setSourceDir("./wx-official-orm/src/main/java");
globalConfig.setSourceDir(System.getProperty("user.dir") + "/wx-official-orm/src/main/java");
globalConfig.setAuthor(getGitUserName());
globalConfig.setEntityDataSource("mysql");
globalConfig.setEntityDataSource(dataSourceName);
globalConfig.setEntityGenerateEnable(true);
globalConfig.setEntityWithLombok(true);
globalConfig.setEntityGenerateEnable(true);
@ -65,7 +66,7 @@ public class OrmGenerator {
// 不生成的表
globalConfig.setUnGenerateTable("DATABASECHANGELOG", "DATABASECHANGELOGLOCK");
Generator generator = new Generator(mysql, globalConfig);
Generator generator = new Generator(dataSource, globalConfig);
generator.generate();
}
}