WebSocket 通信流程，注解和Spring实现WebSocket ，实战多人聊天室系统

一、前言

实现即时通信常见的有四种方式-分别是：轮询、长轮询(comet)、长连接(SSE)、WebSocket。

①短轮询

很多网站为了实现推送技术，所用的技术都是轮询。轮询是在特定的的时间间隔（如每1秒），由客户端浏览器对服务器发出HTTP请求，然后由服务器返回最新的数据给客户端的浏览器。

优点：后端编码比较简单

缺点：这种传统的模式带来很明显的缺点，由于HTTP请求是单向的,是只能由客户端发起请求,由服务端响应的【请求-响应模式】，即客户端的浏览器需要不断的向服务器发出请求，然而HTTP请求可能包含较长的头部，其中真正有效的数据可能只是很小的一部分，显然这样会浪费很多的带宽等资源。

短轮询

②长轮询

客户端向发起一个到服务端的请求，然后服务端一直保持连接打开，直到数据发送到客户端为止。

优点：避免了服务端在没有信息更新时的频繁请求，节省流量

缺点：服务器一直保持连接会消耗资源，需要同时维护多个线程，而服务器所能承载的 TCP 连接是有上限的，所以这种轮询很容易导致连接上限。

长轮询

③长连接

客户端和服务端建立连接后不进行断开，之后客户端再次访问这个服务端上的内容时，继续使用这一条连接通道（长轮询是一次发送完后就断开了，）

优点：消息即时到达，不发无用请求

缺点：与长轮询一样，服务器一直保持连接是会消耗资源的，如果有大量的长连接的话，对于服务器的消耗是巨大的，而且服务器承受能力是有上限的，不可能维持无限个长连接。

HTTP长连接

④WebSocket

客户端向服务器发送一个携带特殊信息的请求头（Upgrade:WebSocket ）建立连接，建立连接后双方即可实现自由的实时双向通信。

优点：

较少的控制开销。在连接创建后，服务器和客户端之间交换数据时，用于协议控制的数据包头部相对较小。
更强的实时性。由于协议是全双工的，所以服务器可以随时主动给客户端下发数据。相对于HTTP请求需要等待客户端发起请求服务端才能响应，延迟明显更少；即使是和Comet等类似的长轮询比较，其也能在短时间内更多次地传递数据。
保持连接状态。与HTTP不同的是，Websocket需要先创建连接，这就使得其成为一种有状态的协议，之后通信时可以省略部分状态信息。而HTTP请求可能需要在每个请求都携带状态信息（如身份认证等）。

缺点：相对来说，开发成本和难度更高

WebSocket

⑤对比总结

	轮询（Polling）	长轮询（Long-Polling）	Websocket	长连接（SSE）
通信协议	http	http	tcp	http
触发方式	client(客户端)	client(客户端)	client、server(客户端、服务端)	client、server(客户端、服务端)
优点	兼容性好容错性强，实现简单	比短轮询节约资源	全双工通讯协议，性能开销小、安全性高，可扩展性强	实现简便，开发成本低
缺点	安全性差，占较多的内存资源与请求数	安全性差，占较多的内存资源与请求数	传输数据需要进行二次解析，增加开发成本及难度	只适用高级浏览器
延迟	非实时，延迟取决于请求间隔	非实时，延迟取决于请求间隔	实时	非实时，默认3秒延迟，延迟可自定义

长连接和长轮询的区别：长轮询的含义就是客户端发起请求，如果服务端的数据没有发生变更，那么就hold住请求，直到服务端的数据发生了变更，或者达到了一定的时间就会返回。这样就减少了客户端和服务端不断频繁连接和传递数据的过程，并且不会消耗服务端太多资源。长连接指的是TCP链接长久保持复用。

二、WebSocket概述

WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。很多网站为了实现推送技术，所用的技术都是轮询。轮询是在特定的时间间隔（如每1秒），由浏览器对服务器发出，然后由服务器返回最新的数据给客户端的浏览器。这种传统的模式带来很明显的缺点，即浏览器需要不断的向服务器发出请求，然而HTTP请求可能包含较长的头部，其中真正有效的数据可能只是很小的一部分，显然这样会浪费很多的带宽等资源。

而比较新的技术去做轮询的效果是长轮询(comet)。这种技术虽然可以双向通信，但依然需要反复发出请求。而且在Comet中，普遍采用的长链接，也会消耗服务器资源。

在这种情况下，HTML5定义了WebSocket协议，能更好的节省服务器资源和带宽，并且能够更实时地进行通讯。

1.Java接入WebSocket的两种方式

①基于JAVA注解实现

服务端

package com.mc.wsdemo.java;import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;/*** 监听websocket地址/myWs*/
@ServerEndpoint("/myWs")
@Component
@Slf4j
public class WsServerEndpont {static Map<String,Session> sessionMap = new ConcurrentHashMap<>();//连接建立时执行的操作@OnOpenpublic void onOpen(Session session){sessionMap.put(session.getId(),session);log.info("websocket is open");}//收到了客户端消息执行的操作@OnMessagepublic String onMessage(String text){log.info("收到了一条消息："+text);return "已收到你的消息";}//连接关闭的时候执行的操作@OnClosepublic void onClose(Session session){sessionMap.remove(session.getId());log.info("websocket is close");}//每2s发送给客户端心跳消息，设置一个定时任务@Scheduled(fixedRate = 2000)public void sendMsg() throws IOException {for(String key:sessionMap.keySet()){sessionMap.get(key).getBasicRemote().sendText("心跳");}}
}

配置类

package com.mc.wsdemo.java;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;@Configuration 注解表明这是一个Spring的配置类，其中包含了一些用于生成bean的方法。
ServerEndpointExporter 是Spring对WebSocket的支持类，它负责扫描并自动注册所有通过
@ServerEndpoint注解标记的WebSocket端点类。@Bean 注解方法 serverEndpointExporter()告诉Spring容器去实例化一个ServerEndpointExporter 
bean，并将其添加到IoC容器中。这样，在Spring应用启动时，ServerEndpointExporter就会自动发现
并注册那些使用了@ServerEndpoint注解的WebSocket端点，无需手动配置或初始化。@Configuration
public class WebSocketConfig {@Beanpublic ServerEndpointExporter serverEndpointExporter(){return new ServerEndpointExporter();}
}

客户端

<!DOCTYPE html>
<html lang="en">
<head><meta charset="UTF-8"><title>ws client</title>
</head>
<body>
<p style="border:1px solid black;width: 600px;height: 500px" id="talkMsg"></p>
<input id="message" /><button id="sendBtn" onclick="sendMsg()">发送</button>
</body>
<script>let ws = new WebSocket("ws://localhost:8080/myWs1")// ws.onopen=function () {// }ws.onmessage=function (message) {document.getElementById("talkMsg").innerHTML = message.data}function sendMsg() {ws.send(document.getElementById("message").value)document.getElementById("message").value=""}
</script>
</html>

效果

②Spring框架实现WebSocket服务器端：多人聊天室

