Java SpringBoot 中，动态执行 bean 对象中的方法

根据不同的条件，调用不同的 bean 对象，执行对象中的方法
SpringUtils 工具类

1. package com.vipsoft.web.utils;
4. import cn.hutool.core.util.ArrayUtil;
5. import org.springframework.aop.framework.AopContext;
6. import org.springframework.beans.BeansException;
7. import org.springframework.beans.factory.NoSuchBeanDefinitionException;
8. import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
9. import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
10. import org.springframework.context.ApplicationContext;
11. import org.springframework.context.ApplicationContextAware;
12. import org.springframework.stereotype.Component;
14. /**
15. * Spring工具类 方便在非 Spring 管理环境中获取bean
16. *
17. */
18. @Component
19. public final class SpringUtils implements BeanFactoryPostProcessor, ApplicationContextAware {
21.     /**
22.      * Spring 应用上下文环境
23.      */
24.     private static ConfigurableListableBeanFactory beanFactory;
26.     private static ApplicationContext applicationContext;
28.     @Override
29.     public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
30.         SpringUtils.beanFactory = beanFactory;
31.     }
33.     @Override
34.     public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
35.         SpringUtils.applicationContext = applicationContext;
36.     }
38.     /**
39.      * 获取对象
40.      *
41.      * @param name
42.      * @return Object 一个以所给名字注册的bean的实例
43.      * @throws BeansException
44.      */
45.     @SuppressWarnings("unchecked")
46.     public static <T> T getBean(String name) throws BeansException {
47.         return (T) beanFactory.getBean(name);
48.     }
50.     /**
51.      * 获取类型为requiredType的对象
52.      *
53.      * @param clz
54.      * @return
55.      * @throws BeansException
56.      */
57.     public static <T> T getBean(Class<T> clz) throws BeansException {
58.         T result = (T) beanFactory.getBean(clz);
59.         return result;
60.     }
62.     /**
63.      * 如果BeanFactory包含一个与所给名称匹配的bean定义，则返回true
64.      *
65.      * @param name
66.      * @return boolean
67.      */
68.     public static boolean containsBean(String name) {
69.         return beanFactory.containsBean(name);
70.     }
72.     /**
73.      * 判断以给定名字注册的bean定义是一个singleton还是一个prototype。 如果与给定名字相应的bean定义没有被找到，将会抛出一个异常（NoSuchBeanDefinitionException）
74.      *
75.      * @param name
76.      * @return boolean
77.      * @throws NoSuchBeanDefinitionException
78.      */
79.     public static boolean isSingleton(String name) throws NoSuchBeanDefinitionException {
80.         return beanFactory.isSingleton(name);
81.     }
83.     /**
84.      * @param name
85.      * @return Class 注册对象的类型
86.      * @throws NoSuchBeanDefinitionException
87.      */
88.     public static Class<?> getType(String name) throws NoSuchBeanDefinitionException {
89.         return beanFactory.getType(name);
90.     }
92.     /**
93.      * 如果给定的bean名字在bean定义中有别名，则返回这些别名
94.      *
95.      * @param name
96.      * @return
97.      * @throws NoSuchBeanDefinitionException
98.      */
99.     public static String[] getAliases(String name) throws NoSuchBeanDefinitionException {
100.         return beanFactory.getAliases(name);
101.     }
103.     /**
104.      * 获取aop代理对象
105.      *
106.      * @param invoker
107.      * @return
108.      */
109.     @SuppressWarnings("unchecked")
110.     public static <T> T getAopProxy(T invoker) {
111.         return (T) AopContext.currentProxy();
112.     }
114.     /**
115.      * 获取当前的环境配置，无配置返回null
116.      *
117.      * @return 当前的环境配置
118.      */
119.     public static String[] getActiveProfiles() {
120.         return applicationContext.getEnvironment().getActiveProfiles();
121.     }
123.     /**
124.      * 获取当前的环境配置，当有多个环境配置时，只获取第一个
125.      *
126.      * @return 当前的环境配置
127.      */
128.     public static String getActiveProfile() {
129.         final String[] activeProfiles = getActiveProfiles();
130.         return ArrayUtil.isNotEmpty(activeProfiles) ? activeProfiles[0] : null;
131.     }
132. }

复制代码

执行类

1. package com.vipsoft.web.task;
3. import cn.hutool.core.date.DateUtil;
4. import org.slf4j.Logger;
5. import org.slf4j.LoggerFactory;
6. import org.springframework.stereotype.Component;
9. @Component("scheduletask")
10. public class ScheduleTask {
12.     protected final Logger logger = LoggerFactory.getLogger(this.getClass());
14.     public void execute(String param) {
15.         logger.info("执行 Schedule Task，当前时间：{}，任务参数：{}", DateUtil.now(), param);
16.     }
17. }

复制代码

测试类

1. package com.vipsoft.web;
3. import cn.hutool.core.util.StrUtil;
4. import com.vipsoft.web.utils.SpringUtils;
5. import org.junit.jupiter.api.Test;
6. import org.springframework.boot.test.context.SpringBootTest;
8. import java.lang.reflect.Method;
10. @SpringBootTest
11. public class SpringUtilTest {
13.     @Test
14.     void invokeTest() throws Exception {
15.         //点前面的做为 bean 名称，后面是方法名，（中是参数==参数可以放JSON）
16.         String invokeTarget = "scheduletask.execute('VipSoft Quartz')";
17.         invokeMethod(invokeTarget);
18.     }
20.     void invokeMethod(String invokeTarget) throws Exception {
21.         String beanName = StrUtil.subBefore(invokeTarget, ".", true);
22.         String methodName = StrUtil.subBetween(invokeTarget, ".", "(");
23.         String param = StrUtil.subBetween(invokeTarget, "(", ")");
25.         Object bean;
26.         if (StrUtil.count(beanName, ".") == 0) {
27.             bean = SpringUtils.getBean(beanName);
28.         } else {
29.             //Package 的形式得到 Bean，如： beanName="com.vipsoft.web.task.ScheduleTask"
30.             bean = Class.forName(beanName).newInstance();
31.         }
33.         if (bean != null) {
34.             if (StrUtil.isNotEmpty(param)) {
35.                 Method method = bean.getClass().getDeclaredMethod(methodName, String.class);
36.                 method.invoke(bean, param);
37.             } else {
38.                 Method method = bean.getClass().getDeclaredMethod(methodName);
39.                 method.invoke(bean);
40.             }
41.         }
42.     }
44. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！