如何使用WebSocket发送定期消息?

时间:2015-03-17 16:09:56

标签: java multithreading java-ee websocket

我在Tomcat上使用WebSocket(实际的实现是Tyrus,JSR 356的参考实现)。当我必须处理客户端消息并响应它们时,它工作得很好。但是,我想为我的几个客户端控件实现 push 解决方案。实际上我需要两种解决方案:

  • 以特定间隔推送数据,
  • 在系统消息被提升时推出系统消息。

对于第一个,我认为ScheduledExecutorService可以是一个解决方案,我已经有一个或多或少的工作示例,但我有清理问题。对于第二个,我想我需要一个线程,它会在WebSocket端点中触发一个方法,但我也不知道如何干净利落地做到这一点。通过干净,我的意思是我希望只有在连接到我的端点的会话时才能运行线程。

总结我的问题:您将如何正确使用Java EE WebSocket API实现推送消息解决方案?

ps:我更喜欢"纯粹的"解决方案,但Spring也不是不受欢迎的。


当前代码框架

这就是我目前解决第一个问题的方法:

@ServerEndpoint(...)
public class MyEndPoint {
    // own class, abstracting away session handling
    private static SessionHandler sessionHandler = new SessionHandler();
    private static ScheduledExecutorService timer =
            Executors.newSingleThreadScheduledExecutor();
    private static boolean timerStarted = false;

    @OnOpen
    public void onOpen(Session session, EndpointConfig config) {
        sessionHandler.addSession(session);
        if (!timerStarted) {
            timer.scheduleAtFixedRate(new Runnable() {
                @Override
                public void run() {
                    sessionHandler.sendToAllSession("foo");
                }
            }, 0, 3, TimeUnit.SECONDS);
            timerStarted = true;
        }
    }

    @OnClose
    public void onClose(Session session) {
        sessionHandler.removeSession(session);
        if (0 == sessionHandler.countSessions()) {
            // TODO: cleanup thread properly
            timer.shutdown();
            try {
                while (!timer.awaitTermination(10, TimeUnit.SECONDS));
            } catch (InterruptedException e) {
                log.debug("Timer terminated.");
            }
            timerStarted = false;
        }
    }
}

这种方法或多或少有效,但在重新加载几页后,它会以RejectedExecutionException消失,我不太确定如何处理这种情况。

1 个答案:

答案 0 :(得分:1)

不幸的是,在shutdown();

之后你不能使用任何ExecutorService

因此在OnClose()方法之后,下一个OnOpen()方法将崩溃。

只是用于演示的代码很少:

        private void mlkbuykg_TextChanged(object sender, EventArgs e)
        {
            int a,b;
            if (int.TryParse(mlkrate.Text, out a) && int.TryParse(mlkbuykg.Text, out b)){
                int c = a * b;
                mlktotal.Text = c.ToString();
            }
        }

您可以尝试将您的课程也用作网络听众http://docs.oracle.com/javaee/7/api/javax/servlet/annotation/WebListener.html 并在启动和销毁服务器时执行的方法中创建计时器

public class TestThread {

    public static void main(String[] args) {

        final ScheduledExecutorService timer = Executors.newSingleThreadScheduledExecutor();
        boolean timerStarted = false;
        //OnOpen - 1;  - OK
        if (!timerStarted) {
            timer.scheduleAtFixedRate(new Runnable() {
                @Override
                public void run() {
                    System.out.println("foo");
                }
            }, 0, 3, TimeUnit.SECONDS);
            timerStarted = true;
        }

        //OnOpen - 2;  - OK
        if (!timerStarted) {
            timer.scheduleAtFixedRate(new Runnable() {
                @Override
                public void run() {
                    System.out.println("foo");
                }
            }, 0, 3, TimeUnit.SECONDS);
            timerStarted = true;
        }

        //OnClose - 1  - OK
        timer.shutdown();
        timerStarted = false;

        //OnOpen - 2;  - NOT OK, because after stop you can't use timer,  RejectedExecutionException will thrown
        if (!timerStarted) {
            // will crash at this invocke
            timer.scheduleAtFixedRate(new Runnable() {
                @Override
                public void run() {
                    System.out.println("foo");
                }
            }, 0, 3, TimeUnit.SECONDS);
            timerStarted = true;
        }
    }
}