From 6cdbe957a71851680343900abbb11d31f9a8715f Mon Sep 17 00:00:00 2001
From: Redkale <8730487+redkale@users.noreply.github.com>
Date: Wed, 3 Jun 2020 14:03:46 +0800
Subject: [PATCH]

---
 src/org/redkale/boot/NodeHttpServer.java | 1 +
 src/org/redkale/boot/NodeServer.java     | 9 ++++++---
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/org/redkale/boot/NodeHttpServer.java b/src/org/redkale/boot/NodeHttpServer.java
index 91740f735..a19844813 100644
--- a/src/org/redkale/boot/NodeHttpServer.java
+++ b/src/org/redkale/boot/NodeHttpServer.java
@@ -346,6 +346,7 @@ public class NodeHttpServer extends NodeServer {
                 }
             }
         }
+        if (agent != null) this.messageAgents.put(agent.getName(), agent);
         //输出信息
         if (ss != null && !ss.isEmpty() && sb != null) {
             ss.sort((AbstractMap.SimpleEntry<String, String[]> o1, AbstractMap.SimpleEntry<String, String[]> o2) -> o1.getKey().compareTo(o2.getKey()));
diff --git a/src/org/redkale/boot/NodeServer.java b/src/org/redkale/boot/NodeServer.java
index 9732f33ca..a426f307e 100644
--- a/src/org/redkale/boot/NodeServer.java
+++ b/src/org/redkale/boot/NodeServer.java
@@ -85,6 +85,9 @@ public abstract class NodeServer {
     //远程模式的Service对象集合
     protected final Set<Service> remoteServices = new LinkedHashSet<>();
 
+    //MessageAgent对象集合
+    protected final Map<String, MessageAgent> messageAgents = new HashMap<>();
+
     private volatile int maxClassNameLength = 0;
 
     private volatile int maxNameLength = 0;
@@ -540,15 +543,15 @@ public abstract class NodeServer {
                 agent.deregister(this, protocol, localServices, remoteServices);
             }
         }
-        if (application.messageAgents != null) { //MQ
+        if (!this.messageAgents.isEmpty()) { //MQ
 
         }
     }
 
     //Server.start执行之后调用
     protected void postStartServer(Set<Service> localServices, Set<Service> remoteServices) {
-        if (application.messageAgents != null) { //MQ
-            final MessageAgent agent = application.messageAgents[0];
+        if (!this.messageAgents.isEmpty()) { //MQ
+            this.messageAgents.values().forEach(agent -> agent.start());
         }
     }