web socket实例

后台代码：

/**

* web socket拦截器

*/

@Component

public class WebSocketInterceptor extends HttpSessionHandshakeInterceptor {

@Override

public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {

if (request instanceof ServletServerHttpRequest){

ServletServerHttpRequest serverHttpRequest = (ServletServerHttpRequest) request;

HttpSession session = serverHttpRequest.getServletRequest().getSession();

SessionInfo sessionInfo=(SessionInfo)session.getAttribute("sessionInfo");

if (session != null) {

attributes.put("LOGIN_USER_ID", sessionInfo.getLOGIN_USER_ID());

}

}

return true;

}

@Override

public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception ex) {

super.afterHandshake(request, response, wsHandler, ex);

}

}

/**

* web socket配置类

*/

@Configuration

@EnableWebSocket

public class WebSocketConfig implements WebSocketConfigurer {

@Override

public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {

registry.addHandler(myHandler(),"/myHandler").addInterceptors(new WebSocketInterceptor());

}

@Bean

public WebSocketHandler myHandler() {

return new MyHandler();

}

}

/**

* 处理器类

*/

@Service

public class MyHandler extends TextWebSocketHandler {

//在线用户列表

private static final Map<String, WebSocketSession> users;

//用户标识

private static final String CLIENT_ID = "LOGIN_USER_ID";

static {

users = new HashMap<String, WebSocketSession>();

}

@Override

public void afterConnectionEstablished(WebSocketSession session) throws Exception {

System.out.println("成功建立连接");

String userId = getClientId(session);

if (StringUtils.isNotBlank(userId)) {

users.put(userId, session);

session.sendMessage(new TextMessage("成功建立socket连接"));

System.out.println(userId);

}

}

@Override

public void handleTextMessage(WebSocketSession session, TextMessage message) {

System.out.println(message.getPayload());

WebSocketMessage message1 = new TextMessage("server:"+message);

try {

session.sendMessage(message1);

} catch (IOException e) {

e.printStackTrace();

}

}

/**

* 发送信息给指定用户

*

* @param clientId

* @param message

* @return

*/

public boolean sendMessageToUser(String clientId, TextMessage message) {

if (users.get(clientId) == null) {

return false;

}

WebSocketSession session = users.get(clientId);

System.out.println("sendMessage:" + session);

if (!session.isOpen()) {

return false;

}

try {

session.sendMessage(message);

} catch (IOException e) {

return false;

}

return true;

}

/**

* 广播信息

*

* @param message

* @return

*/

public boolean sendMessageToAllUsers(TextMessage message) {

boolean allSendSuccess = true;

Set<String> clientIds = users.keySet();

WebSocketSession session = null;

for (String clientId : clientIds) {

try {

session = users.get(clientId);

if (session.isOpen()) {

session.sendMessage(message);

}

} catch (IOException e) {

e.printStackTrace();

allSendSuccess = false;

}

}

return allSendSuccess;

}

@Override

public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {

if (session.isOpen()) {

session.close();

}

System.out.println("连接出错");

users.remove(getClientId(session));

}

@Override

public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {

System.out.println("连接已关闭：" + status);

users.remove(getClientId(session));

}

@Override

public boolean supportsPartialMessages() {

return false;

}

/**

* 获取用户标识

*

* @param session

* @return

*/

private String getClientId(WebSocketSession session) {

try {

String clientId = (String) session.getHandshakeAttributes().get(CLIENT_ID);

return clientId;

} catch (Exception e) {

return null;

}

}

}

前台代码：

var ws =

$(function(){

ws = new WebSocket("ws://"+location.host+$("#context_global").val()+"/myHandler");

ws.onopen = function () {

console.log("onopen");

}

ws.onclose = function () {

console.log("onclose");

}

ws.onmessage = function (msg) {

if(msg.data){

console.log(msg.data);

}

}

});