spring boot websocket

小灰灰 2022-05-21 09:42 149阅读 0赞

1.在pom文件中添加依赖：

<groupId>org.springframework.boot</groupId>

<artifactId>spring-boot-starter-websocket</artifactId>

</dependency>

2.添加websocket配置类：

import org.springframework.context.annotation.Bean;  
import org.springframework.context.annotation.Configuration;  
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration

public class WebSocketConfig \{

@Bean

public ServerEndpointExporter serverEndpointExporter() \{  
        return new ServerEndpointExporter();  
    \}

3.websocket服务端

import java.io.IOException;  
import java.util.concurrent.CopyOnWriteArraySet;  
import javax.websocket.OnClose;  
import javax.websocket.OnError;  
import javax.websocket.OnMessage;  
import javax.websocket.OnOpen;  
import javax.websocket.Session;  
import javax.websocket.server.ServerEndpoint;  
import org.springframework.stereotype.Component;  
  
@Component  
@ServerEndpoint(value = "/websocket")  
public class WebSocketServer \{  
private static int onlineCount = 0;  
private static CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();  
  
private Session session;  
  
@OnOpen  
public void onOpen(Session session) \{  
this.session = session;  
webSocketSet.add(this); // 加入set中  
addOnlineCount(); // 在线数加1  
System.out.println("新用戶：" + this);  
try \{  
System.out.println("连接成功！");  
sendMessage("连接成功！");  
\} catch (IOException e) \{  
System.out.println("连接失败！");  
\}  
\}  
  
@OnClose  
    public void onClose() \{  
System.out.println("连接中断");  
        webSocketSet.remove(this);  //从set中删除  
        subOnlineCount();           //在线数减1  
    \}  
  
@OnMessage  
    public void onMessage(String message, Session session) \{  
  
        //群发消息  
        for (WebSocketServer item : webSocketSet) \{  
            try \{  
            System.out.println(item.getOnlineCount());  
                item.sendMessage(message);  
            \} catch (IOException e) \{  
                e.printStackTrace();  
            \}  
        \}  
    \}  
  
@OnError  
    public void onError(Session session, Throwable error) \{  
        error.printStackTrace();  
          
    \}  
  
public void sendMessage(String message) throws IOException \{  
        this.session.getBasicRemote().sendText(message);  
    \}  
   
    /\*\*  
     \* 群发自定义消息  
     \* \*/  
    public static void sendInfo(String message) throws IOException \{  
        for (WebSocketServer item : webSocketSet) \{  
            try \{  
                item.sendMessage(message);  
            \} catch (IOException e) \{  
                continue;  
            \}  
        \}  
    \}  
   
    public static synchronized int getOnlineCount() \{  
        return onlineCount;  
    \}  
   
    public static synchronized void addOnlineCount() \{  
        WebSocketServer.onlineCount++;  
    \}  
   
    public static synchronized void subOnlineCount() \{  
        WebSocketServer.onlineCount--;

4.客户端代码：

<%@ page language="java" contentType="text/html; charset=utf-8"  
pageEncoding="utf-8"%>  
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">  
<html>  
<head>  
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">  
<title>Insert title here</title>  
  
<script type="text/javascript">  
function WebSocketTest() \{  
if ("WebSocket" in window) \{  
alert("您的浏览器支持 WebSocket!");  
  
// 打开一个 web socket  
var ws = new WebSocket("ws://localhost:8090/websocket");  
  
ws.onopen = function() \{  
// Web Socket 已连接上，使用 send() 方法发送数据  
ws.send("发送数据");  
alert("数据发送中...");  
\};  
  
ws.onmessage = function(evt) \{  
var received\_msg = evt.data;  
alert("数据已接收...");  
\};  
         ws.onclose = function() \{  
// 关闭 websocket  
alert("连接已关闭...");  
\};  
\}else \{  
// 浏览器不支持 WebSocket  
alert("您的浏览器不支持 WebSocket!");  
\}  
\}  
</script>  
</head>  
<body>  
<a href="javascript:WebSocketTest()">websocket</a>  
</body>  
</html>