首页 > 解决方案 > PlayWS - 达到请求超时时如何抛出异常?

问题描述

当使用 PlayWS 执行长时间下载时CompletableFuture,这些有时会达到定义的请求超时。发生这种情况时,PlayWS 似乎没有抛出异常(至少在我的配置中),因此下载不能标记为失败,尽管数据已损坏,但仍会进行处理。

请原谅这种可憎的代码:

final CompletionStage<WSResponse> futureResponse = this.playWS.client
        .url(importSource.getDownloadUrl())
        .setMethod(HttpMethod.GET)
        .setRequestTimeout(Duration.ofSeconds(5)) // When the timeout is reached, the download gets canceled
        .stream();

try {
    futureResponse
            .thenAccept(res -> {
                try (OutputStream outputStream = Files.newOutputStream(file.toPath())) {
                    final Source<ByteString, ?> responseBody = res.getBodyAsSource();
                    final Sink<ByteString, CompletionStage<Done>> outputWriter =
                            Sink.foreach(bytes -> {
                                outputStream.write(bytes.toArray());
                            });
                    responseBody
                            .runWith(outputWriter, this.playWS.materializer)
                            .whenComplete((value, error) -> {
                                System.out.println("VALUE: "+value); // == "Done"
                                System.out.println("Error: "+error); // == null
                            })
                            .exceptionally(exception -> {
                                throw new IllegalStateException("Download failed for: " + importSource.getDownloadUrl(), exception);
                            })
                            .toCompletableFuture().join();
                } catch (final IOException e) {
                    throw new IllegalStateException("Couldn't open or write to OutputStream.", e);
                }
            })
            .exceptionally(exception -> {
                throw new IllegalStateException("Download failed for: " + importSource.getDownloadUrl(), exception);
            })
            .toCompletableFuture().get();
} catch (InterruptedException | ExecutionException e) {
    throw new IllegalStateException("Couldn't complete CompletableFuture.", e);
}

我是在做一些根本错误的事情还是这是一个错误?

我看到的唯一解决方案是:

  1. 计算接收到的字节并将它们与Content-Length header.
  2. 将请求超时设置为 -1(无限期)。

感谢您的任何建议。

标签: javaplayframeworkjava-8

解决方案


我认为有些过于复杂了。

您可以从Future或附加CompletableStageSourceCompletableFutureAkka Streams 的 API 比(我认为)更强大

final CompletionStage<WSResponse> futureResponse = this.playWS.client
        .url(importSource.getDownloadUrl())
        .setMethod(HttpMethod.GET)
        .setRequestTimeout(Duration.ofSeconds(5)) 
        .stream();

Source<WSResponse> source = Source.fromCompletableStage(futureResponse);
source.map(...).filter(...).recover(...).runforeach(..., playWS.materializer)

推荐阅读