package com.whyc.ws;
|
|
import com.whyc.dto.Response;
|
import com.whyc.service.*;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
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.concurrent.CopyOnWriteArraySet;
|
|
/**
|
* webSocket实现与客户端消息交互
|
*
|
* 页面端,三个电源数据的实时监控,可以在打开第一个电源实时页面的时候,触发ws连接开启,点中某个电源时,页面ws发送message:传入参数:实时类型和电源id.
|
* 一个WsSession一次开启ws,ws连接其实与http极为相似
|
*/
|
@ServerEndpoint(value = "/websocket",encoders = WebSocketEncoder.class)
|
@Component
|
public class DeviceWebSocket {
|
|
private static DynamicLoadStateService dynamicLoadStateService;
|
private static MotorStateService motorStateService;
|
private static ElectricStateService electricStateService;
|
private static Electric2MWStateService electric2MWStateService;
|
|
@Autowired
|
public void setDynamicLoadStateService(DynamicLoadStateService dynamicLoadStateService) {
|
DeviceWebSocket.dynamicLoadStateService = dynamicLoadStateService;
|
}
|
|
@Autowired
|
public void setMotorStateService(MotorStateService motorStateService) {
|
DeviceWebSocket.motorStateService = motorStateService;
|
}
|
|
@Autowired
|
public void setElectricStateService(ElectricStateService electricStateService) {
|
DeviceWebSocket.electricStateService = electricStateService;
|
}
|
|
@Autowired
|
public void setElectric2MWStateService(Electric2MWStateService electric2MWStateService) {
|
DeviceWebSocket.electric2MWStateService = electric2MWStateService;
|
}
|
|
private Integer userId;
|
|
//concurrent包的线程安全，用来存放每个客户端对应的MyWebSocket对象。
|
//若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
|
// private static Map<String, WebSocket> webSockets = new ConcurrentHashMap<>();
|
private static CopyOnWriteArraySet<DeviceWebSocket> webSockets = new CopyOnWriteArraySet<>();
|
|
//与某个客户端的连接会话，需要通过它来给客户端发送数据
|
private Session session;
|
|
/**
|
* 连接建立成功调用的方法
|
*
|
* @param session 可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
|
*/
|
@OnOpen
|
public void onOpen(Session session) {
|
System.out.println("websocket1...");
|
this.session = session;
|
webSockets.add(this);
|
}
|
|
/**
|
* 收到客户端消息后调用的方法
|
* 连续监控到3次消息,会启动3个线程,这样导致 ws到页面 是3个响应数据;所以需要关闭ws后再打开ws 或者...
|
* @param message 客户端发送过来的消息
|
* @param session 可选的参数
|
*/
|
@OnMessage
|
public void onMessage(String message,Session session) {
|
try {
|
this.sendMessage(message);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
/*try {
|
this.sendMessage(message);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}*/
|
/*System.out.println(session.getId());
|
|
final String powerDeviceId = message;
|
final Session sessionThread = session;
|
Thread thread = new Thread("Thread_PowerACDCData"){
|
@Override
|
public void run() {
|
while (true) {
|
try {
|
sleep(4000);
|
//获取powerACData记录
|
ServiceModel model =service.get(powerDeviceId);
|
if(sessionThread.isOpen()) {
|
sessionThread.getBasicRemote().sendObject(model);
|
}else{
|
break;
|
}
|
} catch (IOException | InterruptedException | EncodeException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
};
|
thread.start();*/
|
/*//群发消息
|
for(PowerACDCDataWebSocket item: webSockets){
|
try {
|
item.sendMessage(message);
|
} catch (IOException e) {
|
e.printStackTrace();
|
continue;
|
}
|
}*/
|
}
|
|
/**
|
* 该方法没有用注解，是根据自己需要添加的方法。在自己的业务中调用，发送消息给前端。
|
*/
|
public void sendMessage(String deviceId) throws IOException{
|
Thread thread = new Thread("Thread_DeviceManage"){
|
@Override
|
public void run() {
|
while (true) {
|
try {
|
Response model = null;
|
//获取记录
|
if(deviceId.startsWith("1")) {
|
model = dynamicLoadStateService.get(Integer.parseInt(deviceId));
|
}
|
else if(deviceId.startsWith("2")) {
|
model = motorStateService.getById(Integer.parseInt(deviceId));
|
}
|
else if(deviceId.startsWith("3")) {
|
model = electricStateService.get(Integer.parseInt(deviceId));
|
}
|
else if(deviceId.startsWith("4")) {
|
model = electric2MWStateService.get(Integer.parseInt(deviceId));
|
}
|
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();
|
}
|
|
}
|
