spring整合websocket（一）

妖狐艹你老母 2023-07-20 12:23 2阅读 0赞

1.pom文件引入依赖:spring.version为对应的spring版本

<dependency>
       <groupId>org.springframework</groupId>
        <artifactId>spring-websocket</artifactId>
        <version>${spring.version}</version>
    </dependency>
    <dependency>
              <groupId>org.springframework</groupId>
              <artifactId>spring-messaging</artifactId>
              <version>${spring.version}</version>
    </dependency>

2.配置文件

<?xml version="1.0" encoding="UTF-8"?>
    <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:websocket="http://www.springframework.org/schema/websocket" xsi:schemaLocation=" http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-4.0.xsd http://www.springframework.org/schema/websocket http://www.springframework.org/schema/websocket/spring-websocket-4.0.xsd">
     	
        
        <bean id="WebsocketHandler" class="com.thinkgem.jeesite.modules.meeting.websocket.WebsocketHandler"/>
        <websocket:handlers>
            
            <websocket:mapping path="/websocket" handler="WebsocketHandler"/>
            
            <websocket:handshake-interceptors>
                <bean class="com.thinkgem.jeesite.modules.meeting.websocket.HandshakeInterceptor"/>
            </websocket:handshake-interceptors>
            
            
        </websocket:handlers>
        
        <bean class="org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean">
            <property name="maxTextMessageBufferSize" value="8192"/>
            <property name="maxBinaryMessageBufferSize" value="8192"/>
        </bean>
    </beans>

3.java代码

package com.thinkgem.jeesite.modules.meeting.websocket;
    
    import java.util.Map;
    
    import org.springframework.http.server.ServerHttpRequest;
    import org.springframework.http.server.ServerHttpResponse;
    import org.springframework.stereotype.Service;
    import org.springframework.web.socket.WebSocketHandler;
    import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
    
    import com.thinkgem.jeesite.modules.meeting.utils.Constant;
    import com.thinkgem.jeesite.modules.sys.utils.UserUtils;
    
    @Service
    public class HandshakeInterceptor extends HttpSessionHandshakeInterceptor { 
    
        @Override
        public boolean beforeHandshake(ServerHttpRequest request,
                                       ServerHttpResponse response, WebSocketHandler wsHandler,
                                       Map<String, Object> attributes) throws Exception { 
            System.out.println("Before Handshake ... 握手前......");
            // 解决The extension [x-webkit-deflate-frame] is not supported问题
     		if (request.getHeaders().containsKey("Sec-WebSocket-Extensions")) { 
     			request.getHeaders().set("Sec-WebSocket-Extensions", "permessage-deflate");
     		}
     		// 使用userid区分WebSocketHandler，以便定向发送消息(使用shiro获取session) 
     		String userid = UserUtils.getUser().getId();
     		if (userid == null) { 
     			userid = "default-system";
     		}
     		attributes.put(Constant.WEBSOCKET_USERID, userid);
            
            return super.beforeHandshake(request, response, wsHandler, attributes);
        }
    
        @Override
        public void afterHandshake(ServerHttpRequest request,
                                   ServerHttpResponse response, WebSocketHandler wsHandler,
                                   Exception ex) { 
            System.out.println("After Handshake ... 握手后 ......");
            super.afterHandshake(request, response, wsHandler, ex);
        }
    }

package com.thinkgem.jeesite.modules.meeting.websocket;
    
    import java.io.IOException;
    import java.util.ArrayList;
    import java.util.List;
    
    import org.springframework.web.socket.CloseStatus;
    import org.springframework.web.socket.TextMessage;
    import org.springframework.web.socket.WebSocketSession;
    import org.springframework.web.socket.handler.TextWebSocketHandler;
    
    import com.thinkgem.jeesite.modules.meeting.utils.Constant;
    
    public class WebsocketHandler extends TextWebSocketHandler { 
    	
    	//存储用户
    	private static final List<WebSocketSession> users = new ArrayList<>();
    	
    	//接收到客户端消息时调用
        @Override
        public void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException { 
            System.out.println("text message: " + session.getPrincipal() + "-" + message.getPayload());
            String text = "get message by "+session.getPrincipal()+":"+message.getPayload();
            System.out.println(text);
            session.sendMessage(new TextMessage(text.getBytes()));
        }
    
       //用户进入系统监听
        @Override
        public void afterConnectionEstablished(WebSocketSession session)
                throws Exception { 
            users.add(session);
        }
    
        // 消息传输出错时调用
        @Override
        public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { 
            System.out.println("handleTransportError");
        }
    
    	// 一个客户端连接断开时关闭
        @Override
        public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception { 
            System.out.println("afterConnectionClosed");
            if (session.isOpen()) { 
                session.close();
            }
            users.remove(session);
        }
    
        @Override
        public boolean supportsPartialMessages() { 
            return false;
        }
        
        /** * 给所有的用户发送消息 */
        public void sendMessagesToUsers(TextMessage message) { 
            for (WebSocketSession user : users) { 
                try { 
                    //isOpen()在线就发送
                    if (user.isOpen()) { 
                        user.sendMessage(message);
                    }
                } catch (IOException e) { 
                    e.printStackTrace();
                }
            }
        }
    
        /** * 发送消息给指定的用户 */
        public void sendMessageToUser(String userId, TextMessage message) { 
            for (WebSocketSession user : users) { 
                if (user.getAttributes().get(Constant.WEBSOCKET_USERID).equals(userId)) { 
                    try { 
                        //isOpen()在线就发送
                        if (user.isOpen()) { 
                            user.sendMessage(message);
                        }
                    } catch (IOException e) { 
                        e.printStackTrace();
                    }
                }
            }
        }
    }

