自动注册扩展方法

Author: javamxd

src/main/java/org/ssssssss/magicapi/spring/boot/starter/MagicAPIAutoConfiguration.java

@@ -32,6 +32,8 @@
 import org.ssssssss.script.MagicModuleLoader;
 import org.ssssssss.script.MagicScript;
 import org.ssssssss.script.MagicScriptEngine;
+import org.ssssssss.script.functions.ExtensionMethod;
+import org.ssssssss.script.interpreter.AbstractReflection;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.sql.DataSource;
@@ -135,6 +137,7 @@ public MagicApiService magicApiService(DynamicDataSource dynamicDataSource) {
 
 	@Bean
 	public RequestHandler requestHandler(@Autowired(required = false) List<MagicModule> magicModules, //定义的模块集合
+										 @Autowired(required = false) List<ExtensionMethod> extensionMethods, //自定义的类型扩展
 										 MagicApiService magicApiService,
 										 // 动态数据源
 										 DynamicDataSource dynamicDataSource,
@@ -198,6 +201,15 @@ protected String getColumnKey(String columnName) {
 				MagicModuleLoader.addModule(module.getModuleName(), module);
 			}
 		}
+		if (extensionMethods != null) {
+			for (ExtensionMethod extension : extensionMethods) {
+				List<Class<?>> supports = extension.supports();
+				for (Class<?> support : supports) {
+					logger.info("注册扩展:{} -> {}", support, extension.getClass());
+					AbstractReflection.getInstance().registerExtensionClass(support, extension.getClass());
+				}
+			}
+		}
 		DatabaseQuery query = new DatabaseQuery(dynamicDataSource);
 		query.setResultProvider(resultProvider);
 		query.setPageProvider(pageProvider);