Return Registration object when registering interceptor

core/src/main/java/org/axonframework/queryhandling/SimpleQueryBus.java

@@ -193,19 +193,23 @@ private boolean unsubscribe(QueryDefinition registrationKey, MessageHandler<? su
      * respective handlers. The interceptor is invoked separately for each handler instance (in a separate unit of work).
      *
      * @param interceptor the interceptor to invoke before passing a Query to the handler
+     * @return handle to unregister the interceptor
      */
-    public void registerHandlerInterceptor(MessageHandlerInterceptor<QueryMessage<?, ?>> interceptor) {
-        this.handlerInterceptors.add(interceptor);
+    public Registration registerHandlerInterceptor(MessageHandlerInterceptor<QueryMessage<?, ?>> interceptor) {
+        handlerInterceptors.add(interceptor);
+        return () -> handlerInterceptors.remove(interceptor);
     }
 
     /**
      * Registers an interceptor that intercepts Queries as they are sent. Each interceptor is called
      * once, regardless of the type of query (point-to-point or scatter-gather) executed.
      *
      * @param interceptor the interceptor to invoke when sending a Query
+     * @return handle to unregister the interceptor
      */
-    public void registerDispatchInterceptor(MessageDispatchInterceptor<QueryMessage<?, ?>> interceptor) {
-        this.dispatchInterceptors.add(interceptor);
+    public Registration registerDispatchInterceptor(MessageDispatchInterceptor<QueryMessage<?, ?>> interceptor) {
+        dispatchInterceptors.add(interceptor);
+        return () -> dispatchInterceptors.remove(interceptor);
     }
 
     private <Q, R> Set<MessageHandler<? super QueryMessage<?, ?>>> getHandlersForMessage(QueryMessage<Q, R> queryMessage) {