[JSMOD@2_495_COMMIT] Util

Author: magiclu550

JSMod2Core/src/main/java/cn/jsmod2/core/plugin/BeanRegister.java

@@ -117,7 +117,6 @@ public Plugin loadPlugin(File jar) {
                             continue;
                         }
                         Class<?> pluginClass = classLoader.loadClass(mainName);
-                        SpringbootLoader.loadBean(pluginClass);
                         Main main = pluginClass.getAnnotation(Main.class);
                         if(main!=null){
                             PluginFileVO vo = new PluginFileVO(main.name(),pluginClass.getName(),main.description(),main.version());
@@ -180,6 +179,7 @@ public Plugin loadPluginInfo(Object plugin,PluginFileVO vo,JarFile jarFile,URLCl
                     String name = entry.getName();
                     if(name.endsWith(".class")){
                         Class<?> clz = loader.loadClass(name.substring(0,name.lastIndexOf(".")).replace("/","."));
+                        SpringContextUtil.loadBean(clz);
                         if(register.command()) {
                             if (mostSuperClass(clz).equals(NativeCommand.class) && !exclusionsCommand.contains(clz)) {
                                 Object obj = clz.getConstructor(Plugin.class).newInstance(pluginObject);

JSMod2Core/src/main/java/cn/jsmod2/core/plugin/PluginClassLoader.java

@@ -0,0 +1,75 @@
+package cn.jsmod2.core.plugin;
+
+
+import cn.jsmod2.core.utils.Utils;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.support.BeanDefinitionBuilder;
+import org.springframework.beans.factory.support.DefaultListableBeanFactory;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.stereotype.Component;
+import org.springframework.stereotype.Controller;
+import org.springframework.stereotype.Repository;
+import org.springframework.stereotype.Service;
+import org.springframework.util.ReflectionUtils;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
+import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;
+
+import java.lang.reflect.Method;
+import java.lang.reflect.Modifier;
+
+public class SpringContextUtil {
+
+    private static ApplicationContext applicationContext;
+
+    public static void setApplicationContext(ApplicationContext applicationContext) {
+        SpringContextUtil.applicationContext = applicationContext;
+    }
+
+    public static ApplicationContext getApplicationContext() {
+        return applicationContext;
+    }
+
+    public static void loadBean(Class clz){
+        if(isSpringBean(clz)){
+            String beanName = StringUtils.uncapitalize(clz.getSimpleName());
+            ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) SpringContextUtil.getApplicationContext();
+            DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory)configurableApplicationContext.getBeanFactory();
+            BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(clz);
+            defaultListableBeanFactory.registerBeanDefinition(beanName, beanDefinitionBuilder.getRawBeanDefinition());
+            RequestMappingHandlerMapping mapping = configurableApplicationContext.getBean(RequestMappingHandlerMapping.class);
+            Method method = ReflectionUtils.findMethod(RequestMappingHandlerMapping.class,"getMappingForMethod",Method.class,Class.class);
+            method.setAccessible(true);
+            Method[] methods = clz.getDeclaredMethods();
+            for(Method m:methods){
+                //RequestMapping("/aa")+RequestMapping("/name");
+                RequestMapping rm = m.getAnnotation(RequestMapping.class);
+                String[] paths = rm.value();
+                for(String path:paths) {
+                    try {
+                        RequestMappingInfo mappingInfo = (RequestMappingInfo) method.invoke(mapping, m, path);
+                        mapping.registerMapping(mappingInfo, configurableApplicationContext.getBean(beanName), m);
+                    }catch (Exception e){
+                        Utils.printException(e);
+                    }
+                }
+            }
+
+        }
+    }
+
+    public static boolean isSpringBean(Class clz){
+        if(clz == null)return false;
+        if(clz.isInterface())return false;
+        if(Modifier.isAbstract(clz.getModifiers()))return false;
+        if (clz.getAnnotation(Controller.class)!=null)return true;
+        if(clz.getAnnotation(RestController.class)!=null)return true;
+        if(clz.getAnnotation(Component.class)!=null)return true;
+        if(clz.getAnnotation(Repository.class)!=null)return true;
+        if(clz.getAnnotation(Service.class)!=null)return true;
+        return false;
+    }
+
+}

JSMod2Core/src/main/java/cn/jsmod2/core/plugin/SpringContextUtil.java

@@ -3,6 +3,7 @@
 import cn.jsmod2.core.Application;
 import cn.jsmod2.core.annotations.ServerApplication;
 import cn.jsmod2.core.log.ServerLogger;
+import cn.jsmod2.core.plugin.SpringContextUtil;
 import cn.jsmod2.core.utils.Utils;
 
 import java.util.concurrent.CountDownLatch;
@@ -14,7 +15,7 @@ public void start(String[] args){
         Utils.TryCatch(()->{
             CountDownLatch latch = new CountDownLatch(1);
             new Thread(()->{
-                Starter.run(args);
+                SpringContextUtil.setApplicationContext(Starter.run(args));
                 latch.countDown();
             }).start();
             latch.await();

JSMod2Core/src/main/java/cn/jsmod2/core/plugin/SpringbootLoader.java

@@ -5,7 +5,7 @@
     !author MagicLu550
 -->
 
-<?import java.lang.*?>
+
 <?import javafx.scene.control.*?>
 <?import javafx.scene.layout.*?>