Что такое веб-сокет
WebSocket — это протокол, предоставляемый HTML5 для полнодуплексной связи по одному TCP-соединению.
WebSocket упрощает обмен данными между клиентом и сервером, позволяя серверу активно передавать данные клиенту. В API WebSocket браузеру и серверу нужно выполнить только одно рукопожатие, и между ними может быть создано постоянное соединение, и может выполняться двусторонняя передача данных.
В WebSocket API браузеру и серверу достаточно сделать рукопожатие, после чего между браузером и сервером формируется быстрый канал. Данные могут передаваться напрямую между ними.
Как написать джава
Настроить бины
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
Java-ссылка
/**
* token为前端连接时的标识，后端根据此token维护用户与webSocket的绑定
*/
@Component
@ServerEndpoint("/webSocket/{token}")
public class WebSocket {
private Session session;
private static ConcurrentHashMap<String, WebSocket> webSocketMap = new ConcurrentHashMap<>();
@OnOpen
public void onOpen(@PathParam(value = "token") String token, Session session) {
this.session = session;
webSocketMap.put(token, this);
System.out.println("new connect：" + token + ". total :" + webSocketMap.size());
}
@OnClose
public void onClose(@PathParam(value = "token") String token) {
webSocketMap.remove(token);
System.out.println("disconnect:" + token + ", total:" + webSocketMap.size());
}
@OnMessage
public void onMessage(String message) {
System.out.println("get message: " + message);
}
public void sendMessage(String message) {
// 根据业务逻辑，找到token指向的webSocket
WebSocket webSocket = webSocketMap.get(String.valueOf(message.charAt(0)));
// 找不到返回
if (webSocket == null) {
return;
}
try {
webSocket.session.getBasicRemote().sendText(message);
} catch (IOException e) {
e.printStackTrace();
}
}
}
Как написать фронтенд
/**
* WebSocket客户端
*
* 使用说明：
* 1、WebSocket客户端通过回调函数来接收服务端消息。例如：webSocket.onmessage
* 2、WebSocket客户端通过send方法来发送消息给服务端。例如：webSocket.send();
*/
function getWebSocket() {
/**
* WebSocket客户端 PS：URL开头表示WebSocket协议 中间是域名端口 结尾是服务端映射地址
*/
var webSocket = new WebSocket(/*[[${webSocketUrl}]]*/ 'ws://localhost:8080/webSocket/2');
/**
* 当服务端打开连接
*/
webSocket.onopen = function (event) {
console.log('WebSocket打开连接');
};
/**
* 当服务端发来消息：1.广播消息 2.更新在线人数
*/
webSocket.onmessage = function (event) {
console.log('WebSocket收到消息：%c' + event.data, 'color:green');
};
/**
* 关闭连接
*/
webSocket.onclose = function (event) {
console.log('WebSocket关闭连接');
};
/**
* 通信失败
*/
webSocket.onerror = function (event) {
console.log('WebSocket发生异常');
};
return webSocket;
}