WebSocket实现前后端消息推送

这篇具有很好参考价值的文章主要介绍了WebSocket实现前后端消息推送。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

WebSocket的代码编写会根据业务逻辑而进行变化,需要去理解编写思路,这样才能在工作中使用得游刃有余。

1. 引入依赖

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

2.  编写WebSocketConfig配置类

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {

    @Bean
    public ServerEndpointExporter getServerEndpointExporter() {
        return new ServerEndpointExporter();
    }

}

3. 编写WebSocket服务类

下面的服务类中,可以编写多个sendMeg方法(写法比较多样化),作用是发送消息回前端,使用方式就是你在自己的业务代码中自行调用,例(serviceImpl中调用):文章来源地址https://www.toymoban.com/news/detail-508119.html

WebSocketServer.sendCountToUaa(ownerId, unreadNewsCount);
import com.sms.service.InternalNewsService;
import com.sms.service.impl.InternalNewsServiceImpl;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;

import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Websocket服务,推送信息
 */

@Slf4j
@Component
@ServerEndpoint("/webSocket/sms")
public class WebSocketServer {

    /**
     * 定义集合,用于存储连接信息
     * 根据业务需求自定义,数据格式可随意变化
     */
    //浏览器端连接,这种数据格式是因为考虑到同一个用户账号可以在不同的浏览器登录,所以一个用户需要保存多个连接
    private static final ConcurrentHashMap<String, ArrayList<Session>> sessionMap = new ConcurrentHashMap<>();
    //微信端连接
    private static final List<Session> wxSessionList = new ArrayList<>();

    /**
     * 作用:
     * 这里使用@Autowared是无法正常注入对象的,需要注入外部对象就需要使用这种方式
     * 需在启动类也就是main方法中加入:WebSocketServer.setApplicationContext(context);
     * context参数是启动类中run方法的返回值
     * 使用:
     * 在需要对象的地方使用
     * 例:
     * UserServiceImpl userServiceImpl = applicationContext.getBean(UserService.class);
     */
    private static ApplicationContext applicationContext;

    public static void setApplicationContext(ApplicationContext applicationContext) {
        WebSocketServer.applicationContext = applicationContext;
    }

    /**
     * 前端关闭页面或者主动关闭websocket链接,都会执行@OnClose
     */
    @OnClose
    public void close(Session session) {
        if (null == session) {
            return;
        }
        /**
         * 以下代码都是根据业务逻辑编写,写法并不固定
         */
        //关闭微信端中的session
        ListIterator<Session> sessionListIterator = wxSessionList.listIterator();
        while (sessionListIterator.hasNext()) {
            Session next = sessionListIterator.next();
            if (session == next) {
                sessionListIterator.remove();
                return;
            }
        }
        //关闭浏览器端中的session
        Iterator<Map.Entry<String, ArrayList<Session>>> iterator = sessionMap.entrySet().iterator();
        while (iterator.hasNext()) {
            ArrayList<Session> values = iterator.next().getValue();
            if (null == values || 1 > values.size()) {
                continue;
            }
            for (int i = 0; i < values.size(); i++) {
                Session sess = values.get(i);
                if (null == sess) {
                    continue;
                }
                if (session == sess) {
                    values.remove(i--);
                    return;
                }
            }
        }
    }

