Restore authorization context for live response enforcement in Concie…

…rge. Signed-off-by: Yufei Cai <yufei.cai@bosch.io>
eclipse-ditto · Nov 15, 2021 · 56bca4b · 56bca4b
1 parent 19aaa47
commit 56bca4b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...main/java/org/eclipse/ditto/concierge/service/enforcement/AbstractEnforcementWithAsk.java b/...main/java/org/eclipse/ditto/concierge/service/enforcement/AbstractEnforcementWithAsk.java
@@ -73,6 +73,7 @@ protected CompletionStage<R> askAndBuildJsonView(
                 .thenApply(response -> filterJsonView(response, enforcer));
     }
 
+    @SuppressWarnings("unchecked")
     protected CompletionStage<R> askAndBuildJsonView(
             final ActorRef actorToAsk,
             final C commandWithReadSubjects,
@@ -82,7 +83,12 @@ protected CompletionStage<R> askAndBuildJsonView(
             final Executor executor) {
 
         return ask(actorToAsk, commandWithReadSubjects, publish, "before building JsonView", scheduler, executor)
-                .thenApply(response -> filterJsonView(response, enforcer));
+                .thenApply(response -> filterJsonView((R) response.setDittoHeaders(
+                        response.getDittoHeaders()
+                                .toBuilder()
+                                .authorizationContext(commandWithReadSubjects.getDittoHeaders().getAuthorizationContext())
+                                .build()
+                ), enforcer));
     }
 
     /**