lxw
2023-04-24 bb6f09aeedfd58b871ad0afa08392056da39e77f
src/main/java/com/whyc/webSocket/InterfaceAlarmSocket.java
@@ -1,12 +1,12 @@
package com.whyc.webSocket;
import com.whyc.dto.InterfaceParam;
import com.whyc.config.WebSocketConfig;
import com.whyc.dto.Response;
import com.whyc.service.InterfaceService;
import com.whyc.util.ActionUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.servlet.http.HttpSession;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
@@ -15,7 +15,7 @@
import java.util.Map;
@Component
@ServerEndpoint(value = "/interfaceAlarm/{interfaceNumber}",encoders = WebSocketEncoder.class)
@ServerEndpoint(value = "/interfaceAlarm/{interfaceNumber}", encoders = WebSocketEncoder.class, configurator = WebSocketConfig.class)
public class InterfaceAlarmSocket {
    private volatile Thread thread;
@@ -30,20 +30,25 @@
    private static InterfaceService service;
    private Session session;
    private HttpSession httpSession;
    @Autowired
    public void setService(InterfaceService service){
        InterfaceAlarmSocket.service = service;
    }
    @OnOpen
    public void onOpen(Session session, @PathParam("interfaceNumber") String interfaceNumber){
    public void onOpen(Session session, @PathParam("interfaceNumber") String interfaceNumber, EndpointConfig config) {
        this.session=session;
        this.httpSession = (HttpSession) config.getUserProperties().get("httpSession");
        String lang = (String) this.httpSession.getAttribute("lang");
        Thread thread = new Thread() {
            @Override
            public void run() {
                try {
                    while (!currentThread().isInterrupted()) {
                        Response res = service.getAlarmDiffMap(interfaceNumber);
                        Response res = service.getAlarmDiffMap(interfaceNumber, lang);
                        session.getBasicRemote().sendObject(res);
                        sleep(4000);
                    }