From e58ac72bbbc14cfbd3d21dbcc3f1f55b3013bbb5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=99=B4=E9=9B=A8=E5=A4=9C?= <758366855@qq.com>
Date: Fri, 4 Sep 2020 11:56:03 +0800
Subject: [PATCH] =?UTF-8?q?feat(SpringUtil):=20=E5=A2=9E=E5=8A=A0=E6=96=B0?=
 =?UTF-8?q?=E7=89=B9=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

1.添加向Spring动态注册Bean的方法
---
 .../cn/hutool/extra/spring/SpringUtil.java    | 17 ++++++++++
 .../hutool/extra/spring/SpringUtilTest.java   | 32 ++++++++++++++++---
 2 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/hutool-extra/src/main/java/cn/hutool/extra/spring/SpringUtil.java b/hutool-extra/src/main/java/cn/hutool/extra/spring/SpringUtil.java
index af0be5e8c..59cd3e3cd 100644
--- a/hutool-extra/src/main/java/cn/hutool/extra/spring/SpringUtil.java
+++ b/hutool-extra/src/main/java/cn/hutool/extra/spring/SpringUtil.java
@@ -4,6 +4,7 @@ import cn.hutool.core.lang.TypeReference;
 import cn.hutool.core.util.ArrayUtil;
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.ApplicationContextAware;
+import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.core.ResolvableType;
 import org.springframework.stereotype.Component;
 
@@ -148,6 +149,22 @@ public class SpringUtil implements ApplicationContextAware {
 		final String[] activeProfiles = getActiveProfiles();
 		return ArrayUtil.isNotEmpty(activeProfiles) ? activeProfiles[0] : null;
 	}
+
+    /**
+     * 动态向Spring注册Bean
+     * <p>
+     * 由{@link org.springframework.beans.factory.BeanFactory} 实现，通过工具开放API
+     *
+     * @param beanName 名称
+     * @param bean     bean
+     * @param <T>      泛型
+     * @author shadow
+     * @date 2020-09-04
+     */
+    public static <T> void registerBean(String beanName, T bean) {
+        ConfigurableApplicationContext context = (ConfigurableApplicationContext) applicationContext;
+        context.getBeanFactory().registerSingleton(beanName, bean);
+    }
 }
 
 
diff --git a/hutool-extra/src/test/java/cn/hutool/extra/spring/SpringUtilTest.java b/hutool-extra/src/test/java/cn/hutool/extra/spring/SpringUtilTest.java
index e7fbaed87..50553751c 100644
--- a/hutool-extra/src/test/java/cn/hutool/extra/spring/SpringUtilTest.java
+++ b/hutool-extra/src/test/java/cn/hutool/extra/spring/SpringUtilTest.java
@@ -10,6 +10,7 @@ import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.context.annotation.Bean;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 
+import javax.swing.*;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -18,11 +19,34 @@ import java.util.Map;
 //@Import(cn.hutool.extra.spring.SpringUtil.class)
 public class SpringUtilTest {
 
+	/**
+	 * 注册bean
+	 *
+	 * 为了保证顺序采用顺序abcdef...形式命名方法
+	 */
+	@Test
+	public void a() {
+		Demo2 registerBean = new Demo2();
+		registerBean.id = 123;
+		registerBean.name = "222";
+		SpringUtil.registerBean("registerBean", registerBean);
+	}
+
+	/**
+	 * 验证注册的bean
+	 */
+	@Test
+	public void b() {
+		Demo2 registerBean = SpringUtil.getBean("registerBean");
+		Assert.assertEquals(123, registerBean.id);
+		Assert.assertEquals("222", registerBean.name);
+	}
+
 	@Test
 	public void getBeanTest(){
 		final Demo2 testDemo = SpringUtil.getBean("testDemo");
-		Assert.assertEquals(12345, testDemo.getId());
-		Assert.assertEquals("test", testDemo.getName());
+		Assert.assertEquals(12345, testDemo.id);
+		Assert.assertEquals("test", testDemo.name);
 	}
 
 	@Test
@@ -41,8 +65,8 @@ public class SpringUtilTest {
 		@Bean(name="testDemo")
 		public Demo2 generateDemo() {
 			Demo2 demo = new Demo2();
-			demo.setId(12345);
-			demo.setName("test");
+			demo.id = 12345;
+			demo.name= "test";
 			return demo;
 		}