package com.whyc.ws;
|
|
import com.whyc.dto.Response;
|
import com.whyc.pojo.MotorState;
|
import com.whyc.pojo.TestPlan;
|
import com.whyc.service.MotorStateService;
|
import com.whyc.service.TestPlanService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import javax.websocket.*;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
import java.io.IOException;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.concurrent.CopyOnWriteArraySet;
|
|
@ServerEndpoint(value = "/testDataWebSocket",encoders = WebSocketEncoder.class)
|
@Component
|
@Slf4j
|
public class TestDataWebSocket {
|
private static MotorStateService motorStateService;
|
private static TestPlanService testPlanService;
|
|
@Autowired
|
public void setMotorStateService(MotorStateService motorStateService){
|
TestDataWebSocket.motorStateService = motorStateService;
|
}
|
@Autowired
|
public void setTestPlanService(TestPlanService testPlanService){
|
TestDataWebSocket.testPlanService = testPlanService;
|
}
|
|
|
/**
|
* 与某个客户端的连接对话，需要通过它来给客户端发送消息
|
*/
|
private Session session;
|
|
/**
|
* 标识当前连接客户端的参数 试验计划id
|
*/
|
//rivate Integer testPlanId;
|
|
/**
|
* 用于存所有的连接服务的客户端，这个对象存储是安全的
|
*/
|
//private static ConcurrentHashMap<String,WebSocket> webSocketSet = new ConcurrentHashMap<>();
|
|
private static CopyOnWriteArraySet<TestDataWebSocket> webSockets = new CopyOnWriteArraySet<>();
|
@OnOpen
|
public void onOpen(Session session){
|
System.out.println("试验计划实时数据websocket...");
|
this.session = session;
|
//this.testPlanId = testPlanId;
|
webSockets.add(this);
|
}
|
|
|
@OnMessage
|
public void onMessage(String message,Session session) {
|
try {
|
this.sendMessage(message);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
|
public void sendMessage(String testPlanId) throws IOException{
|
Thread thread = new Thread("TestDataWebSocket"){
|
@Override
|
public void run() {
|
while (true) {
|
try {
|
Response model = new Response();
|
//获取试验计划数据
|
TestPlan testPlan = testPlanService.getOneById(Integer.parseInt(testPlanId));
|
//获取试验参试设备
|
if(testPlan==null){
|
break;
|
}
|
String devices = testPlan.getDevices();
|
String[] devIds = devices.split(",");
|
List<MotorState> list = new ArrayList<>();
|
for (String deviceId:devIds) {
|
MotorState motorState = motorStateService.getByDeviceId(Integer.parseInt(deviceId));
|
list.add(motorState);
|
}
|
model.set(1,list);
|
if(session.isOpen()) {
|
session.getBasicRemote().sendObject(model);
|
}else{
|
break;
|
}
|
sleep(4000);
|
} catch (IOException | InterruptedException | EncodeException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
};
|
thread.start();
|
}
|
|
|
/**
|
* 连接关闭调用的方法
|
*/
|
@OnClose
|
public void onClose() {
|
System.out.println("closing");
|
webSockets.remove(this);
|
}
|
|
/**
|
* 发生错误时调用
|
* @param session
|
* @param error
|
*/
|
@OnError
|
public void onError(Session session, Throwable error){
|
System.out.println("发生错误");
|
error.printStackTrace();
|
}
|
|
}
|
