`

Spring+Websocket实现消息的推送

阅读更多
本文主要有三个步骤 1、用户登录后建立websocket连接,默认选择websocket连接,如果浏览器不支持,则使用sockjs进行模拟连接 2、建立连接后,服务端返回该用户的未读消息 3、服务端进行相关操作后,推送给某一个用户或者所有用户新消息 相关环境 Spring4.0.6(要选择4.0+),tomcat7.0.55

Websocet服务端实现
WebSocketConfig.java
@Configuration
@EnableWebMvc
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(systemWebSocketHandler(),"/webSocketServer").addInterceptors(new WebSocketHandshakeInterceptor());
 
        registry.addHandler(systemWebSocketHandler(), "/sockjs/webSocketServer").addInterceptors(new WebSocketHandshakeInterceptor())
                .withSockJS();
    }
 
    @Bean
    public WebSocketHandler systemWebSocketHandler(){
        return new SystemWebSocketHandler();
    }
 
}

不要忘记在springmvc的配置文件中配置对此类的自动扫描
<context:component-scan base-package="com.ldl.origami.websocket" />

@Configuration

@EnableWebMvc
@EnableWebSocket
这三个大致意思是使这个类支持以@Bean的方式加载bean,并且支持springmvc和websocket,不是很准确大致这样,试了一下@EnableWebMvc不加也没什么影响,@Configuration本来就支持springmvc的自动扫描
registry.addHandler(systemWebSocketHandler(),"/webSocketServer").addInterceptors(new WebSocketHandshakeInterceptor())

用来注册websocket server实现类,第二个参数是访问websocket的地址
registry.addHandler(systemWebSocketHandler(), "/sockjs/webSocketServer").addInterceptors(new WebSocketHandshakeInterceptor())
                .withSockJS();
    }

这个是使用Sockjs的注册方法

首先SystemWebSocketHandler.java

public class SystemWebSocketHandler implements WebSocketHandler {
 
    private static final Logger logger;
 
    private static final ArrayList<WebSocketSession> users;
 
    static {
        users = new ArrayList<>();
        logger = LoggerFactory.getLogger(SystemWebSocketHandler.class);
    }
 
    @Autowired
    private WebSocketService webSocketService;
 
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        logger.debug("connect to the websocket success......");
        users.add(session);
        String userName = (String) session.getAttributes().get(Constants.WEBSOCKET_USERNAME);
        if(userName!= null){
            //查询未读消息
            int count = webSocketService.getUnReadNews((String) session.getAttributes().get(Constants.WEBSOCKET_USERNAME));
 
            session.sendMessage(new TextMessage(count + ""));
        }
    }
 
    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
 
        //sendMessageToUsers();
    }
 
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        if(session.isOpen()){
            session.close();
        }
        logger.debug("websocket connection closed......");
        users.remove(session);
    }
 
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        logger.debug("websocket connection closed......");
        users.remove(session);
    }
 
    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
 
    /**
     * 给所有在线用户发送消息
     *
     * @param message
     */
    public void sendMessageToUsers(TextMessage message) {
        for (WebSocketSession user : users) {
            try {
                if (user.isOpen()) {
                    user.sendMessage(message);
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
 
    /**
     * 给某个用户发送消息
     *
     * @param userName
     * @param message
     */
    public void sendMessageToUser(String userName, TextMessage message) {
        for (WebSocketSession user : users) {
            if (user.getAttributes().get(Constants.WEBSOCKET_USERNAME).equals(userName)) {
                try {
                    if (user.isOpen()) {
                        user.sendMessage(message);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;
            }
        }
    }
}


相关内容大家一看就能明白,就不多解释了

然后WebSocketHandshakeInterceptor.java
public class WebSocketHandshakeInterceptor implements HandshakeInterceptor {
 
    private static Logger logger = LoggerFactory.getLogger(HandshakeInterceptor.class);
    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object
                > attributes) throws Exception {
        if (request instanceof ServletServerHttpRequest) {
            ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
            HttpSession session = servletRequest.getServletRequest().getSession(false);
            if (session != null) {
                //使用userName区分WebSocketHandler,以便定向发送消息
                String userName = (String) session.getAttribute(Constants.SESSION_USERNAME);
                attributes.put(Constants.WEBSOCKET_USERNAME,userName);
            }
        }
        return true;
    }
 
    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
 
    }
}

这个的主要作用是取得当前请求中的用户名,并且保存到当前的WebSocketHandler中,以便确定WebSocketHandler所对应的用户,具体可参考HttpSessionHandshakeInterceptor

用户登录建立websocket连接

index.jsp

<script type="text/javascript" src="http://localhost:8080/Origami/websocket/sockjs-0.3.min.js"></script>
        <script>
            var websocket;
            if ('WebSocket' in window) {
                websocket = new WebSocket("ws://localhost:8080/Origami/webSocketServer");
            } else if ('MozWebSocket' in window) {
                websocket = new MozWebSocket("ws://localhost:8080/Origami/webSocketServer");
            } else {
                websocket = new SockJS("http://localhost:8080/Origami/sockjs/webSocketServer");
            }
            websocket.onopen = function (evnt) {
            };
            websocket.onmessage = function (evnt) {
                $("#msgcount").html("(<font color='red'>"+evnt.data+"</font>)")
            };
            websocket.onerror = function (evnt) {
            };
            websocket.onclose = function (evnt) {
            }
 
        </script>

使用sockjs时要注意

1、这两个的写法
<script type="text/javascript" src="http://localhost:8080/Origami/websocket/sockjs-0.3.min.js"></script>

websocket = new SockJS("http://localhost:8080/Origami/sockjs/webSocketServer");

2、web.xml中
<web-app version="3.0" xmlns="http://java.sun.com/xml/ns/javaee"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_3_1.xsd">

version
web-app_3_1.xsd

这两个的版本都要是3.0+

然后在这个servlet中加入
<async-supported>true</async-supported>

<servlet>
        <servlet-name>appServlet</servlet-name>
        <servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
        <init-param>
            <param-name>contextConfigLocation</param-name>
            <param-value>classpath*:servlet-context.xml</param-value>
        </init-param>
        <load-on-startup>1</load-on-startup>
        <async-supported>true</async-supported>
    </servlet>

然后所有的filter中也加入
<async-supported>true</async-supported>

3、添加相关依赖
<dependency>
            <groupId>com.fasterxml.jackson.core</groupId>
            <artifactId>jackson-annotations</artifactId>
            <version>2.3.0</version>
        </dependency>
        <dependency>
            <groupId>com.fasterxml.jackson.core</groupId>
            <artifactId>jackson-core</artifactId>
            <version>2.3.1</version>
        </dependency>
        <dependency>
            <groupId>com.fasterxml.jackson.core</groupId>
            <artifactId>jackson-databind</artifactId>
            <version>2.3.3</version>
        </dependency>


好了,现在websocket可以正常建立起来了



返回用户未读的消息

当连接建立后,会进入SystemWebSocketHandler的afterConnectionEstablished方法,代码看上边,取出WebSocketHandshakeInterceptor中保存的用户名

查询信息后使用session.sendMessage(new TextMessage(count + ""));返回给用户,从哪来回哪去


服务端推送消息给用户
@Controller
public class AdminController {
 
    static Logger logger = LoggerFactory.getLogger(AdminController.class);
 
    @Autowired(required = false)
    private AdminService adminService;
 
    @Bean
    public SystemWebSocketHandler systemWebSocketHandler() {
        return new SystemWebSocketHandler();
    }
 
 
    @RequestMapping("/auditing")
    @ResponseBody
    public String auditing(HttpServletRequest request){
        //无关代码都省略了
        int unReadNewsCount = adminService.getUnReadNews(username);
        systemWebSocketHandler().sendMessageToUser(username, new TextMessage(unReadNewsCount + ""));
        return result;
    }
}

在这里可以使用sendMessageToUser给某个用户推送信息,也可以使用sendMessageToUsers给所有用户推送信息
http://blog.chenzuhuang.com/archive/28.html
分享到:
评论

相关推荐

    Spring Boot 整合 Netty + WebSocket 实时消息推送

    在"Spring Boot 整合 Netty + WebSocket 实时消息推送"项目中,我们主要关注以下几个核心知识点: 1. **Spring Boot集成WebSocket**:Spring Boot提供了Spring WebSocket模块,可以方便地集成WebSocket功能。通过...

    如何使用Spring Boot 3.X + WebSocket来实现推送消息功能

    如何使用Postman来发送数据到后端Spring Boot应用程序,并使用WebSocket建立实时通信通道,以便后端应用程序可以向Postman推送消息。 学会这个方法,就能缩短你的项目开发周期,以及提高你的开发效率。如果提出您的...

    SpringBoot+WebSocket实现消息推送及简单的聊天功能

    本文将探讨如何结合SpringBoot和WebSocket技术来实现消息推送和简单的聊天功能。 首先,要理解WebSocket协议的核心特性。WebSocket是一种在单个TCP连接上进行全双工通信的协议,它允许服务器和客户端之间进行持久...

    spring boot+vue+websocket带token身份认证推送消息实现

    本知识点介绍了Spring Boot集成WebSocket,并在前后端分离的架构下,通过身份认证实现消息推送功能的关键实现步骤。重点涉及了WebSocket的配置和编程模型、消息推送机制、以及前后端实现细节。通过掌握这些知识点,...

    spring+websocket+hibernate小例子

    3. **Controller层**:Spring MVC的Controller层负责处理HTTP请求,可能有一个或多个Controller类,它们接收HTTP请求,根据请求类型决定是否需要通过WebSocket向客户端推送数据。 4. **Service层**:业务逻辑通常在...

    SpringBoot+SpringSecurity+WebSocket

    5. 实现消息推送:在服务端,可以通过WebSocketSession接口向特定用户或所有用户推送消息。 6. 错误处理:处理WebSocket连接异常关闭,如网络中断等情况,确保系统的健壮性。 在项目中,"websocket"可能是包含...

    spring+websocketdemo

    WebSocket是Web交互技术的一种革新,它为实时双向通信提供了标准协议。在传统的HTTP协议中,客户端与服务器之间的通信是请求-响应...在实际项目中,这种技术可以应用于在线聊天、股票实时推送、游戏同步等多种场景。

    springboot+rabbitMQ+websocket

    接下来,我们将深入探讨如何利用Spring Boot、RabbitMQ和WebSocket来实现一个简单的消息推送聊天应用。 首先,让我们了解Spring Boot如何集成RabbitMQ。在Spring Boot项目中引入`spring-boot-starter-amqp`依赖,它...

    SpringBoot+WebSocket+Netty实现消息推送的示例代码

    在本文中,我们将深入探讨如何使用SpringBoot、WebSocket和Netty来实现消息推送功能。首先,让我们了解一下这三个组件的关键概念。 SpringBoot是一个流行的Java框架,它简化了Spring应用程序的开发,提供了快速启动...

    spring+springMVC+WebSocket 的学习笔记

    WebSocket 是一种在客户端和服务器之间建立长连接的协议,它允许双方进行双向通信,即服务器可以主动向客户端推送数据。相较于传统的 HTTP 协议,WebSocket 提供更低的延迟和更高的效率,尤其适用于实时通信应用,如...

    spring + websocket demo

    在IT行业中,WebSockets是一种实时通信协议,它允许服务器与客户端进行双向通信,从而实现高效的数据推送。在本示例"spring + websocket demo"中,我们看到的是一个基于Spring框架实现WebSocket功能的简单示例。以下...

    spring boot+websocket

    WebSocket是一种在单个TCP连接上进行全双工通信的协议,允许服务器主动向客户端推送数据,非常适合实时交互应用。 **1. 配置Spring Boot支持WebSocket** 在Spring Boot项目中,我们需要引入`spring-boot-starter-...

    springboot+websocket进度条实战

    在实际应用中,`ProgressService`中的`sendProgressUpdate`方法可能会被业务逻辑调用,例如在执行耗时操作(如文件上传或大数据处理)时,根据任务的进度调用此方法,将进度信息推送到前端,从而在用户界面上动态...

    SpringBoot + Thymeleaf + WebSocket + Spring Security + SpringData-Jpa

    在SpringBoot中,我们可以使用Spring WebSocket库来支持WebSocket通信,实现低延迟、实时的数据推送,例如在线聊天、股票报价等实时应用场景。 4. **Spring Security**: Spring Security是Spring框架的一个子项目...

    Spring boot整合websocket实现即时通讯demo(适合新手或者开发人员学习)

    本项目基于Spring平台,整合websocket协议,实现一个简易web聊天室的功能。主要特性如下: 1.包含聊天室登录、退出的功能。登录时,浏览器自动向服务器发起websocket连接,退出时自动切断。登录页面为...

    如何使用Spring Boot 3.X + WebSocket来实现推送消息功能,并可动态开启、关闭和取消定时任务

    如何使用Postman来发送数据到后端Spring Boot应用程序,并使用WebSocket建立实时通信通道,以便后端应用程序可以向Postman推送消息,并可动态开启、关闭和取消定时任务。 学会这个方法,就能缩短你的项目开发周期,...

    springboot+websocket+redis实现聊天室

    在聊天室场景下,WebSocket使得消息的实时推送成为可能,用户无需频繁刷新页面即可接收到新消息。Spring Boot提供了WebSocket的支持,可以通过`@ServerEndpoint`注解定义WebSocket端点,并使用`Session`对象进行消息...

    使用dwr+spring实现消息推送

    Spring是一个广泛使用的Java应用框架,提供了依赖注入、面向切面编程、事务管理等功能,同时也支持多种集成技术,如WebSocket、HTTP长轮询等,用于实现消息推送。 将DWR与Spring结合,可以创建出高效且易于维护的...

    springboot+netty+websocket+redis

    WebSocket是一种在单个TCP连接上进行全双工通信的协议,允许服务器主动推送数据给客户端,无需频繁的轮询。在Spring Boot中,我们可以利用WebSocket实现即时通讯,用户之间的消息可以实时传递,提高用户体验。Stomp...

Global site tag (gtag.js) - Google Analytics