Как проверить, живо ли соединение с Websocket - PullRequest
Новая
       56

Как проверить, живо ли соединение с Websocket

0 голосов
/ 01 мая 2018

У меня подключение к серверу через веб-сокет: 

import javax.websocket.*;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;

@ClientEndpoint
public class WebsocketExample {

    private Session userSession;

    private void connect() {

        try {
            WebSocketContainer container = ContainerProvider.getWebSocketContainer();
            container.connectToServer(this, new URI("someaddress"));
        } catch (DeploymentException | URISyntaxException | IOException e) {
            e.printStackTrace();
        }
    }

    @OnOpen
    public void onOpen(Session userSession) {
        // Set the user session
        this.userSession = userSession;
        System.out.println("Open");
    }

    @OnClose
    public void onClose(Session userSession, CloseReason reason) {
        this.userSession = null;
        System.out.println("Close");
    }

    @OnMessage
    public void onMessage(String message) {
        // Do something with the message
        System.out.println(message);
    }
}

Через некоторое время кажется, что я больше не получаю сообщений от сервера, но метод onClose не был вызван.

Я хотел бы иметь своего рода таймер, который, по крайней мере, регистрировал бы ошибку (и, в лучшем случае, пытался восстановить соединение), если бы я не получил, например, сообщения в течение последних пяти минут. Таймер будет сброшен при получении нового сообщения.

Как я могу это сделать?

Ответы [ 2 ]

0 голосов
/ 12 мая 2018

Вы должны обойти эту проблему, внедрив систему сердцебиения, одна сторона которой отправляет ping, а другая отвечает pong. Почти каждый клиент и сервер веб-сокета (насколько я знаю) поддерживают эту функцию внутренне. Эти кадры для пинг-понга можно отправлять с обеих сторон. Я обычно реализую это на стороне сервера, потому что обычно знаю, что у него больше шансов остаться в живых, чем у клиентов (мое мнение). Если клиенты не отправляют обратно понг в течение долгого времени, я знаю, что связь разорвана. На стороне клиента я проверяю то же самое: если сервер долгое время не отправлял сообщения ping, я знаю, что соединение разорвано.

Если ping / pong не реализован в используемых вами библиотеках (которые, я думаю, есть в javax websocket), вы можете создать для этого собственный протокол.

0 голосов
/ 01 мая 2018

Вот что я сделал. Я изменил javax.websocket на Jetty и реализовал пинг-вызов: 

import org.eclipse.jetty.util.ssl.SslContextFactory;
import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
import org.eclipse.jetty.websocket.api.annotations.WebSocket;
import org.eclipse.jetty.websocket.client.WebSocketClient;

import java.io.IOException;
import java.net.URI;
import java.nio.ByteBuffer;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

@WebSocket
public class WebsocketExample {

    private Session userSession;
    private final ScheduledExecutorService executorService = Executors.newScheduledThreadPool(1);

    private void connect() {
        try {
            SslContextFactory sslContextFactory = new SslContextFactory();
            WebSocketClient client = new WebSocketClient(sslContextFactory);
            client.start();
            client.connect(this, new URI("Someaddress"));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @OnWebSocketConnect
    public void onOpen(Session userSession) {
        // Set the user session
        this.userSession = userSession;
        System.out.println("Open");

        executorService.scheduleAtFixedRate(() -> {
                    try {
                        String data = "Ping";
                        ByteBuffer payload = ByteBuffer.wrap(data.getBytes());
                        userSession.getRemote().sendPing(payload);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                },
                5, 5, TimeUnit.MINUTES);
    }

    @OnWebSocketClose
    public void onClose(int code, String reason) {
        this.userSession = null;
        System.out.println("Close");
    }

    @OnWebSocketMessage
    public void onMessage(String message) {
        // Do something with the message
        System.out.println(message);
    }
}

Редактировать : Это просто пример пинга ... Я не знаю, все ли серверы должны отвечать понгом ...

Edit2 : Вот как поступить с сообщением понг. Хитрость заключалась не в том, чтобы слушать сообщения String, а в сообщения Frame: 

@OnWebSocketFrame
@SuppressWarnings("unused")
public void onFrame(Frame pong) {
    if (pong instanceof PongFrame) {
        lastPong = Instant.now();
    }
}

Чтобы управлять временем ожидания сервера, я изменил запланированное задание следующим образом: 

scheduledFutures.add(executorService.scheduleAtFixedRate(() -> {
                    try {
                        String data = "Ping";
                        ByteBuffer payload = ByteBuffer.wrap(data.getBytes());
                        userSession.getRemote().sendPing(payload);

                        if (lastPong != null
                                && Instant.now().getEpochSecond() - lastPong.getEpochSecond() > 60) {
                            userSession.close(1000, "Timeout manually closing dead connection.");
                        }

                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                },
                10, 10, TimeUnit.SECONDS));

... и обработать переподключение в методе onClose

...