package com.mc.wsdemo.spring;import lombok.extern.slf4j.Slf4j;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;import java.util.Map;/*** 握手拦截器*HttpSessionHandshakelnterceptor (抽象类):握手拦截器，在握手前后添加操作*/
@Component
@Slf4j
public class MyWsInterceptor extends HttpSessionHandshakeInterceptor {@Overridepublic boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {//在这里实现自己逻辑，最后都要调用父类的方法log.info(request.getRemoteAddress().toString()+"开始握手");return super.beforeHandshake(request, response, wsHandler, attributes);}@Overridepublic void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception ex) {log.info(request.getRemoteAddress().toString()+"完成握手");super.afterHandshake(request, response, wsHandler, ex);}
}

/*** web socket 主处理程序* AbstractWebSocketHandler (抽象类) :WebSocket处理程序，监听连接前，连接中，连接后*/@Slf4j
@Component
public class MyWsHandler extends AbstractWebSocketHandler {private static Map<String,SessionBean> sessionBeanMap ;//这里定义为一个map，这个Map<String,Session>这样写不能封装自己的信息，所以我们封装了一个sessionBeanMapprivate static AtomicInteger clientIdMaker;//定义一个整型private static StringBuffer stringBuffer;//定义一个字符串缓冲区static {sessionBeanMap = new ConcurrentHashMap<>();//在静态代码快里初始化map.clientIdMaker = new AtomicInteger(0);//线程安全初始化整型stringBuffer = new StringBuffer();}//连接建立@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {在 MyWsHandler 中，自定义了在连接建立之后存储 SessionBean 并发送通知消息的操作，但同时通过调
用 super.afterConnectionEstablished(session) 确保没有遗漏掉 Spring WebSocket 框架内部预设的
连接管理逻辑。如果不调用 super 方法，可能会错过某些重要的框架内置功能，影响整个 WebSocket 服务
的正确运行。super.afterConnectionEstablished(session);SessionBean sessionBean = new SessionBean(session,clientIdMaker.getAndIncrement());//调用我们封装类的构造方法。sessionBeanMap.put(session.getId(),sessionBean);log.info(sessionBeanMap.get(session.getId()).getClientId()+"建立了连接");stringBuffer.append(sessionBeanMap.get(session.getId()).getClientId()+"进入了群聊<br/>");sendMessage(sessionBeanMap);}//收到消息@Overrideprotected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {super.handleTextMessage(session, message);log.info(sessionBeanMap.get(session.getId()).getClientId()+":"+message.getPayload());stringBuffer.append(sessionBeanMap.get(session.getId()).getClientId()+":"+message.getPayload()+"<br/>");sendMessage(sessionBeanMap);}//传输异常@Overridepublic void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {super.handleTransportError(session, exception);if(session.isOpen()){session.close();}sessionBeanMap.remove(session.getId());}//连接关闭@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {super.afterConnectionClosed(session, status);int clientId = sessionBeanMap.get(session.getId()).getClientId();sessionBeanMap.remove(session);log.info(clientId+"关闭了连接");stringBuffer.append(clientId+"退出了群聊<br/>");sendMessage(sessionBeanMap);}//    //每2s发送给客户端心跳消息
//    @Scheduled(fixedRate = 2000)
//    public void sendMsg() throws IOException {
//        for(String key:sessionBeanMap.keySet()){
//            sessionBeanMap.get(key).getWebSocketSession().sendMessage(new TextMessage("心跳"));
//        }
//    }public void sendMessage(Map<String,SessionBean> sessionBeanMap){for(String key:sessionBeanMap.keySet()){try {sessionBeanMap.get(key).getWebSocketSession().sendMessage(new TextMessage(stringBuffer.toString()));} catch (IOException e) {
//                e.printStackTrace();log.error(e.getMessage());}}}
}

package com.mc.wsdemo.spring;import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;import javax.annotation.Resource;/**
*WebSocketConfigurer (接口):配置程序，比如配置监听哪个端口
*上面的握手拦截器，处理程序的使用
*/@Configuration
@EnableWebSocket
public class MyWsConfig implements WebSocketConfigurer {@ResourceMyWsHandler myWsHandler;@ResourceMyWsInterceptor myWsInterceptor;@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {registry.addHandler(myWsHandler,"/myWs1").addInterceptors(myWsInterceptor).setAllowedOrigins("*");}
}

package com.mc.wsdemo.spring;import lombok.AllArgsConstructor;
import lombok.Data;
import org.springframework.web.socket.WebSocketSession;@AllArgsConstructor
@Data
public class SessionBean {这个是对原Session类的二次封装，加入了一些我们想要的信息，封装曾一个新的类，园session作为属性private WebSocketSession webSocketSession;private Integer clientId;
}