RESTEasy Reactive thread fixes

Relates to quarkusio#18977 There are still issues though, as the vert.x pool is not enforcing affinity.
stuartwdouglas · Aug 5, 2021 · eb009f3 · eb009f3
1 parent bf3fdd0
commit eb009f3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/...e/src/main/java/org/jboss/resteasy/reactive/client/handlers/ClientSendRequestHandler.java b/...e/src/main/java/org/jboss/resteasy/reactive/client/handlers/ClientSendRequestHandler.java
@@ -59,6 +59,7 @@ public void handle(Throwable event) {
         future.onSuccess(new Handler<HttpClientRequest>() {
             @Override
             public void handle(HttpClientRequest httpClientRequest) {
+                requestContext.setHttpClientRequest(httpClientRequest);
                 Future<HttpClientResponse> sent;
                 if (requestContext.isMultipart()) {
                     Promise<HttpClientRequest> requestPromise = Promise.promise();

diff --git a/...ntime/src/main/java/org/jboss/resteasy/reactive/client/impl/RestClientRequestContext.java b/...ntime/src/main/java/org/jboss/resteasy/reactive/client/impl/RestClientRequestContext.java
@@ -6,6 +6,8 @@
 import io.vertx.core.http.HttpClient;
 import io.vertx.core.http.HttpClientRequest;
 import io.vertx.core.http.HttpClientResponse;
+import io.vertx.core.impl.ContextInternal;
+import io.vertx.core.net.impl.ConnectionBase;
 import io.vertx.ext.web.multipart.MultipartForm;
 import java.io.IOException;
 import java.io.InputStream;
@@ -154,6 +156,17 @@ public ClientResponseContextImpl getOrCreateClientResponseContext() {
         return clientResponseContext;
     }
 
+    @Override
+    protected Executor getContextExecutor() {
+        if (httpClientRequest != null) {
+            return ((ConnectionBase) httpClientRequest.connection()).getContext().nettyEventLoop();
+        }
+        if (Context.isOnEventLoopThread()) {
+            return ((ContextInternal) restClient.getVertx().getOrCreateContext()).nettyEventLoop();
+        }
+        return null;
+    }
+
     public ClientRequestContextImpl getOrCreateClientRequestContext() {
         if (clientRequestContext == null) {
             clientRequestContext = new ClientRequestContextImpl(this, this.getRestClient(), this.getConfiguration());