Support CP for undertow servlet request quarkusio#2981

extensions/smallrye-context-propagation/deployment/src/test/java/io/quarkus/context/test/ContextEndpoint.java

@@ -7,6 +7,7 @@
 import java.util.concurrent.TimeUnit;
 
 import javax.inject.Inject;
+import javax.servlet.http.HttpServletRequest;
 import javax.transaction.Status;
 import javax.transaction.SystemException;
 import javax.transaction.Transaction;
@@ -43,6 +44,8 @@ public class ContextEndpoint {
     ManagedExecutor all;
     @Inject
     ThreadContext allTc;
+    @Inject
+    HttpServletRequest servletRequest;
 
     @GET
     @Path("/resteasy")
@@ -54,6 +57,16 @@ public CompletionStage<String> resteasyTest(@Context UriInfo uriInfo) {
         });
     }
 
+    @GET
+    @Path("/servlet")
+    public CompletionStage<String> servletTest(@Context UriInfo uriInfo) {
+        CompletableFuture<String> ret = all.completedFuture("OK");
+        return ret.thenApplyAsync(text -> {
+            servletRequest.getContentType();
+            return text;
+        });
+    }
+
     @GET
     @Path("/thread-context")
     public CompletionStage<String> threadContextTest(@Context UriInfo uriInfo) {

extensions/smallrye-context-propagation/deployment/src/test/java/io/quarkus/context/test/ContextUnitTest.java

@@ -30,6 +30,12 @@ public void testRESTEasyContextPropagation() {
                 .statusCode(Response.Status.OK.getStatusCode());
     }
 
+    @Test()
+    public void testServletContextPropagation() {
+        RestAssured.when().get("/context/servlet").then()
+                .statusCode(Response.Status.OK.getStatusCode());
+    }
+
     @Test()
     public void testThreadContextPropagation() {
         RestAssured.when().get("/context/thread-context").then()

extensions/undertow/runtime/pom.xml

@@ -37,6 +37,10 @@
             <groupId>io.quarkus</groupId>
             <artifactId>quarkus-core</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.eclipse.microprofile.context-propagation</groupId>
+            <artifactId>microprofile-context-propagation-api</artifactId>
+        </dependency>
     </dependencies>
 
     <build>

extensions/undertow/runtime/src/main/java/io/quarkus/undertow/runtime/ServletThreadContextProvider.java

@@ -0,0 +1,42 @@
+package io.quarkus.undertow.runtime;
+
+import java.util.Map;
+
+import org.eclipse.microprofile.context.spi.ThreadContextProvider;
+import org.eclipse.microprofile.context.spi.ThreadContextSnapshot;
+
+import io.undertow.servlet.handlers.ServletRequestContext;
+
+public class ServletThreadContextProvider implements ThreadContextProvider {
+
+    @Override
+    public ThreadContextSnapshot currentContext(Map<String, String> props) {
+        ServletRequestContext captured = ServletRequestContext.current();
+        return () -> {
+            ServletRequestContext current = restore(captured);
+            return () -> restore(current);
+        };
+    }
+
+    private ServletRequestContext restore(ServletRequestContext context) {
+        ServletRequestContext currentContext = ServletRequestContext.current();
+        if (context == null)
+            ServletRequestContext.clearCurrentServletAttachments();
+        else
+            ServletRequestContext.setCurrentRequestContext(context);
+        return currentContext;
+    }
+
+    @Override
+    public ThreadContextSnapshot clearedContext(Map<String, String> props) {
+        return () -> {
+            ServletRequestContext current = restore(null);
+            return () -> restore(current);
+        };
+    }
+
+    @Override
+    public String getThreadContextType() {
+        return "Servlet";
+    }
+}

extensions/undertow/runtime/src/main/resources/META-INF/services/org.eclipse.microprofile.context.spi.ThreadContextProvider

@@ -0,0 +1 @@
+io.quarkus.undertow.runtime.ServletThreadContextProvider