Browse Source

Use headerFilter on SimpAnnotationMethodMessageHandler

See gh-36179
pull/36273/head
rstoyanchev 1 month ago
parent
commit
cb4b87cce2
  1. 9
      spring-messaging/src/main/java/org/springframework/messaging/simp/annotation/support/SimpAnnotationMethodMessageHandler.java

9
spring-messaging/src/main/java/org/springframework/messaging/simp/annotation/support/SimpAnnotationMethodMessageHandler.java

@ -372,11 +372,17 @@ public class SimpAnnotationMethodMessageHandler extends AbstractMethodMessageHan @@ -372,11 +372,17 @@ public class SimpAnnotationMethodMessageHandler extends AbstractMethodMessageHan
SendToMethodReturnValueHandler sendToHandler =
new SendToMethodReturnValueHandler(this.brokerTemplate, true);
sendToHandler.setHeaderInitializer(this.headerInitializer);
if (this.headerFilter != null) {
sendToHandler.addHeaderFilter(this.headerFilter);
}
handlers.add(sendToHandler);
SubscriptionMethodReturnValueHandler subscriptionHandler =
new SubscriptionMethodReturnValueHandler(this.clientMessagingTemplate);
subscriptionHandler.setHeaderInitializer(this.headerInitializer);
if (this.headerFilter != null) {
subscriptionHandler.addHeaderFilter(this.headerFilter);
}
handlers.add(subscriptionHandler);
// Custom return value types
@ -387,6 +393,9 @@ public class SimpAnnotationMethodMessageHandler extends AbstractMethodMessageHan @@ -387,6 +393,9 @@ public class SimpAnnotationMethodMessageHandler extends AbstractMethodMessageHan
sendToHandler = new SendToMethodReturnValueHandler(this.brokerTemplate, false);
sendToHandler.setHeaderInitializer(this.headerInitializer);
if (this.headerFilter != null) {
sendToHandler.addHeaderFilter(this.headerFilter);
}
handlers.add(sendToHandler);
return handlers;

Loading…
Cancel
Save