diff --git a/core/src/main/java/org/axonframework/queryhandling/SimpleQueryBus.java b/core/src/main/java/org/axonframework/queryhandling/SimpleQueryBus.java index 73386927dc..44b429efe5 100644 --- a/core/src/main/java/org/axonframework/queryhandling/SimpleQueryBus.java +++ b/core/src/main/java/org/axonframework/queryhandling/SimpleQueryBus.java @@ -193,9 +193,11 @@ private boolean unsubscribe(QueryDefinition registrationKey, MessageHandler> interceptor) { - this.handlerInterceptors.add(interceptor); + public Registration registerHandlerInterceptor(MessageHandlerInterceptor> interceptor) { + handlerInterceptors.add(interceptor); + return () -> handlerInterceptors.remove(interceptor); } /** @@ -203,9 +205,11 @@ public void registerHandlerInterceptor(MessageHandlerInterceptor> interceptor) { - this.dispatchInterceptors.add(interceptor); + public Registration registerDispatchInterceptor(MessageDispatchInterceptor> interceptor) { + dispatchInterceptors.add(interceptor); + return () -> dispatchInterceptors.remove(interceptor); } private Set>> getHandlersForMessage(QueryMessage queryMessage) {