package com.whyc.ws;
|
|
import com.whyc.dto.Response;
|
import com.whyc.service.DeviceInfService;
|
import com.whyc.service.DeviceManageService;
|
import io.swagger.annotations.Api;
|
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/deviceManage/{userId}",encoders = WebSocketEncoder.class)
|
@Component
|
@Api(tags = "设备实时状态")
|
public class DeviceInfWebSocket {
|
|
private static DeviceInfService service;
|
|
@Autowired
|
public void setService(DeviceInfService service) {
|
DeviceInfWebSocket.service = service;
|
}
|
|
private Integer userId;
|
|
//concurrent包的线程安全，用来存放每个客户端对应的MyWebSocket对象。
|
//若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
|
// private static Map<String, WebSocket> webSockets = new ConcurrentHashMap<>();
|
private static CopyOnWriteArraySet<DeviceInfWebSocket> webSockets = new CopyOnWriteArraySet<>();
|
|
//与某个客户端的连接会话，需要通过它来给客户端发送数据
|
private Session session;
|
|
/**
|
* 连接建立成功调用的方法
|
*
|
* @param session 可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
|
*/
|
@OnOpen
|
public void onOpen(Session session, @PathParam("userId") Integer userId) {
|
System.out.println("websocket1...");
|
this.session = session;
|
this.userId = userId;
|
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 pageInf) throws IOException{
|
Thread thread = new Thread("Thread_DeviceManage"){
|
@Override
|
public void run() {
|
while (true) {
|
try {
|
//获取powerACData记录
|
Response model =service.getAll();
|
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();
|
}
|
|
}
|
