服务端:
1、pom文件中引入webScoket组件
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-thymeleaf</artifactId>
</dependency>
2、建立webSocket配置类
首先要注入ServerEndpointExporter,这个bean会自动注册使用了@ServerEndpoint注解声明的Websocket endpoint。要注意,如果使用独立的servlet容器,而不是直接使用springboot的内置容器,就不要注入ServerEndpointExporter,因为它将由容器自己提供和管理。
package com.javamalls.config;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
@Component
public class WebSocketConfig {
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
}
3、建立webSocket服务端
package com.javamalls.socket;
import org.springframework.stereotype.Component;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.util.concurrent.CopyOnWriteArrayList;
@Component
@ServerEndpoint("/webScoket")
public class WebScoketUtil {
private Session session;
private static CopyOnWriteArrayList<WebScoketUtil> webSocketSet=new CopyOnWriteArrayList<>();
@OnOpen
public void onOpen(Session session){
this.session=session;
webSocketSet.add(this);
System.out.println("【websocket消息】有新的连接, 总数:"+webSocketSet.size());
}
@OnClose
public void onClose() {
webSocketSet.remove(this);
System.out.println("【websocket消息】连接断开, 总数:"+webSocketSet.size());
}
@OnMessage
public void onMessage(String message) {
System.out.println("【websocket消息】收到客户端发来的消息:"+ message);
}
public void sendMessage(String message) {
for (WebScoketUtil webSocket: webSocketSet) {
System.out.println("【websocket消息】广播消息, message="+message);
try {
webSocket.session.getBasicRemote().sendText(message);
} catch (Exception e) {
e.printStackTrace();
}
}
}
}
4、调用
package com.javamalls.controller;
import com.javamalls.socket.WebScoketUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class SendMessage {
@Autowired
private WebScoketUtil webScoketUtil;
@RequestMapping("/sendMsg")
public String send(){
webScoketUtil.sendMessage("你好,这是消息测试");
return "发送socket消息成功";
}
}
客户端调用:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>首页</title>
</head>
<script>
var websocket = null;
if('WebSocket' in window) {
websocket = new WebSocket('ws://localhost:8087/webScoket');
}else {
alert('该浏览器不支持websocket!');
}
websocket.onopen = function (event) {
console.log('建立连接');
}
websocket.onclose = function (event) {
console.log('连接关闭');
}
websocket.onmessage = function (event) {
console.log('收到消息:' + event.data)
alert(event.data)
}
websocket.onerror = function () {
alert('websocket通信发生错误!');
}
window.onbeforeunload = function () {
websocket.close();
}
</script>
<body>
这是首页
</body>
</html>