    /**
     * 前端连接后端socket时执行@OnMessage,前后端建立起连接,后端保存连接
     */
    @OnMessage
    public void onMessage(String userId, Session session) {
        if (StringUtils.isEmpty(userId) || null == session) {
            return;
        }
        if ("ykj".equals(userId)) {
            //微信端消息标识,自定义,没有微信端可直接将这个if删掉
            wxSessionList.add(session);
        } else {
            //接收到消息后,找到对应的session
            ArrayList<Session> sessions = sessionMap.get(userId);
            if (null == sessions) {
                sessions = new ArrayList<>();
            } else {
                //遍历,看该session是否存在,如果存在代表是在心跳检测操作
                for (int i = 0; i < sessions.size(); i++) {
                    if (sessions.get(i) == session) {
                        try {
                            //发送消息回前端
                            session.getAsyncRemote().sendText("{\"heartbeat\":\"socket心跳检测成功!!!\"}");
                            return;
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
                }
            }
            //说明session不存在,添加到列表
            sessions.add(session);
            sessionMap.put(userId, sessions);

            //查询当前用户未读消息条数(业务代码,查询数据库)
            InternalNewsService internalNewsService = applicationContext.getBean(InternalNewsService.class);
            Integer newsCount = internalNewsService.selectUnreadNewsCount(userId, 0, 4);

            if (newsCount == null) {
                session.getAsyncRemote().sendText("0");//发送消息回前端
            } else {
                session.getAsyncRemote().sendText(newsCount.toString());//发送消息回前端
            }
            return;
        }
        try {
            session.getAsyncRemote().sendText("200");//发送消息回前端
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 发送数据到前端
     *
     * @param msg
     */
    public static void sendMsg(String userIds, String msg) {
        if (StringUtils.isEmpty(userIds) || StringUtils.isEmpty(msg)) {
            return;
        }
        String[] user_id = userIds.split(",");
        int uis = user_id.length;
        if (1 > uis) {
            return;
        }
        for (int ii = 0; ii < uis; ii++) {
            String userId = user_id[ii];
            if (StringUtils.isEmpty(userId)) {
                continue;
            }
            ArrayList<Session> sessions = sessionMap.get(userId);
            if (null == sessions || 1 > sessions.size()) {
                continue;
            }
            Iterator<Session> iterator = sessions.iterator();
            while (iterator.hasNext()) {
                Session next = iterator.next();
                if (null == next || !next.isOpen()) {
                    continue;
                }
                try {
                    next.getBasicRemote().sendText(msg);
                    //System.out.println("推送给socket:" + next);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 发送数据到前端uaa端
     */
    public static void sendCountToUaa(String userId, Integer count) {
        if (StringUtils.isEmpty(userId)) {
            return;
        }
        ArrayList<Session> sessions = sessionMap.get(userId);
        if (null == sessions || 1 > sessions.size()) {
            return;
        }
        Iterator<Session> iterator = sessions.iterator();
        while (iterator.hasNext()) {
            Session next = iterator.next();
            if (null == next || !next.isOpen()) {
                continue;
            }
            try {
                //next.getBasicRemote().sendText(count.toString());
                next.getAsyncRemote().sendText(count.toString());
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 给微信端用户推送消息
     *
     * @param msg
     */
    public static void sendToVXMsg(String msg) {
        ListIterator<Session> sessionListIterator = wxSessionList.listIterator();
        while (sessionListIterator.hasNext()) {
            Session session = sessionListIterator.next();
            try {
                session.getBasicRemote().sendText(msg);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

}

到了这里,关于WebSocket实现前后端消息推送的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • SpringBoot+Netty+Websocket实现消息推送

    这样一个需求:把设备异常的状态每10秒推送到页面并且以弹窗弹出来,这个时候用Websocket最为合适,今天主要是后端代码展示。 添加依赖 定义netty端口号 netty服务器 Netty配置 管理全局Channel以及用户对应的channel(推送消息) 管道配置 自定义CustomChannelHandler 推送消息接口及

    2024年02月04日
    浏览(49)
  • WebSocket+Redis实现消息推送机制以及离线消息推送(vue+sping boot)

    vue端涉及业务就不贴了 WebSocket 是一种在单个TCP连接上进行全双工通信的协议。WebSocket通信协议于2011年被IETF定为标准RFC 6455,并由RFC7936补充规范。WebSocket API也被W3C定为标准。 WebSocket使得客户端和服务器之间的数据交换变得更加简单,允许服务端主动向客户端推送数据。在

    2024年02月09日
    浏览(51)
  • SpringBoot整合Netty+Websocket实现消息推送

           Netty是一个高性能、异步事件驱动的网络应用框架,用于快速开发可维护的高性能协议服务器和客户端。以下是Netty的主要优势: 高性能 :Netty基于NIO(非阻塞IO)模型,采用事件驱动的设计,具有高性能的特点。它通过零拷贝技术、内存池化技术等手段,进一步提高

    2024年01月20日
    浏览(45)
  • Spring Boot集成WebSocket实现消息推送

    项目中经常会用到消息推送功能,关于推送技术的实现,我们通常会联想到轮询、comet长连接技术,虽然这些技术能够实现,但是需要反复连接,对于服务资源消耗过大,随着技术的发展,HtML5定义了WebSocket协议,能更好的节省服务器资源和带宽,并且能够更实时地进行通讯。

    2023年04月08日
    浏览(45)
  • Vue使用WebSocket实现实时获取后端推送的数据。

    Vue可以使用WebSocket实现实时获取后端推送的数据。 1.在Vue项目中安装WebSocket库 可以使用npm或yarn安装WebSocket库: 2.创建WebSocket连接 在Vue组件中创建WebSocket连接,连接到后端WebSocket服务器,代码如下: 上面的代码中,使用WebSocket连接到后端WebSocket服务器,通过监听onmessage事件,

    2024年02月08日
    浏览(50)
  • 分布式WebSocket消息推送系统设计与实现

    作者:禅与计算机程序设计艺术 现如今,随着物联网、云计算、移动互联网、大数据等新技术的兴起,分布式系统成为越来越多企业面临的挑战。在分布式系统中,服务间通信是一个重要且复杂的课题,基于TCP/IP协议族的传输层协议之上的应用层协议比如HTTP协议、RPC(Remo

    2024年02月05日
    浏览(41)
  • Springboot整合WebSocket实现主动向前端推送消息

            在上篇文章tcp编程中,我们实现了C++客户端与java服务器之间的通信,客户端发送了一个消息给服务器,今天我们要实现基于WebSocket实现服务器主动向前端推送消息,并且以服务器接收到C++客户端的消息主动向前端推送消息的触发条件。 WebSocket 的诞生背景       

    2024年03月16日
    浏览(43)
  • 记录--你还在使用websocket实现实时消息推送吗?

    在日常的开发中,我们经常能碰见服务端需要主动推送给客户端数据的业务场景,比如数据大屏的实时数据,比如消息中心的未读消息,比如聊天功能等等。 本文主要介绍SSE的使用场景和如何使用SSE。 我们常规实现这些需求的方案有以下三种 轮询 websocket SSE 在很久很久以前

    2024年02月19日
    浏览(39)
  • Springboot集成websocket实现消息推送和在线用户统计

    在启动类上添加一个bean 核心代码 实现消息推送只要在业务代码中调用sendMessageSpecial()方法即可。 然后调用刚才的业务接口测试:http://localhost:8080/websocket/t1 调用成功后可以看到三个窗口中都收到了消息

    2023年04月08日
    浏览(55)
  • Golang 搭建 WebSocket 应用(三) - 实现一个消息推送中心

    有了前两篇的铺垫,相信大家已经对 Golang 中 WebSocket 的使用有一定的了解了, 今天我们以一个更加真实的例子来学习如何在 Golang 中使用 WebSocket 。 在实际的项目中,往往有一些任务耗时比较长,然后我们会把这些任务做异步的处理,但是又要及时给客户端反馈任务的处理进

    2024年01月23日
    浏览(40)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包