- 添加依赖
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency>
- 添加 MyWebSocket 类
package com.jfsoft.socket;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.Map;
/**
* @author ChenXc
* @version V1.0
* @Date 2018/10/8 15:07
* @Description TODO(一句话描述类作用)
*/
@ServerEndpoint(value = "/websocket") //接受websocket请求路径
@Component //注册到spring容器中
public class MyWebSocket {
//保存所有在线socket连接
private static Map, MyWebSocket> webSocketMap = new LinkedHashMap<>();
//记录当前在线数目
private static int count = 0;
//当前连接（每个websocket连入都会创建一个MyWebSocket实例
private Session session;
private Logger log = LoggerFactory.getLogger(this.getClass());
//处理连接建立
@OnOpen
public void onOpen(Session session) {
this.session = session;
webSocketMap.put(session.getId(), this);
addCount();
log.info("新的连接加入：{}", session.getId());
}
//接受消息
@OnMessage
public void onMessage(String message, Session session) {
log.info("收到客户端{}消息：{}", session.getId(), message);
try {
this.sendMessage("收到消息：" + message);
} catch (Exception e) {
e.printStackTrace();
}
}
//处理错误
@OnError
public void onError(Throwable error, Session session) {
log.info("发生错误{},{}", session.getId(), error.getMessage());
}
//处理连接关闭
@OnClose
public void onClose() {
webSocketMap.remove(this.session.getId());
reduceCount();
log.info("连接关闭:{}", this.session.getId());
}
//群发消息
//发送消息
public void sendMessage(String message) throws IOException {
this.session.getBasicRemote().sendText(message);
}
//广播消息
public static void broadcast(String msg) {
MyWebSocket.webSocketMap.forEach((k, v) -> {
try {
v.sendMessage(msg);
} catch (Exception e) {
}
});
}
//获取在线连接数目
public static int getCount() {
return count;
}
//操作count，使用synchronized确保线程安全
public static synchronized void addCount() {
MyWebSocket.count++;
}
public static synchronized void reduceCount() {
MyWebSocket.count--;
}
}
- 添加 WebSocketConfig 类
package com.jfsoft.socket;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
/**
* @author ChenXc
* @version V1.0
* @Date 2018/10/8 15:06
* @Description TODO(一句话描述类作用)
*/
@Configuration
public class WebSocketConfig {
@Bean
public ServerEndpointExporter serverEndpointExporter(){
return new ServerEndpointExporter();
}
}
- 添加测试页面 test.html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Title</title> </head> <body> <h1>websocket测试</h1> </body> <script> let socket = new WebSocket("ws://localhost:8080/websocket"); socket.onerror = err => { console.log(err); }; socket.onopen = event => { console.log(event); }; socket.onmessage = msg => { console.log(msg) }; socket.onclose = () => { console.log("连接关闭"); }; function sendMessage() { if (socket.readyState === 1) socket.send("这是一个测试数据"); else alert("尚未建立websocket连接"); } </script> </html>
注意：springboot 下 websocket 运行没问题，但是打包会报错，解决如下
欢迎来到这里！
我们正在构建一个小众社区，大家在这里相互信任，以平等 • 自由 • 奔放的价值观进行分享交流。最终，希望大家能够找到与自己志同道合的伙伴，共同成长。
注册 关于