springboot websocket server服务端代码demo
 2025-01-04  79507fa069048f0cfa2d4e5d843a390a 99+   8 分钟   1.1 k0次访问

springboot websocket server服务端代码demo

后端java springboot 项目

简单websocket server 端代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
<?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">
	<modelVersion>4.0.0</modelVersion>

	<groupId>com.cloudkd</groupId>
	<artifactId>websocket-server</artifactId>
	<version>1.0</version>
	<packaging>jar</packaging>

	<name>websocket</name>
	<description>websocket for Spring Boot</description>

	<parent>
		<groupId>org.springframework.boot</groupId>
		<artifactId>spring-boot-starter-parent</artifactId>
		<version>2.0.5.RELEASE</version>
		<relativePath/> <!-- lookup parent from repository -->
	</parent>

	<properties>
		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
		<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
		<java.version>1.8</java.version>
	</properties>

	<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>

		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-devtools</artifactId>
			<scope>runtime</scope>
		</dependency>
		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-test</artifactId>
			<scope>test</scope>
		</dependency>
		<dependency>
			<groupId>com.alibaba</groupId>
			<artifactId>fastjson</artifactId>
			<version>2.0.20</version>
		</dependency>
	</dependencies>

	<build>
		<plugins>
			<plugin>
				<groupId>org.springframework.boot</groupId>
				<artifactId>spring-boot-maven-plugin</artifactId>
			</plugin>
		</plugins>
	</build>


</project>

// 定义相关地址，以及session集合，用户存储管理client

ws://{ip}:{port}/websocket/data

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
package com.cloudkd.websocket;

import java.io.IOException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
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 com.alibaba.fastjson.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

@ServerEndpoint("/websocket/data")
@Component
public class WebSocketServer {
    private static final Logger log = LoggerFactory.getLogger(WebSocketServer.class);
    //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;
    //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
    private static CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();
    //与某个客户端的连接会话，需要通过它来给客户端发送数据

    private Session session;

    public static void kill(String id) {
        for (WebSocketServer item : webSocketSet) {
            try {
                if (item.session.getId().equals(id)) {
                    log.info("Kill {} success", id);
                    item.session.close();
                }
            } catch (IOException ignored) {
            }
        }
    }

    public static Map<String, Object> getAllSessionIds() {
        Map<String, Object> map = new HashMap<>();
        for (WebSocketServer item : webSocketSet) {
            map.put(item.session.getId(),item.session.getId());
        }
        return map;
    }

    /**
     * 连接建立成功调用的方法
     */
    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        //加入set中
        webSocketSet.add(this);
        //在线数加1
        addOnlineCount();
        log.info("有新连接加入！当前在线人数为" + getOnlineCount());
        try {
            sendMessage("连接成功");
        } catch (IOException e) {
            log.error("websocket IO异常");
        }
    }
    //	//连接打开时执行
    //	@OnOpen
    //	public void onOpen(@PathParam("user") String user, Session session) {
    //		currentUser = user;
    //		System.out.println("Connected ... " + session.getId());
    //	}

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose() {
        webSocketSet.remove(this);  //从set中删除
        subOnlineCount();           //在线数减1
        log.info("有一连接关闭！当前在线人数为" + getOnlineCount());
    }

    /**
     * 收到客户端消息后调用的方法
     *
     * @param message 客户端发送过来的消息
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        log.info("来自客户端的消息 sessionId = {} ,data :{}", session.getId(), message);
        //群发消息
        //        for (WebSocketServer item : webSocketSet) {
        //            try {
        //                item.sendMessage(message);
        //            } catch (IOException e) {
        //                e.printStackTrace();
        //            }
//        }
    }

    /**
     * @param session
     * @param error
     */
    @OnError
    public void onError(Session session, Throwable error) {
        log.error("发生错误");
        error.printStackTrace();
    }

    public void sendMessage(String message) throws IOException {
        this.session.getBasicRemote().sendText(message);
    }

    /**
     * 群发自定义消息
     */
    public static void sendInfo(BodyEntity message) {
        log.info("SEND MESSAGE:{}", message);

        for (WebSocketServer item : webSocketSet) {
            try {
                if (item.session.getId().equals(message.getSessionId())) {
                    JSONObject sa = JSONObject.parseObject(message.getMessage());
                    item.sendMessage(sa.toString());
                }
            } catch (IOException ignored) {
            }
        }
    }

    private static synchronized int getOnlineCount() {
        return onlineCount;
    }

    private static synchronized void addOnlineCount() {
        WebSocketServer.onlineCount++;
    }

    private static synchronized void subOnlineCount() {
        WebSocketServer.onlineCount--;
    }
}

// WebSocketConfig 相关配置

1
2
3
4
5
6
7
8
9
10
11
12
13
package com.cloudkd.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();
    }
}

// 管理api，可往对应连接发送消息

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package com.cloudkd.websocket;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;

@RestController
public class ManageController {
    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    @PostMapping(value = "/pushData")
    public Map<String, Object> pushVideoListToWeb(@RequestBody BodyEntity message) {
        Map<String, Object> result = new HashMap<String, Object>();
        try {
            WebSocketServer.sendInfo(message);
            result.put("operationResult", true);
        } catch (Exception e) {
            result.put("operationResult", true);
        }
        return result;
    }

    @GetMapping(value = "/kill")
    public Map<String, Object> pushVideoListToWeb(String id) {
        WebSocketServer.kill(id);
        return new HashMap<>();
    }

    @GetMapping(value = "/getAll")
    public Map<String, Object> getAll() {
        return WebSocketServer.getAllSessionIds();
    }
}


package com.cloudkd.websocket;
public class MessageEntity {
    private String message;
    private String sessionId;

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public String getSessionId() {
        return sessionId;
    }

    public void setSessionId(String sessionId) {
        this.sessionId = sessionId;
    }
}

// 测试页面

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
 </head>
<body>
<h3>server地址 :</h3>  <input id="serveraddress" type="text" value="ws://192.168.102.17:28080/elevator/realData" style="width: 500px;%"/><br/>
<h3>您的用户id :</h3>  <input id="userId" type="text"/><br/>
<button onclick="initSocket()">连接</button>
<br/>

=====================================================<br/>
消息 : <input id="message" type="text"/><br/>
<button onclick="send()">发送</button>
<br/>
=====================================================<br/>
连接状态 :
<button onclick="clearConnectStatu()">清空</button>
<br/>
<div id="connectStatu"></div>
<br/>

=====================================================<br/>
收到消息 :<br/>
<div id="receivedMessage"></div>
<br/>
=====================================================<br/>
心跳 :<br/>
<div id="heartdiv"></div>
<br/>

</body>

<script src="https://code.jquery.com/jquery-3.1.1.min.js"></script>
<script type="text/javascript">
    var heartflag = false;
    var webSocket = null;
    var tryTime = 0;
    $(function () {

//        initSocket();
        window.onbeforeunload = function () {

        };
    });

    /**
     * 初始化websocket，建立连接
     */
    function initSocket() {
        var serveraddress = $("#serveraddress").val();
        var userId = $("#userId").val();

        if (!window.WebSocket) {
            $("#connectStatu").append(getNowFormatDate() + "  您的浏览器不支持ws<br/>");
            return false;
        }

        webSocket = new WebSocket(serveraddress/* + "/" + userId*/);

        // 收到服务端消息
        webSocket.onmessage = function (msg) {
            if (msg.data !== "&") {
                $("#receivedMessage").append(getNowFormatDate() + "  收到消息 : " + msg.data + "<br/>");
            }
        };

        // 异常
        webSocket.onerror = function (event) {
            heartflag = false;
            $("#connectStatu").append(getNowFormatDate() + "  异常<br/>");
        };

        // 建立连接
        webSocket.onopen = function (event) {
            heartflag = true;
            heart();
            $("#connectStatu").append(getNowFormatDate() + "  建立连接成功<br/>");
            tryTime = 0;
        };

        // 断线重连
        webSocket.onclose = function () {
            heartflag = false;
            // 重试10次，每次之间间隔10秒
            if (tryTime < 10) {
                setTimeout(function () {
                    webSocket = null;
                    tryTime++;
                    initSocket();
                    $("#connectStatu").append(getNowFormatDate() + "  第" + tryTime + "次重连<br/>");
                }, 3 * 1000);
            } else {
                alert("重连失败.");
            }
        };

    }

    function send() {
        var message = $("#message").val();
        webSocket.send(message);
    }

    function clearConnectStatu() {
        $("#connectStatu").empty();
    }

    function getNowFormatDate() {
        var date = new Date();
        var seperator1 = "-";
        var seperator2 = ":";
        var month = date.getMonth() + 1;
        var strDate = date.getDate();
        if (month >= 1 && month <= 9) {
            month = "0" + month;
        }
        if (strDate >= 0 && strDate <= 9) {
            strDate = "0" + strDate;
        }
        var currentdate = date.getFullYear() + seperator1 + month + seperator1 + strDate
            + " " + date.getHours() + seperator2 + date.getMinutes()
            + seperator2 + date.getSeconds();
        return currentdate;
    }

    function heart() {
        if (heartflag) {
            webSocket.send("&");
            $("#heartdiv").append(getNowFormatDate() + "  心跳 <br/>");
        }
        setTimeout("heart()", 10 * 60 * 1000);

    }
</script>
</html>

// 配置文件

1
server.port=8081

// 启动Main

1
2
3
4
5
6
7
@SpringBootApplication
public class WebsocketApplication {

	public static void main(String[] args) {
		SpringApplication.run(WebsocketApplication.class, args);
	}
}

启动后默认,返回 http://{ip}:8081 ，进入测试页面

springboot websocket server服务端代码demo

https://removeif.github.io/java/frame/springboot-websocket-server代码demo示例.html

发布于

2025-01-04

更新于

2025-01-06

许可协议

# 相关文章
  1.Spring 如何解决循环依赖
  2.Spring cloud feign重试问题排查
# 推荐文章
  1.博客源码分享
  2.2019年国庆70周年阅兵完整图文解说(收藏!)
  3.博客中gitalk最新评论的获取
  4.中华人民共和国刑事诉讼法
  5.Github-Action自动部署Hexo博客
  6.springboot接入dubbo-nacos注册中心

喜欢这篇文章？打赏一下作者吧

评论
关注我

:D 一言句子获取中...

最新评论

加载中，最新评论有1分钟缓存...

最新文章

分类

归档

标签

java20
English19
grammar19
法律8
设计模式8
mysql7
springboot6
think5
工具教程5
二分查找3
Git2
dp2
elasticsearch62
hexo主题2
icarus主题配置2
redis2
spring2
动态规划2
双指针2
支付架构2
查看全部>>

订阅更新

输入邮箱开始订阅，更博后邮件通知！

广告


×