Play Framework 2.5 JavaAsync 抛出 CompletionException

Question

我正在使用 Play 2.5 构建一个简单的应用程序。为了获得更好的性能，我将 Akka 分块响应与 Java 8 CompletionStage 策略一起使用。下面是生成分块响应的代码（不使用 ComperableFuture 时它工作正常）：

@Singleton
public class AbstractSource {

    public Source<ByteString, ?> getChunked(String html) {

        return Source.<ByteString>actorRef(256, OverflowStrategy.dropNew())
                .mapMaterializedValue(sourceActor -> {
                    sourceActor.tell(ByteString.fromString(html), null);
                    sourceActor.tell(new Status.Success(NotUsed.getInstance()), null);
                    return null;
                });

    }

}

这是我的控制器：

@Singleton
@AddCSRFToken
public class Application extends Controller {

    @Inject
    private AbstractSource abstractSource;

    public CompletionStage<Result> index() {


        CompletionStage<Source<ByteString, ?>> source = CompletableFuture.supplyAsync(() -> 
                                                  abstractSource.getChunked(index.render(CSRF.getToken(request()).map(t -> 
                                                    t.value()).orElse("no token")).body()
                                                   )
                                                );

        return source.thenApply( chunks -> ok().chunked(chunks));

    }

}

现在，当我运行应用程序时，它会引发以下异常：

play.api.http.HttpErrorHandlerExceptions$$anon$1: Execution exception[[CompletionException: java.lang.RuntimeException: There is no HTTP Context available from here.]]
    at play.api.http.HttpErrorHandlerExceptions$.throwableToUsefulException(HttpErrorHandler.scala:269)
    at play.api.http.DefaultHttpErrorHandler.onServerError(HttpErrorHandler.scala:195)
    at play.api.GlobalSettings$class.onError(GlobalSettings.scala:160)
    at play.api.DefaultGlobal$.onError(GlobalSettings.scala:188)
    at play.api.http.GlobalSettingsHttpErrorHandler.onServerError(HttpErrorHandler.scala:98)
    at play.core.server.netty.PlayRequestHandler$$anonfun$2$$anonfun$apply$1.applyOrElse(PlayRequestHandler.scala:99)
    at play.core.server.netty.PlayRequestHandler$$anonfun$2$$anonfun$apply$1.applyOrElse(PlayRequestHandler.scala:98)
    at scala.concurrent.Future$$anonfun$recoverWith$1.apply(Future.scala:344)
    at scala.concurrent.Future$$anonfun$recoverWith$1.apply(Future.scala:343)
    at scala.concurrent.impl.CallbackRunnable.run(Promise.scala:32)
Caused by: java.util.concurrent.CompletionException: java.lang.RuntimeException: There is no HTTP Context available from here.
    at java.util.concurrent.CompletableFuture.encodeThrowable(CompletableFuture.java:273)
    at java.util.concurrent.CompletableFuture.completeThrowable(CompletableFuture.java:280)
    at java.util.concurrent.CompletableFuture$AsyncSupply.run(CompletableFuture.java:1592)
    at java.util.concurrent.CompletableFuture$AsyncSupply.exec(CompletableFuture.java:1582)
    at java.util.concurrent.ForkJoinTask.doExec(ForkJoinTask.java:289)
    at java.util.concurrent.ForkJoinPool$WorkQueue.runTask(ForkJoinPool.java:1056)
    at java.util.concurrent.ForkJoinPool.runWorker(ForkJoinPool.java:1692)
    at java.util.concurrent.ForkJoinWorkerThread.run(ForkJoinWorkerThread.java:157)
Caused by: java.lang.RuntimeException: There is no HTTP Context available from here.
    at play.mvc.Http$Context.current(Http.java:57)
    at play.mvc.Controller.request(Controller.java:36)
    at com.mabsisa.ui.web.controllers.Application.lambda$index$1(Application.java:31)
    at java.util.concurrent.CompletableFuture$AsyncSupply.run(CompletableFuture.java:1590)
    ... 5 common frames omitted

我没有在任何地方使用 HTTP 上下文，所以为什么这不起作用我不明白。当返回带有分块响应的正常结果时，相同的代码正在工作。请帮忙解决这个问题

Answer 1

您必须在处理CompletableFuture / CompletionStage 时提供HTTP 执行上下文。在 Scala 中，上下文信息通过隐式传递，这些在 Java 中不可用 - 这就是 Play 使用 ThreadLocal 的原因。

但是，您在切换线程时可能会丢失此信息，这就是您遇到问题的原因。您可能认为您不访问 HTTP 上下文，但实际上您访问了 - 您正在使用 request()。

因此，您必须更改代码以将 supplyAsync 与 Executor 一起使用：

https://docs.oracle.com/javase/8/docs/api/java/util/concurrent/CompletableFuture.html#supplyAsync-java.util.function.Supplier-java.util.concurrent.Executor-

从这里：

CompletableFuture.supplyAsync(() -> abstractSource.getChunked(index.render(CSRF.getToken(request()).map(t -> 
                                                    t.value()).orElse("no token")).body()
                                                   )
                                                );

到这里：

CompletableFuture.supplyAsync(() -> abstractSource.getChunked(index.render(CSRF.getToken(request()).map(t -> 
                                                    t.value()).orElse("no token")).body()
                                                   )
                                                , ec.current());

ec 是您的上下文：@Inject HttpExecutionContext ec;

Answer 2

我补充了安东的回答。

如果您正在使用 Play Java API 构建非阻塞应用程序，每次需要调用 CompletionStage 上的方法时，注入 HttpExecutionContext 并传递 ec.current()) 可能会变得非常麻烦。

为了让生活更轻松，您可以使用装饰器，它将保留调用之间的上下文。

public class ContextPreservingCompletionStage<T> implements CompletionStage<T> {

    private HttpExecutionContext context;
    private CompletionStage<T> delegate;

    public ContextPreservingCompletionStage(CompletionStage<T> delegate,
                                            HttpExecutionContext context) {
        this.delegate = delegate;
        this.context = context;
    }
    ...
}

所以你只需要传递一次上下文：

return new ContextPreservingCompletionStage<>(someCompletableFuture, context)
                            .thenCompose(something -> {...});
                            .thenApply(something -> {...});

代替

return someCompletableFuture.thenComposeAsync(something -> {...}, context.current())
                            .thenApplyAsync(something -> {...}, context.current());

如果您正在构建一个多层应用程序并在不同类之间传递CompletionStages，这将特别有用。

完整的装饰器实现示例is here.