SpringBoot 整合 Socket 实战案例，实现单点发送、广播群发，1对1，1对多

逃离我推掉我的手 2023-09-23 16:11 9阅读 0赞

后端 + 前端简单HTML页面

### 功能场景点： ###

1.  群发，所有人都能收到
2.  局部群发，部分人群都能收到
3.  单点推送， 指定某个人的页面

惯例，先看看本次实战示例项目结构：

![1de6dc53665f946b080464a52e38403d.png][]

可以看到内容不多，也就是说，springboot 整合socket， 跟着我学，轻轻松松。

不多说，开始：

① pom引入核心依赖

<dependencies>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.75</version>
        </dependency>
        <dependency>
            <groupId>com.corundumstudio.socketio</groupId>
            <artifactId>netty-socketio</artifactId>
            <version>1.7.7</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>

② yml加上配置项

server:
      port: 8089
     
    socketio:
       host: localhost
       port: 8503
       maxFramePayloadLength: 1048576
       maxHttpContentLength: 1048576
       bossCount: 1
       workCount: 100
       allowCustomRequests: true
       upgradeTimeout: 10000
       pingTimeout: 60000
       pingInterval: 25000

③ 创建socket配置加载类 MySocketConfig.java

import com.corundumstudio.socketio.SocketConfig;
    import com.corundumstudio.socketio.SocketIOServer;
    import com.corundumstudio.socketio.annotation.SpringAnnotationScanner;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
     
    /**
     * @Author: JCccc
     * @Description:
     * @Date: 2022/06/13 21:50
     */
    @Configuration
    public class MySocketConfig{
     
        @Value("${socketio.host}")
        private String host;
     
        @Value("${socketio.port}")
        private Integer port;
     
        @Value("${socketio.bossCount}")
        private int bossCount;
     
        @Value("${socketio.workCount}")
        private int workCount;
     
        @Value("${socketio.allowCustomRequests}")
        private boolean allowCustomRequests;
     
        @Value("${socketio.upgradeTimeout}")
        private int upgradeTimeout;
     
        @Value("${socketio.pingTimeout}")
        private int pingTimeout;
     
        @Value("${socketio.pingInterval}")
        private int pingInterval;
     
        @Bean
        public SocketIOServer socketIOServer() {
            SocketConfig socketConfig = new SocketConfig();
            socketConfig.setTcpNoDelay(true);
            socketConfig.setSoLinger(0);
            com.corundumstudio.socketio.Configuration config = new com.corundumstudio.socketio.Configuration();
            buildSocketConfig(socketConfig, config);
            return new SocketIOServer(config);
        }
     
        /**
         * 扫描netty-socketIo的注解( @OnConnect、@OnEvent等）
         */
        @Bean
        public SpringAnnotationScanner springAnnotationScanner() {
            return new SpringAnnotationScanner(socketIOServer());
        }
     
        private void buildSocketConfig(SocketConfig socketConfig, com.corundumstudio.socketio.Configuration config) {
            config.setSocketConfig(socketConfig);
            config.setHostname(host);
            config.setPort(port);
            config.setBossThreads(bossCount);
            config.setWorkerThreads(workCount);
            config.setAllowCustomRequests(allowCustomRequests);
            config.setUpgradeTimeout(upgradeTimeout);
            config.setPingTimeout(pingTimeout);
            config.setPingInterval(pingInterval);
        }
    }

④创建消息实体 MyMessage.java

/**
     * @Author: JCccc
     * @Date: 2022-07-23 9:05
     * @Description:
     */
    public class MyMessage {
     
        private String type;
     
        private String content;
     
        private String from;
     
        private String to;
     
        private String channel;
     
     
        public String getType() {
            return type;
        }
     
        public void setType(String type) {
            this.type = type;
        }
     
        public String getContent() {
            return content;
        }
     
        public void setContent(String content) {
            this.content = content;
        }
     
        public String getFrom() {
            return from;
        }
     
        public void setFrom(String from) {
            this.from = from;
        }
     
        public String getTo() {
            return to;
        }
     
        public void setTo(String to) {
            this.to = to;
        }
     
        public String getChannel() {
            return channel;
        }
     
        public void setChannel(String channel) {
            this.channel = channel;
        }
    }

代码简析：

![d040907a11c627fd2b797d08233bca02.png][]

⑤创建 socket handler 负责记录客户端 连接、下线

MySocketHandler.java

import com.corundumstudio.socketio.SocketIOClient;
    import com.corundumstudio.socketio.SocketIOServer;
    import com.corundumstudio.socketio.annotation.OnConnect;
    import com.corundumstudio.socketio.annotation.OnDisconnect;
    import com.socket.mysocket.util.SocketUtil;
    import org.springframework.beans.factory.annotation.Autowired;
    import javax.annotation.PostConstruct;
    import javax.annotation.PreDestroy;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.stereotype.Component;
     
     
    /**
     * @Author: JCccc
     * @Description:
     * @Date: 2022/6/23 21:21
     */
    @Component
    public class MySocketHandler {
        private final Logger log = LoggerFactory.getLogger(this.getClass());
        @Autowired
        private SocketIOServer socketIoServer;
        @PostConstruct
        private void start(){
            try {
                socketIoServer.start();
            }catch (Exception e){
                e.printStackTrace();
            }
        }
        @PreDestroy
        private void destroy(){
            try {
            socketIoServer.stop();
            }catch (Exception e){
                e.printStackTrace();
            }
        }
        @OnConnect
        public void connect(SocketIOClient client) {
            String userFlag = client.getHandshakeData().getSingleUrlParam("userFlag");
            SocketUtil.connectMap.put(userFlag, client);
            log.info("客户端userFlag: "+ userFlag+ "已连接");
        }
        @OnDisconnect
        public void onDisconnect(SocketIOClient client) {
            String userFlag = client.getHandshakeData().getSingleUrlParam("userFlag");
            log.info("客户端userFlag:" + userFlag + "断开连接");
            SocketUtil.connectMap.remove(userFlag, client);
        }
    }

代码简析：

![c17a4cb5695bff6223adcc576e5d3d65.png][]

⑥ 封装的socket 小函数

SocketUtil.java

import com.alibaba.fastjson.JSON;
    import com.alibaba.fastjson.JSONObject;
    import com.corundumstudio.socketio.AckRequest;
    import com.corundumstudio.socketio.SocketIOClient;
    import com.corundumstudio.socketio.annotation.OnEvent;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Component;
    import org.springframework.util.StringUtils;
     
    import java.util.Map;
    import java.util.Objects;
    import java.util.concurrent.ConcurrentHashMap;
    import java.util.concurrent.ConcurrentMap;
     
    /**
     * @Author: JCccc
     * @Description:
     * @Date: 2022/6/23 21:28
     */
    @Component
    public class SocketUtil {
     
        private final Logger log = LoggerFactory.getLogger(this.getClass());
     
        //暂且把用户&客户端信息存在缓存
        public static ConcurrentMap<String, SocketIOClient> connectMap = new ConcurrentHashMap<>();
     
        @OnEvent(value = "CHANNEL_SYSTEM")
        public void systemDataListener(String receiveMsg) {
            if (!StringUtils.hasLength(receiveMsg)){
                return;
            }
            JSONObject msgObject = (JSONObject) JSON.parse(receiveMsg);
            String userFlag = String.valueOf(msgObject.get("from"));
            String content = String.valueOf(msgObject.get("content"));
            log.info("收到用户 ： {} 推送到系统频道的一条消息 :{}",userFlag,content );
        }
     
        public void sendToAll(Map<String, Object> msg,String sendChannel) {
            if (connectMap.isEmpty()){
                return;
            }
            //给在这个频道的每个客户端发消息
            for (Map.Entry<String, SocketIOClient> entry : connectMap.entrySet()) {
                entry.getValue().sendEvent(sendChannel, msg);
            }
        }
     
        public void sendToOne(String userFlag, Map<String, Object> msg,String sendChannel) {
            //拿出某个客户端信息
            SocketIOClient socketClient = getSocketClient(userFlag);
            if (Objects.nonNull(socketClient) ){
                //单独给他发消息
                socketClient.sendEvent(sendChannel,msg);
            }
        }
     
     
        /**
         * 识别出客户端
         * @param userFlag
         * @return
         */
        public SocketIOClient getSocketClient(String userFlag){
            SocketIOClient client = null;
            if (StringUtils.hasLength(userFlag) &&  !connectMap.isEmpty()){
                for (String key : connectMap.keySet()) {
                    if (userFlag.equals(key)){
                        client = connectMap.get(key);
                    }
                }
            }
            return client;
        }
    }

代码简析：

![58bc82bbf20b472c918199fb89ef7086.png][]

⑦写1个接口，模拟场景，前端页面调用后端接口，做消息推送

TestController.java

import com.socket.mysocket.dto.MyMessage;
    import com.socket.mysocket.util.SocketUtil;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.*;
     
    import java.util.HashMap;
    import java.util.Map;
     
    /**
     * @Author: JCccc
     * @Description:
     * @Date: 2022/06/13 21:50
     */
    @RestController
    public class TestController {
        public final static String SEND_TYPE_ALL = "ALL";
        public final static String SEND_TYPE_ALONE = "ALONE";
        @Autowired
        SocketUtil socketUtil;
     
        @PostMapping("/testSendMsg")
        public String testSendMsg(@RequestBody MyMessage myMessage){
            Map<String, Object> map = new HashMap<>();
            map.put("msg",myMessage.getContent());
     
            //群发
            if (SEND_TYPE_ALL.equals(myMessage.getType())){
                socketUtil.sendToAll( map,myMessage.getChannel());
                return "success";
            }
            //指定单人
            if (SEND_TYPE_ALONE.equals(myMessage.getType())){
                socketUtil.sendToOne(myMessage.getTo(), map, myMessage.getChannel());
                return "success";
            }
     
            return "fail";
        }
    }

代码简析：

![d058516fdedb40db83d437fa1ed1e674.png][]

好了，7步了。一切已经就绪了。

### 前端简单页面 ###

接下来搞点前端HTML页面， 玩一玩看看效果:

![b5238da898214db89eb388ed62556b83.png][]

**第一个页面：**

TestClientStudentJC.html

<!DOCTYPE html>
    <html>
    <head>
        <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
        <title>我要连SOCKET</title>
        <base>
        <script src="https://cdn.bootcss.com/jquery/3.4.0/jquery.min.js"></script>
        <script src="https://cdn.bootcss.com/socket.io/2.2.0/socket.io.js"></script>
        <style>
            body {
                padding: 20px;
            }
            #console {
                height: 450px;
                overflow: auto;
            }
            .msg-color {
                color: green;
            }
        </style>
    </head>
     
    <body>
    <div id="console" class="well"></div>
     
     
    <div id="conversationDiv">
        <labal>给系统推消息</labal>
        <input type="text" id="content"/>
        <button id="btnSendToSystem" onclick="sendSys();">发送</button>
    </div>
     
     
    </body>
    <script type="text/javascript">
        var socket;
        connect();
     
        function connect() {
            var userFlag = 'user_JC';
            var opts = {
                query: 'userFlag=' + userFlag
            };
            socket = io.connect('http://localhost:8503', opts);
            socket.on('connect', function () {
                console.log("连接成功");
                output('当前用户是：' + userFlag );
                output('<span class="msg-color">连接成功了。</span>');
            });
            socket.on('disconnect', function () {
                output('<span class="msg-color">下线了。 </span>');
            });
     
            socket.on('CHANNEL_STUDENT', function (data) {
                let msg= JSON.stringify(data)
                output('收到学生频道消息了：' + msg );
                console.log(data);
     
            });
            socket.on('CHANNEL_SYSTEM', function (data) {
                let msg= JSON.stringify(data)
                output('收到系统全局消息了：' + msg );
                console.log(data);
     
            });
     
        }
     
        function sendSys() {
            console.log('发送消息给服务端');
            var content = document.getElementById('content').value;
     
            socket.emit('CHANNEL_SYSTEM',JSON.stringify({
                'content': content,
                'from': 'user_JC'
            }));
     
        }
        function output(message) {
            var element = $("<div>" + message + "</div>");
            $('#console').prepend(element);
        }
        
    </script>
    </html>

代码简析：

![dd55cc29a58e4b5eb4ac18f6f8ca0037.png][]

**第二个页面，跟第一个基本一样，改一下用户唯一标识：**

TestClientStudentPU.html

<!DOCTYPE html>
    <html>
    <head>
        <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
        <title>我要连SOCKET</title>
        <base>
        <script src="https://cdn.bootcss.com/jquery/3.4.0/jquery.min.js"></script>
        <script src="https://cdn.bootcss.com/socket.io/2.2.0/socket.io.js"></script>
        <style>
            body {
                padding: 20px;
            }
            #console {
                height: 450px;
                overflow: auto;
            }
            .msg-color {
                color: green;
            }
        </style>
    </head>
     
    <body>
    <div id="console" class="well"></div>
     
     
    <div id="conversationDiv">
        <labal>给系统推消息</labal>
        <input type="text" id="content"/>
        <button id="btnSendToSystem" onclick="sendSys();">发送</button>
    </div>
     
     
    </body>
    <script type="text/javascript">
        var socket;
        connect();
     
        function connect() {
            var userFlag = 'user_PU';
            var opts = {
                query: 'userFlag=' + userFlag
            };
            socket = io.connect('http://localhost:8503', opts);
            socket.on('connect', function () {
                console.log("连接成功");
                output('当前用户是：' + userFlag );
                output('<span class="msg-color">连接成功了。</span>');
            });
            socket.on('disconnect', function () {
                output('<span class="msg-color">下线了。 </span>');
            });
     
            socket.on('CHANNEL_STUDENT', function (data) {
                let msg= JSON.stringify(data)
                output('收到学生频道消息了：' + msg );
                console.log(data);
     
            });
            socket.on('CHANNEL_SYSTEM', function (data) {
                let msg= JSON.stringify(data)
                output('收到系统全局消息了：' + msg );
                console.log(data);
     
            });
     
        }
     
        function sendSys() {
            console.log('发送消息给服务端');
            var content = document.getElementById('content').value;
     
            socket.emit('CHANNEL_SYSTEM',JSON.stringify({
                'content': content,
                'from': 'user_PU'
            }));
     
        }
        function output(message) {
            var element = $("<div>" + message + "</div>");
            $('#console').prepend(element);
        }
     
    </script>
    </html>

OK，把项目跑起来，开始玩。

直接访问客户端页面 模拟学生 JC连接socket：

> http://127.0.0.1:8089/TestClientStudentJC.html

![3d3ca4d45649f24b02f3c68d3bc47380.png][]

可以看到服务端有监测到：

![dafc9de015acb5f696e4eaa8b602a2ff.png][]

这里监测的：

![f182c4f9123158e358ae58b5de6a5462.png][]

先试试客户端给系统推消息先：

![782f3d8a2a0c4185b473ac00392c0ef1.png][]

可以看到服务端成功收到消息：

![b129ed8c6e832ac4314ce5007ced0323.png][]

这种方式，其实是因为服务监听了相关的频道：

![85a10b2ef7aca8fc922585ef4da6fc1d.png][]

前端使用JS推到这个系统频道：

![bc606b506b781c6af14ae2c2ef6bdc84.png][]

> ps：其实前端给服务端推消息，其实调用接口就可以。

OK，进入核心应用场景1：

群发，所有人都能收到

系统给连上的客户端都推送消息

![2886d6f2f386107d7e9d9fcd38669a51.png][]

{
    
    "type": "ALL",
    
    "content":"你们好,这是一条广播消息,全部人都能收到",
    
    "channel":"CHANNEL_SYSTEM"
    
    }

看看效果：

![63b242200b5f4430bede2ed48db27384.png][]

然后是场景2，局部群发，部分人群都能收到

其实也就是通过HTML 客户端监听主题做区分就好：

直接拉人口，升3 ：

模拟2个学生，1个老师都连接上了socket

当然，老师监听的是 老师频道：

![741e497bf4117a3fec2a1948e939ea96.png][]

然后我们模拟推送一下消息到指定的老师频道：

![3f31485094e1434abad7cb456a49133d.png][]

{
    
    "type": "ALL",
    
    "content":"给老师们推一条消息！！！",
    
    "channel":"CHANNEL_TEACHER"
    
    }

![a66d9bfd6d6b4d372d2c29c9924afde3.png][]

最后一个场景，也就是单点推送，指定某个人收到

模拟 学生 PU 给 学生JC 推消息：

![019ba2f61a0155d480e113e95b25c449.png][]

可以看到在学生频道的JC正常收到了PU的消息：

![6023c8cc0909445c8f7def02a610371d.png][]

好了，该篇就到这吧。

[1de6dc53665f946b080464a52e38403d.png]: https://img-blog.csdnimg.cn/img_convert/1de6dc53665f946b080464a52e38403d.png
[d040907a11c627fd2b797d08233bca02.png]: https://img-blog.csdnimg.cn/img_convert/d040907a11c627fd2b797d08233bca02.png
[c17a4cb5695bff6223adcc576e5d3d65.png]: https://img-blog.csdnimg.cn/img_convert/c17a4cb5695bff6223adcc576e5d3d65.png
[58bc82bbf20b472c918199fb89ef7086.png]: https://img-blog.csdnimg.cn/58bc82bbf20b472c918199fb89ef7086.png
[d058516fdedb40db83d437fa1ed1e674.png]: https://img-blog.csdnimg.cn/d058516fdedb40db83d437fa1ed1e674.png
[b5238da898214db89eb388ed62556b83.png]: https://img-blog.csdnimg.cn/b5238da898214db89eb388ed62556b83.png
[dd55cc29a58e4b5eb4ac18f6f8ca0037.png]: https://img-blog.csdnimg.cn/dd55cc29a58e4b5eb4ac18f6f8ca0037.png
[3d3ca4d45649f24b02f3c68d3bc47380.png]: https://img-blog.csdnimg.cn/img_convert/3d3ca4d45649f24b02f3c68d3bc47380.png
[dafc9de015acb5f696e4eaa8b602a2ff.png]: https://img-blog.csdnimg.cn/img_convert/dafc9de015acb5f696e4eaa8b602a2ff.png
[f182c4f9123158e358ae58b5de6a5462.png]: https://img-blog.csdnimg.cn/img_convert/f182c4f9123158e358ae58b5de6a5462.png
[782f3d8a2a0c4185b473ac00392c0ef1.png]: https://img-blog.csdnimg.cn/782f3d8a2a0c4185b473ac00392c0ef1.png
[b129ed8c6e832ac4314ce5007ced0323.png]: https://img-blog.csdnimg.cn/img_convert/b129ed8c6e832ac4314ce5007ced0323.png
[85a10b2ef7aca8fc922585ef4da6fc1d.png]: https://img-blog.csdnimg.cn/img_convert/85a10b2ef7aca8fc922585ef4da6fc1d.png
[bc606b506b781c6af14ae2c2ef6bdc84.png]: https://img-blog.csdnimg.cn/img_convert/bc606b506b781c6af14ae2c2ef6bdc84.png
[2886d6f2f386107d7e9d9fcd38669a51.png]: https://img-blog.csdnimg.cn/img_convert/2886d6f2f386107d7e9d9fcd38669a51.png
[63b242200b5f4430bede2ed48db27384.png]: https://img-blog.csdnimg.cn/63b242200b5f4430bede2ed48db27384.png
[741e497bf4117a3fec2a1948e939ea96.png]: https://img-blog.csdnimg.cn/img_convert/741e497bf4117a3fec2a1948e939ea96.png
[3f31485094e1434abad7cb456a49133d.png]: https://img-blog.csdnimg.cn/3f31485094e1434abad7cb456a49133d.png
[a66d9bfd6d6b4d372d2c29c9924afde3.png]: https://img-blog.csdnimg.cn/img_convert/a66d9bfd6d6b4d372d2c29c9924afde3.png
[019ba2f61a0155d480e113e95b25c449.png]: https://img-blog.csdnimg.cn/img_convert/019ba2f61a0155d480e113e95b25c449.png
[6023c8cc0909445c8f7def02a610371d.png]: https://img-blog.csdnimg.cn/6023c8cc0909445c8f7def02a610371d.png