Kaynağa Gözat

代码优化

mxd 3 yıl önce
ebeveyn
işleme
b5c289dbdf

+ 4 - 3
magic-api-spring-boot-starter/src/main/java/org/ssssssss/magicapi/spring/boot/starter/MagicAPIAutoConfiguration.java

@@ -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);

+ 1 - 1
magic-api/src/main/java/org/ssssssss/magicapi/modules/SQLModule.java

@@ -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);