將自訂 HandlerMethodArgumentResolver 新增至 @RabbitListener

從 2.3.7 版開始,您可以新增自己的 HandlerMethodArgumentResolver 並解析自訂方法參數。您只需要實作 RabbitListenerConfigurer 並使用 RabbitListenerEndpointRegistrar 類別中的 setCustomMethodArgumentResolvers() 方法。

@Configuration
class CustomRabbitConfig implements RabbitListenerConfigurer {

    @Override
    public void configureRabbitListeners(RabbitListenerEndpointRegistrar registrar) {
        registrar.setCustomMethodArgumentResolvers(
				new HandlerMethodArgumentResolver() {

					@Override
					public boolean supportsParameter(MethodParameter parameter) {
						return CustomMethodArgument.class.isAssignableFrom(parameter.getParameterType());
					}

					@Override
					public Object resolveArgument(MethodParameter parameter, org.springframework.messaging.Message<?> message) {
						return new CustomMethodArgument(
								(String) message.getPayload(),
								message.getHeaders().get("customHeader", String.class)
						);
					}

				}
			);
    }

}