WIP - Support cancelling in flight REST requests

Fixes: #41971

Author: geoand

Diff for: independent-projects/resteasy-reactive/client/runtime/src/main/java/org/jboss/resteasy/reactive/client/handlers/ClientSendRequestHandler.java

@@ -98,6 +98,8 @@ public void handle(RestClientRequestContext requestContext) {
         future.subscribe().with(new Consumer<>() {
             @Override
             public void accept(HttpClientRequest httpClientRequest) {
+                requestContext.setHttpClientRequest(httpClientRequest);
+
                 // adapt headers to HTTP/2 depending on the underlying HTTP connection
                 ClientSendRequestHandler.this.adaptRequest(httpClientRequest);
 

Diff for: independent-projects/resteasy-reactive/client/runtime/src/main/java/org/jboss/resteasy/reactive/client/impl/UniInvoker.java

@@ -1,6 +1,8 @@
 package org.jboss.resteasy.reactive.client.impl;
 
+import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionStage;
+import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Function;
 import java.util.function.Supplier;
 
@@ -13,7 +15,7 @@
 
 public class UniInvoker extends AbstractRxInvoker<Uni<?>> {
 
-    private InvocationBuilderImpl invocationBuilder;
+    private final InvocationBuilderImpl invocationBuilder;
 
     public UniInvoker(InvocationBuilderImpl invocationBuilder) {
         this.invocationBuilder = invocationBuilder;
@@ -22,10 +24,16 @@ public UniInvoker(InvocationBuilderImpl invocationBuilder) {
     @Override
     public <R> Uni<R> method(String name, Entity<?> entity, GenericType<R> responseType) {
         AsyncInvokerImpl invoker = (AsyncInvokerImpl) invocationBuilder.rx();
+        AtomicReference<RestClientRequestContext> restClientRequestContextRef = new AtomicReference<>();
         return Uni.createFrom().completionStage(new Supplier<CompletionStage<R>>() {
             @Override
             public CompletionStage<R> get() {
-                return invoker.method(name, entity, responseType);
+                RestClientRequestContext restClientRequestContext = invoker.performRequestInternal(name, entity,
+                        responseType == null ? new GenericType<>(String.class) : responseType,
+                        true);
+                restClientRequestContextRef.set(restClientRequestContext);
+                CompletableFuture response = restClientRequestContext.getResult();
+                return invoker.mapResponse(response, responseType == null ? String.class : responseType.getRawType());
             }
         }).onFailure().transform(new Function<>() {
             @Override
@@ -35,6 +43,11 @@ public Throwable apply(Throwable t) {
                 }
                 return t;
             }
+        }).onCancellation().invoke(new Runnable() {
+            @Override
+            public void run() {
+                restClientRequestContextRef.get().getHttpClientRequest().reset();
+            }
         });
     }