简单的负载均衡随机分配策略

license/src/main/java/iot/sixiang/license/auth/AuthManager.java

+package iot.sixiang.license.auth;
+
+import iot.sixiang.license.entity.Server;
+import iot.sixiang.license.service.ServerService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+import org.springframework.stereotype.Service;
+
+import javax.annotation.PostConstruct;
+import java.util.*;
+
+@Component
+@Slf4j
+public class AuthManager {
+    @Autowired
+    private ServerService serverService;
+
+    private Map<String, Server> allServers = null;
+
+    public AuthManager() {
+        allServers = new HashMap<String, Server>();
+    }
+
+    @PostConstruct
+    public void init() {
+        List<Server> servers = serverService.getServerList(0, 20);
+        for (Server server : servers) {
+            String serverIp = server.getServerIp();
+            allServers.put(serverIp, server);
+        }
+
+    }
+
+    public Server getBalanceServer(){
+        int count = allServers.size();
+        if(count==0){
+            return null;
+        }else{
+            Random random = new Random();
+            int index = random.nextInt(count);
+            List<Server> servers = new ArrayList<>(allServers.values());
+            return servers.get(index);
+        }
+
+
+    }
+
+
+}

license/src/main/java/iot/sixiang/license/device/DeviceManager.java

@@ -27,10 +27,6 @@ public class DeviceManager {
     @Autowired
     private DeviceServerHandler handler;
 
-    @Getter
-    @Setter
-    SessionContext sessionContext = null;
-
 
     public DeviceManager() {
         sessionContexts = new HashMap<String, SessionContext>();

license/src/main/java/iot/sixiang/license/event/CreateForwarClientEventHandler.java

 package iot.sixiang.license.event;
 
+import iot.sixiang.license.auth.AuthManager;
+import iot.sixiang.license.entity.Server;
 import iot.sixiang.license.forward.ForwardManager;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -15,18 +17,29 @@ public class CreateForwarClientEventHandler {
     ForwardManager forwardManager;
     @Autowired
     EventPublisher eventPublisher;
+    @Autowired
+    AuthManager authManager;
 
 
     public CreateForwarClientEventHandler() {
 
     }
 
-    //	@Async("asyncExecutor")
     @EventListener
     public void handlerEvent(CreateForwarClientEvent event) {
 
         String appId = event.getAppId();
-        forwardManager.startTcpClient(appId);
+
+        Server balanceServer = authManager.getBalanceServer();
+        if (balanceServer != null) {
+            String serverIp = balanceServer.getServerIp();
+            Integer port = balanceServer.getPort();
+            forwardManager.startTcpClient(serverIp, port, appId);
+        } else {
+            log.error("balanceServer is null");
+        }
+
+
     }
 
 

license/src/main/java/iot/sixiang/license/forward/ForwardManager.java

@@ -20,8 +20,6 @@ public class ForwardManager {
     @Autowired
     public ForwardClient client;
 
-    String serviceIP = "121.46.25.14";
-    int port = 18889;
     private Map<String, SessionContext> sessionContexts = null;
 
 
@@ -30,7 +28,7 @@ public class ForwardManager {
 
     }
 
-    public void startTcpClient(String appId) {
+    public void startTcpClient(String serviceIP,int port,String appId) {
         client.startTcp(serviceIP, port,appId);
     }
 

license/src/main/resources/application.properties

 server.port=8868
 logging.level.root=debug
 spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
-spring.datasource.url=jdbc:mysql://localhost:3306/firstwork?serverTimezone=GMT%2B8
+spring.datasource.url=jdbc:mysql://localhost:3306/iot_license?serverTimezone=GMT%2B8
 spring.datasource.username=root
 spring.datasource.password=123456
 mybatis-plus.mapper-locations=classpath:/mapper/**.xml