介绍
WebSocket是一种在单个TCP连接上进行全双工通信的协议。它使得客户端和服务器之间的实时通信变得更加容易。SpringBoot提供了对WebSocket的支持，使得我们可以轻松地实现实时通信功能。
在本文中，我们将使用SpringBoot和WebSocket来实现一个简单的实时聊天室。
准备工作
在开始之前，我们需要确保我们已经安装了以下软件：
- JDK 1.8或更高版本
- Maven
- IDE（例如Eclipse或IntelliJ IDEA）
创建SpringBoot项目
首先，我们需要创建一个SpringBoot项目。我们可以使用Spring Initializr来创建一个基本的SpringBoot项目。在创建项目时，我们需要选择以下依赖项：
- Spring Web
- Spring WebSocket
创建WebSocket配置类
接下来，我们需要创建一个WebSocket配置类。这个类将会配置WebSocket的相关参数。
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(chatWebSocketHandler(), "/chat").setAllowedOrigins("*");
}
@Bean
public WebSocketHandler chatWebSocketHandler() {
return new ChatWebSocketHandler();
}
}
在这个配置类中，我们使用了@EnableWebSocket注解来启用WebSocket支持。然后，我们实现了WebSocketConfigurer接口，并重写了registerWebSocketHandlers方法。在这个方法中，我们注册了一个ChatWebSocketHandler实例，并指定了它的URL为/chat。最后，我们使用setAllowedOrigins("*")方法来允许所有来源的WebSocket连接。
创建WebSocket处理器
接下来，我们需要创建一个WebSocket处理器。这个处理器将会处理WebSocket连接和消息。
public class ChatWebSocketHandler extends TextWebSocketHandler {
private static final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
sessions.add(session);
}
@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
for (WebSocketSession s : sessions) {
s.sendMessage(message);
}
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
sessions.remove(session);
}
}
在这个处理器中，我们继承了TextWebSocketHandler类，并重写了afterConnectionEstablished、handleTextMessage和afterConnectionClosed方法。
在afterConnectionEstablished方法中，我们将新的WebSocket会话添加到一个静态列表中。
在handleTextMessage方法中，我们遍历所有WebSocket会话，并将消息发送给每个会话。
在afterConnectionClosed方法中，我们将关闭的WebSocket会话从列表中移除。
创建聊天室页面
接下来，我们需要创建一个聊天室页面。这个页面将会使用JavaScript来连接WebSocket，并发送和接收消息。
<!DOCTYPE html>
<html>
<head>
<title>Chat Room</title>
<script src="https://code.jquery.com/jquery-3.5.1.min.js"></script>
<script>
$(function () {
var socket = new WebSocket("ws://" + window.location.host + "/chat");
socket.onmessage = function (event) {
var message = event.data;
$("#messages").append("<p>" + message + "</p>");
};
$("#send").click(function () {
var message = $("#message").val();
socket.send(message);
$("#message").val("");
});
});
</script>
</head>
<body>
<div id="messages"></div>
<input type="text" id="message">
<button id="send">Send</button>
</body>
</html>
在这个页面中，我们使用了jQuery来连接WebSocket。在页面加载时，我们创建了一个WebSocket实例，并指定了它的URL为/chat。然后，我们使用onmessage方法来接收WebSocket消息，并将它们添加到页面上的消息列表中。最后，我们使用click方法来发送消息。
运行应用程序
现在，我们已经完成了应用程序的所有部分。我们可以使用Maven来构建和运行应用程序。在命令行中，我们可以使用以下命令来构建和运行应用程序：
mvn spring-boot:run
然后，我们可以在浏览器中访问http://localhost:8080来打开聊天室页面。在页面上输入消息并点击“发送”按钮，就可以发送消息并在页面上看到它们了。
总结
在本文中，我们使用了SpringBoot和WebSocket来实现一个简单的实时聊天室。通过使用WebSocket，我们可以轻松地实现实时通信功能，从而提升用户体验。
