Springboot基于websocket實現簡單在線聊天功能
添加maven依賴
<?xml version='1.0' encoding='UTF-8'?><project xmlns='http://maven.apache.org/POM/4.0.0' xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' xsi:schemaLocation='http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd'> <parent> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-parent</artifactId> <version>2.2.5.RELEASE</version> </parent> <modelVersion>4.0.0</modelVersion> <artifactId>springboot-websocket</artifactId> <name>springboot-websocket</name> <dependencies> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> </dependencies></project>
添加websocket配置
@Configuration@EnableWebSocketpublic class MyWebSocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(myHandler(), '/myHandler').setAllowedOrigins('*'); } @Bean public WebSocketHandler myHandler() { return new MyTextWebSocketHandler(); }}
實現具體的handler
public class MyTextWebSocketHandler extends TextWebSocketHandler { private Set<WebSocketSession> sessions = new HashSet<>(); @Override protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { if (session.isOpen()) { sessions.add(session); } sendToAll(message); } private void sendToAll(TextMessage message) throws IOException { for (WebSocketSession session : sessions) { session.sendMessage(message); } } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { sessions.remove(session); }}
即可通過ws://localhost:8080/myHandler訪問websocket
添加測試頁面：
<html><script type='text/javascript'> if ('WebSocket' in window) { var ws = new WebSocket('ws://localhost:8080/myHandler'); ws.onopen = function () { }; ws.onmessage = function (evt) { document.getElementById(’messageDiv’).innerHTML += evt.data + '</br>'; }; ws.onclose = function () { console.log('close connect'); }; } else { alert('您的瀏覽器不支持 WebSocket!'); } function send() { ws.send(document.getElementById('input').value + ': ' + document.getElementById('message').value); }</script></head><body>當前用戶: <input /><br/><a href='http://www.aoyou183.cn/bcjs/13700.html#' rel='external nofollow' onclick='send();'>發送消息</a>: <input /><div id='messageDiv'></div></body></html>
即可實現簡單的通信功能
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持好吧啦網。
相關文章：
網公網安備