4.前端页面

<%@ page language="java" import="java.util.*"
    	contentType="text/html;charset=UTF-8"%>
    <%@ include file="/WEB-INF/views/include/taglib.jsp"%>
    <%@include file="/WEB-INF/views/include/head.jsp"%>
    <html>
    <head>
    	<title>test</title>
    	<meta name="decorator" content="default" />
    	<meta charset="UTF-8">
    	
    	<script src="https://cdn.jsdelivr.net/npm/sockjs-client@1/dist/sockjs.min.js"></script>
    	<script src="https://cdn.bootcss.com/stomp.js/2.3.3/stomp.min.js"></script>
    	<script src="https://cdn.bootcss.com/stomp.js/2.3.3/stomp.js"></script>
    	
    	<link rel="stylesheet" href="//cdn.bootcss.com/bootstrap/3.3.5/css/bootstrap.min.css">
    	<link rel="stylesheet" href="//cdn.bootcss.com/bootstrap/3.3.5/css/bootstrap-theme.min.css">
    	<script src="//cdn.bootcss.com/jquery/1.11.3/jquery.min.js"></script>
    	<script src="//cdn.bootcss.com/bootstrap/3.3.5/js/bootstrap.min.js"></script>
    	<title>测试</title>
    	<script type="text/javascript"> var stomp,sock; $(document).ready(function () { if (window.WebSocket){ websocketConfig(); } else { alert("错误","浏览器不支持websocket技术通讯."); } }); //websocket配置 function websocketConfig() { /* if('WebSocket' in window) { console.log("此浏览器支持websocket"); sock = new WebSocket("ws://localhost:8080"+ctx+"/webSocket"); } else { //连接url为endpointChat的endpoint,对应后台WebSoccketConfig的配置 } */ //SockJS所处理的URL是http://或https://,不再是ws://和wss:// //使用相对URL。例如，如果包含JavaScript的页面位于"http://localhost:8080/websocket"的路径下 sock = new SockJS("http://localhost:8080/hprd_business/webSocket"); sockHandle(); stompConfig(); } function stompConfig() { //使用STOMP子协议的websocket客户端 console.log("Connection status "+sock.readyState); stomp = Stomp.over(sock); stomp.connect({ }, function(frame) { //这里的订阅地址和controller的convertAndSendToUser地址一致，多出的user必须的，指的是用户之间发送消息 //这里的地址和后台的Controller sendChatMsg方法的chatAddress 一致 stomp.subscribe("/user/queue/showQRCode", function (message) { //处理响应消息 var json = JSON.parse(message.body); console.log(json); }); //订阅处理错误的消息地址,接收错误消息 stomp.subscribe("/user/queue/contactErrors${chatId}", function (message) { var response = JSON.parse(message.body); var error = response.error; alert("发送失败"+error); }); }); } //发送消息 function sendMessage() { //文本框的内容 var content = $("#content").val(); if ($.trim($("#content").val()).length <= 0){ return ; } //拼接成json传到后台 var data = { chatId:'${chatId}', schoolId:'${schoolId}', toUserId:'${receiverId}', content:content, fromUserId:'${accountId}', appId:'${appId}', msgType:1 }; //发送 stomp.send("/sendChatMsg",{ },JSON.stringify(data) ); } function sockHandle() { sock.onopen = function () { console.log("------连接成功------"); }; sock.onmessage = function (event) { console.log('-------Received: ' + event.data); }; sock.onclose = function (event) { console.log('--------Info: connection closed.------'); }; //连接发生错误 sock.onerror = function () { alert("连接错误", "网络超时或通讯地址错误."); disconnect(); }; } //关闭websocket function disconnect() { if (sock != null) { sock.close(); sock = null; } } </script>
    </head>
    
    <body>
    	<div class="page-header" id="tou">webSocket多终端聊天测试</div>
    	<div class="well" id="msg"></div>
    	<div class="col-lg">
    		<div class="input-group">
    			<input type="text" class="form-control" placeholder="发送信息..." width="10" id="message"> <span class="input-group-btn">
    				<button class="btn btn-default" type="button" id="send">发送</button>
    			</span>
    		</div>
    	</div>
    </body>
    </html>