package com.whyc.webSocket; import com.whyc.config.WebSocketConfig; import com.whyc.dto.Response; import com.whyc.pojo.db_user.UserInf; import com.whyc.service.DevLithiumInfService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; import javax.servlet.http.HttpSession; import javax.websocket.*; import javax.websocket.server.ServerEndpoint; import java.io.IOException; /** * 左侧列表 */ @Component @ServerEndpoint(value = "/devLithiumLine",encoders = WebSocketEncoder.class,configurator = WebSocketConfig.class) public class DevLithiumLineSocket { private Session session; private Thread thread; private static DevLithiumInfService service; private HttpSession httpSession; @Autowired public void setService(DevLithiumInfService service) { DevLithiumLineSocket.service = service; } @OnOpen public void onOpen(Session session, EndpointConfig config){ this.httpSession = (HttpSession) config.getUserProperties().get("httpSession"); UserInf user = (UserInf) this.httpSession.getAttribute("user"); final int userId = user.getUid(); //final int userId = 101; Thread thread = new Thread() { @Override public void run() { try { while (!currentThread().isInterrupted()) { Response res = service.getDevType(userId); session.getBasicRemote().sendObject(res); sleep(4000); } } catch (Exception e) { this.interrupt(); } } }; thread.start(); this.thread = thread; } @OnClose public void onClose(CloseReason closeReason) throws IOException { System.err.println("closeReason = " + closeReason); if(session.isOpen()){ session.close(); } } @OnError public void onError(Throwable error) throws IOException { error.printStackTrace(); thread.isInterrupted(); if(session.isOpen()){ session.close(); } } }