lilonglong/magic-api-gitee
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

代码优化

Browse Source
This commit is contained in:
mxd 2021-10-28 22:00:48 +08:00
parent ca20cec394
commit b5c289dbdf
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
magic-api-spring-boot-starter/src/main/java/org/ssssssss/magicapi/spring/boot/starter/MagicAPIAutoConfiguration.java
View File

@ -490,9 +490,10 @@ public class MagicAPIAutoConfiguration implements WebMvcConfigurer, WebSocketCon
logger.info("注册模块:{} -> {}", module.getModuleName(), module.getClass());
MagicResourceLoader.addModule(module.getModuleName(), module);
});
MagicResourceLoader.addModule("db", new DynamicModuleImport(SQLModule.class, context -> {
sqlModule.setDataSourceNode(dynamicDataSource.getDataSource(context.getString(Options.DEFAULT_DATA_SOURCE.getValue())));
return sqlModule;
MagicResourceLoader.addModule(sqlModule.getModuleName(), new DynamicModuleImport(SQLModule.class, context -> {
SQLModule newSqlModule = sqlModule.cloneSQLModule();
newSqlModule.setDataSourceNode(dynamicDataSource.getDataSource(context.getString(Options.DEFAULT_DATA_SOURCE.getValue())));
return newSqlModule;
}));
MagicResourceLoader.getModuleNames().stream().filter(importModules::contains).forEach(moduleName -> {
logger.info("自动导入模块：{}", moduleName);

2
magic-api/src/main/java/org/ssssssss/magicapi/modules/SQLModule.java
View File

@ -185,7 +185,7 @@ public class SQLModule extends HashMap<String, SQLModule> implements MagicModule
}
@UnableCall
private SQLModule cloneSQLModule() {
public SQLModule cloneSQLModule() {
SQLModule sqlModule = new SQLModule();
sqlModule.setDynamicDataSource(this.dynamicDataSource);
sqlModule.setDataSourceNode(this.dataSourceNode);