代码优化
parent
3aa4196399
commit
243aeb989f
@ -0,0 +1,12 @@
|
|||||||
|
package com.ipsplm.entity.simulation.vo;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.io.Serializable;
|
||||||
|
@Data
|
||||||
|
public class FieldCnVO implements Serializable {
|
||||||
|
|
||||||
|
private String fieldCode;
|
||||||
|
|
||||||
|
private String fieldName;
|
||||||
|
}
|
||||||
@ -0,0 +1,12 @@
|
|||||||
|
package com.ipsplm.service.simulation;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @Description 仿真Service
|
||||||
|
* @Author FanDongqiang
|
||||||
|
* @Date 2023/3/15 16:23
|
||||||
|
* @Version 1.0
|
||||||
|
*/
|
||||||
|
public interface ISimulationService {
|
||||||
|
|
||||||
|
String getPlantData(Object param);
|
||||||
|
}
|
||||||
@ -0,0 +1,104 @@
|
|||||||
|
package com.ipsplm.ws;
|
||||||
|
|
||||||
|
import com.ipsplm.service.simulation.ISimulationService;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
|
||||||
|
import javax.websocket.OnClose;
|
||||||
|
import javax.websocket.OnMessage;
|
||||||
|
import javax.websocket.OnOpen;
|
||||||
|
import javax.websocket.Session;
|
||||||
|
import javax.websocket.server.ServerEndpoint;
|
||||||
|
import java.util.Timer;
|
||||||
|
import java.util.TimerTask;
|
||||||
|
import java.util.concurrent.CopyOnWriteArraySet;
|
||||||
|
import java.util.concurrent.Executors;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
|
@Slf4j
|
||||||
|
@Component
|
||||||
|
@ServerEndpoint(value = "/websocket/simulation")
|
||||||
|
public class SimulationWebsocket {
|
||||||
|
private Session session;
|
||||||
|
|
||||||
|
private static ISimulationService simulationService;
|
||||||
|
|
||||||
|
private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
|
||||||
|
private boolean isSimulationRunning = false;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
public void setService(ISimulationService simulationService) {
|
||||||
|
SimulationWebsocket.simulationService = simulationService;
|
||||||
|
}
|
||||||
|
|
||||||
|
//存放websocket的集合
|
||||||
|
private static CopyOnWriteArraySet<SimulationWebsocket> webSocketSet = new CopyOnWriteArraySet<>();
|
||||||
|
|
||||||
|
|
||||||
|
@OnOpen
|
||||||
|
public void OnOpen(Session session) {
|
||||||
|
this.session = session;
|
||||||
|
webSocketSet.add(this);
|
||||||
|
log.info("【websocket消息】有新的连接, 总数:{}", webSocketSet.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
@OnMessage
|
||||||
|
public void onMessage(String message) {
|
||||||
|
log.info("【websocket消息】收到客户端发来的消息:{}", message);
|
||||||
|
if ("StartSimulation".equals(message)) {
|
||||||
|
//发送开始仿真的信息
|
||||||
|
String result = simulationService.getPlantData(message);
|
||||||
|
if ("Simulation Started".equals(result)) {
|
||||||
|
isSimulationRunning = true;
|
||||||
|
log.info("【仿真开始】");
|
||||||
|
scheduler.scheduleAtFixedRate(() -> {
|
||||||
|
if (isSimulationRunning) {
|
||||||
|
String data = simulationService.getPlantData("SimulationProgress");
|
||||||
|
// 根据需要处理和发送获取到的数据
|
||||||
|
log.info("【仿真数据】：" + data);
|
||||||
|
sendMessage(data);
|
||||||
|
if("1".equals(data) || "1.00".equals(data)){
|
||||||
|
sendMessage("ending");
|
||||||
|
isSimulationRunning = false;
|
||||||
|
scheduler.shutdown();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}, 0, 10, TimeUnit.SECONDS);
|
||||||
|
} else {
|
||||||
|
sendMessage("Simulation Start Failed");
|
||||||
|
}
|
||||||
|
} else if ("ResetSimulation".equals(message)) {
|
||||||
|
String resetSimulation = simulationService.getPlantData(message);
|
||||||
|
if ("Simulation Reset".equals(resetSimulation)) {
|
||||||
|
sendMessage("ending");
|
||||||
|
} else {
|
||||||
|
sendMessage("Simulation Reset Failed");
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
String resetSimulation = simulationService.getPlantData(message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@OnClose
|
||||||
|
public void onClose() {
|
||||||
|
webSocketSet.remove(this);
|
||||||
|
log.info("【websocket消息】连接断开, 总数:{}", webSocketSet.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public void sendMessage(String message) {
|
||||||
|
for (SimulationWebsocket webSocket : webSocketSet) {
|
||||||
|
try {
|
||||||
|
synchronized (webSocket.session) {
|
||||||
|
if (webSocket.session.isOpen()) {
|
||||||
|
webSocket.session.getBasicRemote().sendText(message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue