websocket基于Springboot本人亲测

冷不防 2022-05-31 02:21 219阅读 0赞

非常感谢https://www.cnblogs.com/bianzy/p/5822426.html

**WebSocket 技术学习，更多知识请访问[https://www.itkc8.com][https_www.itkc8.com]**

<dependency>
    			<groupId>javax</groupId>
    			<artifactId>javaee-api</artifactId>
    			<version>8.0</version>
    			<scope>provided</scope>
    		</dependency>
    
    		
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter-websocket</artifactId>
    			<version>1.5.10.RELEASE</version>
    		</dependency>

package com.example.demo.config;
    
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.web.socket.server.standard.ServerEndpointExporter;
    
    /**
     * websocket
     *
     * @author huxu
     * @create 2018-02-04 11:04
     **/
    
    @Configuration
    public class WebSocketConfig {
        @Bean
        public ServerEndpointExporter serverEndpointExporter() {
            return new ServerEndpointExporter();
        }
    
    }

package com.example.demo;
    
    import org.springframework.stereotype.Component;
    
    import javax.websocket.*;
    import javax.websocket.server.ServerEndpoint;
    import java.io.IOException;
    import java.util.concurrent.CopyOnWriteArraySet;
    
    /**
     * websocket的具体实现类
     *
     * @author huxu
     * @create 2018-02-04 11:07
     **/
    
    @ServerEndpoint(value = "/websocket")
    @Component
    public class MyWebSocket {
        /**
         * 静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
         */
        private static int onlineCount = 0;
    
        /**
         * concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
         */
        private static CopyOnWriteArraySet<MyWebSocket> webSocketSet = new CopyOnWriteArraySet<>();
    
        /**
         * 与某个客户端的连接会话，需要通过它来给客户端发送数据
         */
        private Session session;
    
        /**
         * 连接建立成功调用的方法*/
        @OnOpen
        public void onOpen(Session session) {
            this.session = session;
            webSocketSet.add(this);     //加入set中
            addOnlineCount();           //在线数加1
            System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());
            /*try {
                sendMessage(CommonConstant.CURRENT_WANGING_NUMBER.toString());  
            } catch (IOException e) {
                System.out.println("IO异常");
            }*/
        }
    
        /**
         * 连接关闭调用的方法
         */
        @OnClose
        public void onClose() {
            webSocketSet.remove(this);  //从set中删除
            subOnlineCount();           //在线数减1
            System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());
        }
    
        /**
         * 收到客户端消息后调用的方法
         *
         * @param message 客户端发送过来的消息*/
        @OnMessage
        public void onMessage(String message, Session session) {
            System.out.println("来自客户端的消息:" + message);
    
            //群发消息
            for (MyWebSocket item : webSocketSet) {
                try {
                    item.sendMessage(message);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    
    
         //发生错误时调用
         @OnError
         public void onError(Session session, Throwable error) {
             System.out.println("发生错误");
             error.printStackTrace();
         }
    
    
         public void sendMessage(String message) throws IOException {
             this.session.getBasicRemote().sendText(message);
             //this.session.getAsyncRemote().sendText(message);
         }
    
    
         /**
          * 群发自定义消息
          * */
        public void sendInfo(String message) throws IOException {
            for (MyWebSocket item : webSocketSet) {
                try {
                    item.sendMessage(message);
                } catch (IOException e) {
                    continue;
                }
            }
        }
    
        public static synchronized int getOnlineCount() {
            return onlineCount;
        }
    
        public static synchronized void addOnlineCount() {
            MyWebSocket.onlineCount++;
        }
    
        public static synchronized void subOnlineCount() {
            MyWebSocket.onlineCount--;
        }
    }

package com.example.demo;
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.PostMapping;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RestController;
    
    /**
     * @author huxu
     * @create 2018-02-04 12:39
     **/
    @RestController
    @RequestMapping({"/", "/index"})
    public class Demo {
    
        @Autowired
        private MyWebSocket myWebSocket;
    
        @PostMapping("/hello")
        public String sayBy() throws Exception {
    
            for (int i=0;i<1000; i++) {
                Thread.sleep(1000);
                System.out.println("发消息了："+i);
                myWebSocket.sendInfo("推消息："+i);
            }
    
            return "good bye";
        }
    
    }

<!DOCTYPE html>
    <html lang="en">
    <head>
        <meta charset="UTF-8">
        <title>WebSocket测试</title>
    </head>
    <body>
        Welcome<br/>
        <input id="text" type="text"/>
        <button onclick="send()">发送消息</button>
        <hr/>
        <button onclick="closeWebSocket()">关闭WebSocket连接</button>
        <hr/>
        <div id="message"></div>
    
        <script src="js/WebSocket.js"></script>
    </body>
    </html>

**WebSocket 技术学习，更多知识请访问[https://www.itkc8.com][https_www.itkc8.com]**

[https_www.itkc8.com]: https://www.itkc8.com/