springboot 2.0websocket推送指定页面------------------gxy

矫情吗；* 2022-04-16 04:14 177阅读 0赞

加一个这个就行了，其他别乱加

<dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-websocket</artifactId>
            </dependency>

package com.gjy.mywork.config;
    
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.web.socket.server.standard.ServerEndpointExporter;
    
    /**
     * 开启WebSocket支持
     * @author gxy
     *	2018年11月16日
     *	my-work
     * 	success when you begin
     */
    @Configuration
    public class WebSocketConfig {
        @Bean
        public ServerEndpointExporter serverEndpointExporter() {
            return new ServerEndpointExporter();
        }
    
    }

package com.gjy.mywork.config;
    
    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.PathParam;
    import javax.websocket.server.ServerEndpoint;
    
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.stereotype.Component;
    
    /**
     * websocket基础类
     * @author gxy
     *	2018年11月16日
     *	my-work
     * 	success when you begin
     */
    @ServerEndpoint(value="/mySocket/{userid}")
    @Component
    public class MyWebSocket {
    	
    	
    	private static final Logger log = LoggerFactory.getLogger(MyWebSocket.class);
    
    	
    	//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
        private static int onlineCount = 0;
    
        //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
        private static CopyOnWriteArraySet<MyWebSocket> webSocketSet = new CopyOnWriteArraySet<MyWebSocket>();
    
        //与某个客户端的连接会话，需要通过它来给客户端发送数据
        private Session session;
        
        //接收sid
        private String sid="";
    
        /**
         * 连接建立成功调用的方法*/
        @OnOpen
        public void onOpen(Session session,@PathParam("userid") String sid) {
            this.session = session;
            webSocketSet.add(this);     //加入set中
            addOnlineCount();           //在线数加1
           log.info("有新连接加入！当前在线人数为" + getOnlineCount());
           this.sid = sid;
            try {
                sendMessage("已有连接人数："+getOnlineCount()+",连接人："+sid);
            } catch (IOException e) {
            	e.printStackTrace();
               log.info("IO异常");
            }
        }
    
        /**
         * 连接关闭调用的方法
         */
        @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("来自客户端的消息:" + message);
    
            //群发消息
            for (MyWebSocket item : webSocketSet) {
                try {
                    item.sendMessage(message);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    
        /**
         * 发生错误时调用
         */
        @OnError
        public void onError(Session session, Throwable error) {
           log.info("发生错误");
            error.printStackTrace();
        }
    
    
        public void sendMessage(String message) throws IOException {
            this.session.getBasicRemote().sendText(message);
            //this.session.getAsyncRemote().sendText(message);
        }
    
    
        /**
         * 群发自定义消息
         * */
        public static void sendInfoAll(String message) throws IOException {
            for (MyWebSocket item : webSocketSet) {
                try {
                    item.sendMessage(message);
                } catch (IOException e) {
                    continue;
                }
            }
        }
        
        public static void sendInfo(String message,@PathParam("sid") String sid) throws IOException {
        	log.info("推送消息到窗口"+sid+"，推送内容:"+message);
            for (MyWebSocket item : webSocketSet) {
                try {
                	//这里可以设定只推送给这个sid的，为null则全部推送
                	if(sid==null) {
                		item.sendMessage(message);
                	}else if(item.sid.equals(sid)){
                		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.gjy.mywork.controller;
    
    import java.io.IOException;
    
    import org.springframework.stereotype.Controller;
    import org.springframework.web.bind.annotation.GetMapping;
    import org.springframework.web.bind.annotation.PathVariable;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.ResponseBody;
    import org.springframework.web.servlet.ModelAndView;
    
    import com.gjy.mywork.config.MyWebSocket;
    import com.gjy.mywork.util.AjaxReturnVO;
    
    /**
     * websocket消息推送
     * @author gxy
     *	2018年11月16日
     *	my-work
     * 	success when you begin
     */
    @Controller
    @RequestMapping(value="/pm")
    public class PushMessageController {
    	
    	@GetMapping(value="/personalpage/{person}")
    	public ModelAndView gotoPMPage(@PathVariable String person) {
    		
    		ModelAndView mav = new ModelAndView("/wbs/person");
    		mav.addObject("cid", person);
    	
    		return mav;
    	}
    	
    	//推送数据接口
    	@ResponseBody
    	@RequestMapping("/push/{cid}")
    	public AjaxReturnVO pushToWeb(@PathVariable String cid,String message) {  
    
    		try {
    			MyWebSocket.sendInfo(message,cid);
    		} catch (IOException e) {
    			e.printStackTrace();
    			return AjaxReturnVO.fail(e.getMessage());
    		}  
    		return AjaxReturnVO.ok(cid);
    	} 
    
    }

页面

<!DOCTYPE html>
    <html xmlns:th="http://www.thymeleaf.org" >
    <head>
    <meta charset="UTF-8">
    <title>push message</title>
    </head>
    <body>
    
    	it's <span th:text="${cid}" id="userId"></span> page,message is :<br/>
    <div id="message">
    </div>
    </body>
    <script type="text/javascript" th:src="@{/js/jquery-3.1.0.min.js}"></script>
    <script type="text/javascript">
        var websocket = null;
    
        //判断当前浏览器是否支持WebSocket
        if('WebSocket' in window){
            websocket = new WebSocket("ws://localhost:68/mySocket/"+$('#userId').text());
        }
        else{
            alert('浏览器不支持接收后台推送')
        }
    
        //连接发生错误的回调方法
        websocket.onerror = function(){
            setMessageInnerHTML("error");
        };
    
        //连接成功建立的回调方法
        websocket.onopen = function(event){
            setMessageInnerHTML("open");
        }
    
        //接收到消息的回调方法
        websocket.onmessage = function(event){
            setMessageInnerHTML(event.data);
        }
    
        //连接关闭的回调方法
        websocket.onclose = function(){
            setMessageInnerHTML("close");
        }
    
        //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
        window.onbeforeunload = function(){
            websocket.close();
        }
    
        //将消息显示在网页上
        function setMessageInnerHTML(innerHTML){
            document.getElementById('message').innerHTML += innerHTML + '<br/>';
        }
    
        //关闭连接
        function closeWebSocket(){
            websocket.close();
        }
    
        //发送消息
        function send(){
            var message = document.getElementById('text').value;
            websocket.send(message);
        }
    </script>
    </html>

感谢：Moshow郑锴