Support rabbitmq multi-connection (#541)

hippo4j-adapter/hippo4j-adapter-rabbitmq/src/main/java/cn/hippo4j/adapter/rabbitmq/RabbitMQThreadPoolAdapter.java

@@ -17,6 +17,12 @@
 
 package cn.hippo4j.adapter.rabbitmq;
 
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.ThreadPoolExecutor;
+
 import cn.hippo4j.adapter.base.ThreadPoolAdapter;
 import cn.hippo4j.adapter.base.ThreadPoolAdapterParameter;
 import cn.hippo4j.adapter.base.ThreadPoolAdapterState;
@@ -25,16 +31,11 @@ import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+
 import org.springframework.amqp.rabbit.connection.AbstractConnectionFactory;
 import org.springframework.boot.context.event.ApplicationStartedEvent;
 import org.springframework.context.ApplicationListener;
 
-import java.util.List;
-import java.util.Map;
-import java.util.Objects;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.ThreadPoolExecutor;
-
 import static cn.hippo4j.common.constant.ChangeThreadPoolConstants.CHANGE_DELIMITER;
 
 /**
@@ -48,12 +49,7 @@ public class RabbitMQThreadPoolAdapter implements ThreadPoolAdapter, Application
 
 	private static final String FiledName = "executorService";
 
-    /**
-     * TODO Configurable name
-     */
-    private static final String RABBITMQ_EXECUTOR_SERVICE = "Rabbitmq_Executor_Service";
-
-    private final AbstractConnectionFactory abstractConnectionFactory;
+	private final Map<String, AbstractConnectionFactory> abstractConnectionFactoryMap;
 
 	private final Map<String, ThreadPoolExecutor> RABBITMQ_THREAD_POOL_TASK_EXECUTOR = Maps.newHashMap();
 
@@ -103,15 +99,19 @@ public class RabbitMQThreadPoolAdapter implements ThreadPoolAdapter, Application
 
 	@Override
 	public void onApplicationEvent(ApplicationStartedEvent event) {
-        ExecutorService executor = (ExecutorService) ReflectUtil.getFieldValue(abstractConnectionFactory, FiledName);
+		abstractConnectionFactoryMap.forEach((beanName, abstractConnectionFactor) -> {
+			ExecutorService executor = (ExecutorService) ReflectUtil.getFieldValue(abstractConnectionFactor, FiledName);
 			if (Objects.nonNull(executor)) {
 				if (executor instanceof ThreadPoolExecutor) {
 					ThreadPoolExecutor threadPoolTaskExecutor = (ThreadPoolExecutor) executor;
-                RABBITMQ_THREAD_POOL_TASK_EXECUTOR.put(RABBITMQ_EXECUTOR_SERVICE, threadPoolTaskExecutor);
-                log.info("Rabbitmq executor name {}", RABBITMQ_EXECUTOR_SERVICE);
-            } else {
+					RABBITMQ_THREAD_POOL_TASK_EXECUTOR.put(beanName, threadPoolTaskExecutor);
+					log.info("Rabbitmq executor name {}", beanName);
+				}
+				else {
 					log.warn("Custom thread pools only support ThreadPoolExecutor");
 				}
 			}
+
+		});
 	}
 }

hippo4j-spring-boot/hippo4j-spring-boot-starter-adapter/hippo4j-spring-boot-starter-adapter-rabbitmq/src/main/java/cn/hippo4j/springboot/starter/adapter/rabbitmq/RabbitMQAdapterAutoConfiguration.java

@@ -17,9 +17,12 @@
 
 package cn.hippo4j.springboot.starter.adapter.rabbitmq;
 
+import java.util.Map;
+
 import cn.hippo4j.adapter.rabbitmq.RabbitMQThreadPoolAdapter;
 import cn.hippo4j.common.config.ApplicationContextHolder;
 import lombok.RequiredArgsConstructor;
+
 import org.springframework.amqp.rabbit.connection.AbstractConnectionFactory;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
@@ -33,7 +36,7 @@ import org.springframework.context.annotation.Configuration;
 @RequiredArgsConstructor
 public class RabbitMQAdapterAutoConfiguration {
 
-    private final AbstractConnectionFactory abstractConnectionFactories;
+    private final Map<String, AbstractConnectionFactory> stringAbstractConnectionFactoryMap;
 
     @Bean
     @ConditionalOnMissingBean
@@ -45,6 +48,6 @@ public class RabbitMQAdapterAutoConfiguration {
     @SuppressWarnings("all")
     @ConditionalOnProperty(name = "spring.rabbitmq.host")
     public RabbitMQThreadPoolAdapter rabbitMQThreadPoolAdapter(ApplicationContextHolder applicationContextHolder) {
-        return new RabbitMQThreadPoolAdapter(abstractConnectionFactories);
+        return new RabbitMQThreadPoolAdapter(stringAbstractConnectionFactoryMap);
     }
 }