代码规范

im-server/src/main/java/com/bx/imserver/netty/IMServerGroup.java

@@ -22,6 +22,20 @@ public class IMServerGroup implements CommandLineRunner  {
     @Autowired
     private List<IMServer> imServers;
 
+    /***
+     * 判断服务器是否就绪
+     *
+     * @return
+     **/
+    public boolean isReady(){
+        for(IMServer imServer:imServers){
+            if(!imServer.isReady()){
+                return  false;
+            }
+        }
+        return true;
+    }
+
     @Override
     public void run(String... args) throws Exception {
         // 初始化SERVER_ID

im-server/src/main/java/com/bx/imserver/task/AbstractPullMessageTask.java

@@ -1,6 +1,6 @@
 package com.bx.imserver.task;
 
-import com.bx.imserver.netty.ws.WebSocketServer;
+import com.bx.imserver.netty.IMServerGroup;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -17,7 +17,7 @@ public  abstract class AbstractPullMessageTask{
     private ExecutorService executorService;
 
     @Autowired
-    private WebSocketServer WSServer;
+    private IMServerGroup serverGroup;
 
     public  AbstractPullMessageTask(){
         this.threadNum = 1;
@@ -38,7 +38,7 @@ public  abstract class AbstractPullMessageTask{
                 @Override
                 public void run() {
                     try{
-                        if(WSServer.isReady()){
+                        if(serverGroup.isReady()){
                             pullMessage();
                         }
                         Thread.sleep(100);