使用spring事件完成消息推送功能

需求是查询数据库 ,有无用户未读的消息,发消息通知前端。


第一版:

使用订阅者模式,一个线程轮询数据库是否有新消息,有则放入session属性中;一个线程拉,一直轮询这个session属性,有则通知前端。

  拉 :

 @OnOpen
    public String open(Session session) {
        Cache cache = cacheManager.getCache(EhCacheName.CONST);

        openSessions.add(session);if (session.getUserProperties().get("eventQueue")==null) {
            session.getUserProperties().put("eventQueue", new LinkedBlockingQueue());
        }
        if ( cache.get(session.getId() + "sentResponseForm")==null){
            cache.put(session.getId() + "sentResponseForm",new ArrayList<>());
        }


        if (!isEventOpen) {
        Thread eventQueuePoller = new Thread(() -> {
            //每一用户单独队列
            LinkedBlockingQueue queue = (LinkedBlockingQueue) session.getUserProperties().get("eventQueue");
            while (true) {
                ResponseForm responseForm = (ResponseForm) queue.poll();
                if (responseForm != null) {
                    session.getAsyncRemote().sendText(JSON.toJSONString(responseForm));
                }
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        eventQueuePoller.setName("eventQueuePoller");
        eventQueuePoller.start();

        //判断
            new MessageListener(session, messageService, cacheManager).listen();
            isEventOpen = true;
        }

        return "open";
    }

  其中@OnMessage是J2EE websocket注解,ResponseForm 是自定义的数据格式 。

 

  推:轮询数据库,往队列里添加ResponseForm 。


功能虽然能完成,可是存在问题:1.有新消息就一直往前端推送,直到消息被阅读,影响用户体验。2.无关代码放在一个功能里

解决方法:

第一个问题,使用缓存比对,即查询出来的和 缓存中比对,有新的才通知。

 

第二个问题,使用spring自带的消息通知,其实在spring启动过程中已经使用过了,只是不知道。

推的线程不变,但是不需要用户队列,取而代之得是

ApplicationContext.publishEvent(event);

然后所有实现了EventListener的类都会得到消息通知。


从实现的角度 来说,区别就在于我通过一个队列订阅消息,可以并发往队列里加消息,而 spring的实现可以无需代码对多个订阅者通知。

参考文章:https://jinnianshilongnian.iteye.com/blog/1902886

 

posted @ 2019-05-07 20:33  陈跃进同志  阅读(2975)  评论(1编辑  收藏  举报