springboot controller和service动态注入_springboot controller service-优快云博客

本文链接：https://blog.youkuaiyun.com/oThrowsException/article/details/124022052

该代码示例展示了如何在Spring Boot应用中动态注册和卸载Bean以及Controller。通过`ApplicationBootstrap`类的静态方法，实现了从指定JAR文件加载类并注册为Bean，同时注册对应的Controller。另外，`registerApplication`类提供了类似的功能，包括获取所有Bean、注册和卸载Bean及Controller。这些方法利用了反射和Spring的内部API来操作Bean生命周期。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package test;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.FileSystemResourceLoader;
import org.springframework.stereotype.Component;
import org.springframework.util.ClassUtils;
import org.springframework.util.ReflectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;


@SpringBootApplication
@Component
public class ApplicationBootstrap {

    static ApplicationContext applicationContext;

    static String file = "F:\\download\\apache-maven-3.3.9-bin\\repository\\org\\example\\springboot-test-common\\1.0-SNAPSHOT\\" +
            "springboot-test-common-1.0-SNAPSHOT.jar";

    public static void main(String[] args) throws Exception {

        applicationContext = SpringApplication.run(ApplicationBootstrap.class, args);
        String clazzName = "test1.controller.MyTestController1";
        String clazzName1 = "test1.service.MyTestServiceImpl";
        registerBean(clazzName1);
        registerBean(clazzName);

        registerController(StringUtils.uncapitalize(clazzName.substring(clazzName.lastIndexOf(".") + 1)));
    }


    public static void registerBean(String clazzName) throws Exception{
        URL url = new File(file).toURI().toURL();

//        URLClassLoader loader = (URLClassLoader) ApplicationBootstrap.class.getClassLoader();
//        Method add = URLClassLoader.class.getDeclaredMethod("addURL", new Class[] { URL.class });
//        add.setAccessible(true);
//        add.invoke(loader, url);

        URLClassLoader loader = new URLClassLoader(new URL[]{url});
//        String clazzName = "test1.controller.MyTestController1";
        Class<?> clazz = loader.loadClass(clazzName);


//        applicationContext.getClassLoader().loadClass("test.MyController");
//        DefaultListableBeanFactory factory = (DefaultListableBeanFactory) applicationContext.getAutowireCapableBeanFactory();

        BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(clazz);
        BeanDefinition beanDefinition = beanDefinitionBuilder.getRawBeanDefinition();

        //设置当前bean定义对象是单利的
        beanDefinition.setScope("singleton");

        //将变量首字母置小写
        String beanName = StringUtils.uncapitalize(clazzName);

        beanName =  beanName.substring(beanName.lastIndexOf(".")+1);
        beanName = StringUtils.uncapitalize(beanName);

        //将applicationContext转换为ConfigurableApplicationContext
        ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) applicationContext;

        // 获取bean工厂并转换为DefaultListableBeanFactory
        DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory) configurableApplicationContext.getBeanFactory();


        defaultListableBeanFactory.registerBeanDefinition(beanName, beanDefinition);
    }


    /**
     * 注册Controller
     * @param controllerBeanName
     * @throws Exception
     */
    public static void registerController(String controllerBeanName) throws Exception{
        final RequestMappingHandlerMapping requestMappingHandlerMapping=
                applicationContext.getBean(RequestMappingHandlerMapping.class);
        if(requestMappingHandlerMapping!=null){
            String handler=controllerBeanName;
            Object controller= applicationContext.getBean(handler);
            if(controller==null){
                return;
            }
//            unregisterController(controllerBeanName);
            //注册Controller
            Method method=requestMappingHandlerMapping.getClass().getSuperclass().getSuperclass().
                    getDeclaredMethod("detectHandlerMethods",Object.class);
            //将private改为可使用
            method.setAccessible(true);
            method.invoke(requestMappingHandlerMapping,handler);
        }
    }

    /**
     * 去掉Controller的Mapping
     * @param controllerBeanName
     */
    public static void unregisterController(String controllerBeanName){
        final RequestMappingHandlerMapping requestMappingHandlerMapping=(RequestMappingHandlerMapping)
                applicationContext.getBean("requestMappingHandlerMapping");
        if(requestMappingHandlerMapping!=null){
            String handler=controllerBeanName;
            Object controller= applicationContext.getBean(handler);
            if(controller==null){
                return;
            }
            final Class<?> targetClass=controller.getClass();
            ReflectionUtils.doWithMethods(targetClass, new ReflectionUtils.MethodCallback() {
                public void doWith(Method method) {
                    Method specificMethod = ClassUtils.getMostSpecificMethod(method, targetClass);
                    try {
                        Method createMappingMethod = RequestMappingHandlerMapping.class.
                                getDeclaredMethod("getMappingForMethod", Method.class, Class.class);
                        createMappingMethod.setAccessible(true);
                        RequestMappingInfo requestMappingInfo =(RequestMappingInfo)
                                createMappingMethod.invoke(requestMappingHandlerMapping,specificMethod,targetClass);
                        if(requestMappingInfo != null) {
                            requestMappingHandlerMapping.unregisterMapping(requestMappingInfo);
                        }
                    }catch (Exception e){
                        e.printStackTrace();
                    }
                }
            }, ReflectionUtils.USER_DECLARED_METHODS);
        }
    }
}

