Inject modified headers

Fixes quarkusio#19317
stuartwdouglas · Aug 11, 2021 · db91a03 · db91a03
1 parent 626b2dc
commit db91a03
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/.../test/java/io/quarkus/resteasy/reactive/server/test/simple/SimpleQuarkusRestResource.java b/.../test/java/io/quarkus/resteasy/reactive/server/test/simple/SimpleQuarkusRestResource.java
@@ -33,6 +33,7 @@
 import javax.ws.rs.core.Response;
 import javax.ws.rs.ext.Providers;
 
+import org.jboss.resteasy.reactive.RestHeader;
 import org.jboss.resteasy.reactive.server.SimpleResourceInfo;
 
 import io.quarkus.runtime.BlockingOperationControl;
@@ -138,8 +139,8 @@ public Response filters(@Context Providers providers) {
 
     @GET
     @Path("filters")
-    public Response filters(@Context HttpHeaders headers) {
-        return Response.ok().header("filter-request", headers.getHeaderString("filter-request")).build();
+    public Response filters(@Context HttpHeaders headers, @RestHeader("filter-request") String header) {
+        return Response.ok().header("filter-request", header).build();
     }
 
     @GET

diff --git a/...src/main/java/org/jboss/resteasy/reactive/server/core/ResteasyReactiveRequestContext.java b/...src/main/java/org/jboss/resteasy/reactive/server/core/ResteasyReactiveRequestContext.java
@@ -10,6 +10,7 @@
 import java.net.URISyntaxException;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.Deque;
 import java.util.LinkedList;
 import java.util.List;
@@ -781,10 +782,22 @@ public void initPathSegments() {
 
     @Override
     public Object getHeader(String name, boolean single) {
-        if (single)
-            return serverRequest().getRequestHeader(name);
-        // empty collections must not be turned to null
-        return serverRequest().getAllRequestHeaders(name);
+        if (httpHeaders == null) {
+            if (single)
+                return serverRequest().getRequestHeader(name);
+            // empty collections must not be turned to null
+            return serverRequest().getAllRequestHeaders(name);
+        } else {
+            if (single)
+                return httpHeaders.getMutableHeaders().getFirst(name);
+            // empty collections must not be turned to null
+            List<String> list = httpHeaders.getMutableHeaders().get(name);
+            if (list == null) {
+                return Collections.emptyList();
+            } else {
+                return list;
+            }
+        }
     }
 
     @Override