前端js代码:
<script>
var websocket = null;
if('WebSocket' in window) {
websocket = new WebSocket('ws://localhost:8080/sell/webSocket');
}else {
alert('该浏览器不支持websocket!');
}
websocket.onopen = function (event) {
console.log('建立连接');
}
websocket.onclose = function (event) {
console.log('连接关闭');
}
websocket.onmessage = function (event) {
console.log('收到消息:' + event.data)
//弹窗提醒, 播放音乐
$('#myModal').modal('show');
document.getElementById('notice').play();
}
websocket.onerror = function () {
alert('websocket通信发生错误!');
}
window.onbeforeunload = function () {
websocket.close();
}
</script>
第二步导入websocket依赖:
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
第三步写WebSocketConfig:
@Component
public class WebSocketConfig {
@Bean
public ServerEndpointExporter serverEndpointExporter(){
return new ServerEndpointExporter();
}
}
第四步编写webSocket:
@Component
@ServerEndpoint("/webSocket")
@Slf4j
public class WebSocket {
private Session session;
private static CopyOnWriteArrayList<WebSocket> webSockets = new CopyOnWriteArrayList<>();
@OnOpen
public void opOpen(Session session){
this.session = session;
webSockets.add(this);
log.info("【websocket消息】有新的连接,总数:{}",webSockets.size());
}
@OnClose
public void onClose(){
webSockets.remove(this);
log.info("【websocket消息】连接断开,总数:{}",webSockets.size());
}
@OnMessage
public void onMessage(String message){
log.info("【websocket消息】收到客户端发来的消息:{}",message);
}
public void sendMessage(String message){
for(WebSocket webSocket:webSockets){
log.info("【websocket消息】广播消息,message={}",message);
try {
webSocket.session.getBasicRemote().sendText(message);
}catch (Exception e){
e.printStackTrace();
}
}
}
}
最后引入依赖,然后使用websocket发送消息给客户端:
@Autowired
private WebSocket webSocket;
//发送websocket消息
webSocket.sendMessage(orderDTO.getOrderId());