foo

src/main/java/io/vertx/core/http/impl/HttpServerRequestImpl.java

@@ -153,13 +153,13 @@ void handleBegin() {
       check100();
     }
     VertxThread current = VertxThread.current();
-    ContextInternal prev = current.beginDispatch(context);
+    current.beginDispatch(context);
     try {
       conn.requestHandler.handle(this);
     } catch(Throwable t) {
       context.reportException(t);
     } finally {
-      current.endDispatch(prev);
+      current.endDispatch();
     }
   }
 

src/main/java/io/vertx/core/http/impl/HttpServerResponseImpl.java

@@ -342,10 +342,10 @@ public void end(Buffer chunk) {
       written = true;
       conn.responseComplete();
       if (bodyEndHandler != null) {
-        context.dispatch(bodyEndHandler);
+        bodyEndHandler.handle(null);
       }
       if (endHandler != null) {
-        context.dispatch(endHandler);
+        endHandler.handle(null);
       }
     }
   }

src/main/java/io/vertx/core/impl/AbstractContext.java

@@ -73,13 +73,13 @@ public final void dispatch(Handler<Void> task) {
   @Override
   public final <T> void dispatch(T arg, Handler<T> task) {
     VertxThread currentThread = VertxThread.current();
-    ContextInternal prev = currentThread.beginDispatch(this);
+    currentThread.beginDispatch(this);
     try {
       task.handle(arg);
     } catch (Throwable t) {
       reportException(t);
     } finally {
-      currentThread.endDispatch(prev);
+      currentThread.endDispatch();
     }
   }
 

src/main/java/io/vertx/core/impl/VertxThread.java

@@ -92,34 +92,32 @@ public TimeUnit getMaxExecTimeUnit() {
    * shall be used.
    *
    * @param context the context on which the task is dispatched on
-   * @return the current context that shall be restored
    */
-  public ContextInternal beginDispatch(ContextInternal context) {
+  public void beginDispatch(ContextInternal context) {
+    if (this.context != null) {
+      throw new IllegalStateException("Already dispatching");
+    }
     if (!ContextImpl.DISABLE_TIMINGS) {
       executeStart();
     }
     if (!DISABLE_TCCL) {
       setContextClassLoader(context != null ? context.classLoader() : null);
     }
-    ContextInternal prev = this.context;
     this.context = context;
-    return prev;
   }
 
   /**
    * End the dispatch of a context task.
    * <p>
    * This is a low level interface that should not be used, instead {@link ContextInternal#dispatch(Object, io.vertx.core.Handler)}
    * shall be used.
-   *
-   * @param context the previous context thread to restore, might be {@code null}
    */
-  public void endDispatch(ContextInternal context) {
+  public void endDispatch() {
     // We don't unset the context after execution - this is done later when the context is closed via
     // VertxThreadFactory
-    this.context = context;
+    this.context = null;
     if (!DISABLE_TCCL) {
-      setContextClassLoader(context != null ? context.classLoader() : null);
+      setContextClassLoader(null);
     }
     if (!ContextImpl.DISABLE_TIMINGS) {
       executeEnd();

src/test/java/io/vertx/core/http/HttpTest.java

@@ -4662,12 +4662,12 @@ public void testHttpInvalidConnectResponseChunked() {
 
   @Test
   public void testEndFromAnotherThread() throws Exception {
-
-    waitFor(2);
-
-    // Same with pipelining
-
+    waitFor(3);
+    disableThreadChecks();
     server.requestHandler(req -> {
+      req.response().bodyEndHandler(v -> {
+        complete();
+      });
       req.response().endHandler(v -> {
         complete();
       });
@@ -4680,7 +4680,6 @@ public void testEndFromAnotherThread() throws Exception {
       assertEquals(200, resp.statusCode());
       complete();
     }));
-
     await();
   }