Commit 1f444089 authored by duanruiming's avatar duanruiming

[update] socketClient优化

parent 35a10bee
...@@ -3,6 +3,7 @@ package net.wanji.utc.cache; ...@@ -3,6 +3,7 @@ package net.wanji.utc.cache;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import net.wanji.databus.po.CrossInfoPO; import net.wanji.databus.po.CrossInfoPO;
import net.wanji.utc.websocket.WebSocketClientHandler; import net.wanji.utc.websocket.WebSocketClientHandler;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.CommandLineRunner; import org.springframework.boot.CommandLineRunner;
import org.springframework.core.annotation.Order; import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
...@@ -13,6 +14,7 @@ import javax.websocket.Session; ...@@ -13,6 +14,7 @@ import javax.websocket.Session;
import javax.websocket.WebSocketContainer; import javax.websocket.WebSocketContainer;
import java.net.URI; import java.net.URI;
import java.util.List; import java.util.List;
import java.util.Objects;
/** /**
* @author duanruiming * @author duanruiming
...@@ -24,6 +26,9 @@ import java.util.List; ...@@ -24,6 +26,9 @@ import java.util.List;
public class WebSocketClient implements CommandLineRunner { public class WebSocketClient implements CommandLineRunner {
public static Session session; public static Session session;
@Value("${server.port}")
private int port;
@Override @Override
public void run(String... args) throws Exception { public void run(String... args) throws Exception {
try { try {
...@@ -34,10 +39,12 @@ public class WebSocketClient implements CommandLineRunner { ...@@ -34,10 +39,12 @@ public class WebSocketClient implements CommandLineRunner {
List<CrossInfoPO> crossInfoList = CrossInfoCache.crossInfoList; List<CrossInfoPO> crossInfoList = CrossInfoCache.crossInfoList;
if (!CollectionUtils.isEmpty(crossInfoList)) { if (!CollectionUtils.isEmpty(crossInfoList)) {
for (CrossInfoPO crossInfoPO : crossInfoList) { for (CrossInfoPO crossInfoPO : crossInfoList) {
String crossId = crossInfoPO.getId(); if (!Objects.equals(23, crossInfoPO.getManufacturerId())) {
String uri = "ws://localhost:32000/utc/signalStatus/" + crossId + ",preload"; String crossId = crossInfoPO.getId();
session = container.connectToServer(WebSocketClientHandler.class, URI.create(uri)); String uri = "ws://localhost:" + port + "/utc/signalStatus/" + crossId + ",preload";
log.info("ws客户端建立成功:[{}]", crossId); session = container.connectToServer(WebSocketClientHandler.class, URI.create(uri));
log.info("ws客户端建立成功:[{}]", crossId);
}
} }
} }
} catch (Exception e) { } catch (Exception e) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment