package com.shlanbao.tzsc.pms.websocket;
|
|
import org.apache.commons.collections.MapUtils;
|
import org.springframework.web.socket.CloseStatus;
|
import org.springframework.web.socket.TextMessage;
|
import org.springframework.web.socket.WebSocketSession;
|
import org.springframework.web.socket.handler.TextWebSocketHandler;
|
|
/**
|
|
*/
|
public class MyWebSocketHandler extends TextWebSocketHandler {
|
@Override
|
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
|
super.afterConnectionEstablished(session);
|
String userId = MapUtils.getString(session.getAttributes(),"name");
|
System.out.println("websocket连接：用户-" +userId);
|
//记录连接的session
|
if(userId!=null)
|
WebSocketSessionUtils.getInstance().add(userId,session);
|
|
}
|
@Override
|
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
|
super.handleTextMessage(session, message);
|
String userId = MapUtils.getString(session.getAttributes(),"name");
|
//回复消息
|
// TextMessage returnMessage = new TextMessage("Hi:"+userId+"..your message:"+message.getPayload()+" I have received,now I tell you");
|
// WebSocketSessionUtils.getInstance().sendMessageToTarget(userId,returnMessage);
|
|
}
|
|
@Override
|
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
|
super.handleTransportError(session, exception);
|
String userId = MapUtils.getString(session.getAttributes(),"name");
|
System.out.println("websocket错误：用户-" +userId+"/错误-"+exception.toString());
|
WebSocketSessionUtils.getInstance().remove(userId);
|
}
|
|
@Override
|
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
|
super.afterConnectionClosed(session, status);
|
|
String userId = MapUtils.getString(session.getAttributes(),"name");
|
System.out.println("websocket断开：用户-" +userId);
|
WebSocketSessionUtils.getInstance().remove(userId);
|
}
|
}
|
