使用标准JSR356 WebSocket客户端的连接超时

Question

我有一个连接到远程websocket服务器的Java应用程序。作为客户端，我使用的是标准的Java JSR356 WebSocket API

javax.websocket.WebSocketContainer.connectToServer(...)

但是，我还没有找到使用这个API指定连接超时的方法。当我调用connectToServer(.)方法时，它一直阻塞直到建立连接(这可能永远不会发生)。

是否有一种方法可以使用标准API指定连接超时？如果没有，有什么解决办法吗？

Answer 1

不幸的是，JSR356-JavaAPI WebSocket没有公开这一点。您将需要使用实现特性，如超时 in 泰勒斯 (参考实现)。其他实现很可能有类似的东西。

在等级库中似乎还没有这方面的票据，所以如果您愿意，可以添加一个(我只能找到提到SSL属性的问题-- 规格-210)。

Answer 2

您可以简单地在connectToServer中重写WsWebSocketContainer方法

public class WsWebSocketContainer2 extends WsWebSocketContainer {

    @Override
    public Session connectToServer(Object pojo, URI path) throws DeploymentException {
        ClientEndpoint annotation = pojo.getClass().getAnnotation(ClientEndpoint.class);
        if (annotation == null) {
            throw new DeploymentException("wsWebSocketContainer.missingAnnotation");
        }

        Endpoint ep = new PojoEndpointClient(pojo, Arrays.asList(annotation.decoders()));

        Class<? extends ClientEndpointConfig.Configurator> configuratorClazz = annotation.configurator();

        ClientEndpointConfig.Configurator configurator = null;
        if (!ClientEndpointConfig.Configurator.class.equals(configuratorClazz)) {
            try {
                configurator = configuratorClazz.getConstructor().newInstance();
            } catch (ReflectiveOperationException e) {
                throw new DeploymentException("wsWebSocketContainer.defaultConfiguratorFail", e);
            }
        }

        ClientEndpointConfig.Builder builder = ClientEndpointConfig.Builder.create();
        // Avoid NPE when using RI API JAR - see BZ 56343
        if (configurator != null) {
            builder.configurator(configurator);
        }
        ClientEndpointConfig config = builder.decoders(Arrays.asList(annotation.decoders())).encoders(Arrays.asList(annotation.encoders()))
                .preferredSubprotocols(Arrays.asList(annotation.subprotocols())).build();
        Map<String, Object> userProperties = config.getUserProperties();
        userProperties.put(Constants.IO_TIMEOUT_MS_PROPERTY, 999999);
        return connectToServer(ep, config, path);
    }
}

Answer 3

这件事我自己就搞定了。如果通过未来构造调用connectToServer，则可以在get()方法中使用超时。

您需要一个ThreadPool：

private final ExecutorService pool = Executors.newFixedThreadPool(10);

以下是功能：

private Future<Session> asyncConnectToServer(Object annotatedEndpointInstance, URI uri) {
        return pool.submit(new Callable<Session>() {
            @Override
            public Session call() throws Exception {
                try {
                    WebSocketContainer container = ContainerProvider.getWebSocketContainer();
                    return(container.connectToServer(annotatedEndpointInstance, uri));
                } catch (DeploymentException | IOException | IllegalStateException  e) {
                    //throw new RuntimeException(e);
                    return(null);
                }
            }
        });
    }

这就是你所说的：

 public webSocketClientEndpoint(URI endpointURI, long timeout) {

        final Future<Session> futureSes = asyncConnectToServer(this, endpointURI);

        try {            
            Session ses = futureSes.get(timeout, TimeUnit.MILLISECONDS);
        } catch(InterruptedException | ExecutionException | TimeoutException  e) {
            System.out.println("Time out...");
        }
    }

参考文献：https://dzone.com/articles/javautilconcurrentfuture

https://docs.oracle.com/javase/7/docs/api/java/util/concurrent/Future.html#get(long,%20java.util.concurrent.TimeUnit)