registerApplication.java：注册、卸载、获取Bean,Controller操作

package chen.wset.utils;


import org.springframework.beans.factory.config.BeanDefinition;
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.util.ClassUtils;
import org.springframework.util.ReflectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class registerApplication {

    // 获取ApplicationContext
    private ApplicationContext applicationContext = springContextUtil.getApplicationContext();

    //将applicationContext转换为ConfigurableApplicationContext
    private ConfigurableApplicationContext configurableApplicationContext =
            (ConfigurableApplicationContext) applicationContext;

    // 获取bean工厂并转换为DefaultListableBeanFactory
    private DefaultListableBeanFactory defaultListableBeanFactory =
            (DefaultListableBeanFactory) configurableApplicationContext.getBeanFactory();


    public registerApplication() {
    }

    /**
     * 注册bean
     * @param className
     * @throws Exception
     */
    public void registerBean(String className) throws Exception {

        //获取class
        Class reclass = Class.forName(className);


        BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(reclass);
        BeanDefinition beanDefinition = beanDefinitionBuilder.getRawBeanDefinition();

        //设置当前bean定义对象是单利的
        beanDefinition.setScope("singleton");

        //将变量首字母置小写
        String beanName = StringUtils.uncapitalize(className);

        beanName = beanName.substring(beanName.lastIndexOf(".") + 1);
        beanName = StringUtils.uncapitalize(beanName);

        defaultListableBeanFactory.registerBeanDefinition(beanName, beanDefinition);
    }

    /**
     * 卸载bean
     * @param className
     * @throws Exception
     */
    public void unregisterBean(String className) throws Exception {

        defaultListableBeanFactory.removeBeanDefinition(className);
    }

    /**
     * 获取所有bean
     * @return
     */
    public List<String> getBeans() {
        List<String> list = new ArrayList<>();
        String[] beans = applicationContext.getBeanDefinitionNames();
        Arrays.sort(beans);
        for (String bean : beans) {
            list.add(applicationContext.getBean(bean).getClass().getName());
        }
        return list;
    }

    /**
     * 注册controller
     * @param className
     * @throws Exception
     */
    public void registerController(String className) throws Exception {

        //先注册bean
        registerBean(className);

        final RequestMappingHandlerMapping requestMappingHandlerMapping =
                applicationContext.getBean(RequestMappingHandlerMapping.class);

        if (requestMappingHandlerMapping != null) {
            String handler = StringUtils.uncapitalize(className.substring(className.lastIndexOf(".") + 1));
            Object controller = applicationContext.getBean(handler);
            if (controller == null) {
                return;
            }

            //注册Controller
            Method method = requestMappingHandlerMapping.getClass().getSuperclass().getSuperclass().
                    getDeclaredMethod("detectHandlerMethods", Object.class);
            //将private改为可使用
            method.setAccessible(true);
            method.invoke(requestMappingHandlerMapping, handler);
        }
    }


    /**
     * 卸载controller
     * @param className
     * @throws Exception
     */
    public void unregisterController(String className) throws Exception {
        final RequestMappingHandlerMapping requestMappingHandlerMapping = (RequestMappingHandlerMapping)
                applicationContext.getBean("requestMappingHandlerMapping");
        if (requestMappingHandlerMapping != null) {
            String handler = StringUtils.uncapitalize(className.substring(className.lastIndexOf(".") + 1));
            Object controller = applicationContext.getBean(handler);
            if (controller == null) {
                return;
            }
            final Class<?> targetClass = controller.getClass();
            ReflectionUtils.doWithMethods(targetClass, new ReflectionUtils.MethodCallback() {
                public void doWith(Method method) {
                    Method specificMethod = ClassUtils.getMostSpecificMethod(method, targetClass);
                    try {
                        Method createMappingMethod = RequestMappingHandlerMapping.class.
                                getDeclaredMethod("getMappingForMethod", Method.class, Class.class);
                        createMappingMethod.setAccessible(true);
                        RequestMappingInfo requestMappingInfo = (RequestMappingInfo)
                                createMappingMethod.invoke(requestMappingHandlerMapping, specificMethod, targetClass);
                        if (requestMappingInfo != null) {
                            requestMappingHandlerMapping.unregisterMapping(requestMappingInfo);
                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }, ReflectionUtils.USER_DECLARED_METHODS);

            //卸载bean
            unregisterBean(className);
        }
    }

}

package chen.wset.utils;

import org.springframework.context.ApplicationContext;

public class springContextUtil {
    private static ApplicationContext applicationContext;

    //获取上下文
    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    //设置上下文
    public static void setApplicationContext(ApplicationContext applicationContext) {
        springContextUtil.applicationContext = applicationContext;
    }

}