Examples with Header org.springframework.messaging.handler.annotation.Header used on opensource projects

Example 1 with Header

use of org.springframework.messaging.handler.annotation.Header in project spring-integration by spring-projects.

the class MethodInvokingMessageProcessorTests method testOptionalArgs.

@Test
public void testOptionalArgs() throws Exception {
    class Foo {

        private final Map<String, Object> arguments = new LinkedHashMap<String, Object>();

        @SuppressWarnings("unused")
        public void optionalHeaders(Optional<String> foo, @Header(value = "foo", required = false) String foo1, @Header("foo") Optional<String> foo2) {
            this.arguments.put("foo", (foo.isPresent() ? foo.get() : null));
            this.arguments.put("foo1", foo1);
            this.arguments.put("foo2", (foo2.isPresent() ? foo2.get() : null));
        }
    }
    Foo targetObject = new Foo();
    MessagingMethodInvokerHelper helper = new MessagingMethodInvokerHelper(targetObject, (String) null, false);
    helper.process(new GenericMessage<>(Optional.empty()));
    assertNull(targetObject.arguments.get("foo"));
    assertNull(targetObject.arguments.get("foo1"));
    assertNull(targetObject.arguments.get("foo2"));
    helper.process(MessageBuilder.withPayload("foo").setHeader("foo", "FOO").build());
    assertEquals("foo", targetObject.arguments.get("foo"));
    assertEquals("FOO", targetObject.arguments.get("foo1"));
    assertEquals("FOO", targetObject.arguments.get("foo2"));
}

Example 2 with Header

use of org.springframework.messaging.handler.annotation.Header in project spring-integration by spring-projects.

the class MethodInvokingMessageProcessorTests method testProxyAndHeaderAnnotation.

@Test
public void testProxyAndHeaderAnnotation() {
    final AtomicReference<Object> payloadReference = new AtomicReference<>();
    final AtomicReference<UUID> idReference = new AtomicReference<>();
    class MyHandler {

        public void handle(@Header(MessageHeaders.ID) UUID id, @Payload Object payload) {
            idReference.set(id);
            payloadReference.set(payload);
        }
    }
    MyHandler service = new MyHandler();
    final AtomicBoolean adviceCalled = new AtomicBoolean();
    ProxyFactory proxyFactory = new ProxyFactory(service);
    proxyFactory.addAdvice((MethodInterceptor) i -> {
        adviceCalled.set(true);
        return i.proceed();
    });
    service = (MyHandler) proxyFactory.getProxy(getClass().getClassLoader());
    GenericMessage<String> testMessage = new GenericMessage<>("foo");
    MethodInvokingMessageProcessor processor = new MethodInvokingMessageProcessor(service, "handle");
    processor.processMessage(testMessage);
    assertEquals(testMessage.getPayload(), payloadReference.get());
    assertEquals(testMessage.getHeaders().getId(), idReference.get());
    assertTrue(adviceCalled.get());
}

Example 3 with Header

use of org.springframework.messaging.handler.annotation.Header in project spring-integration by spring-projects.

the class MethodInvokingMessageGroupProcessorTests method testHeaderParameters.

@Test
public void testHeaderParameters() throws Exception {
    class SingleAnnotationTestBean {

        @Aggregator
        public String method1(List<String> input, @Header("foo") String foo) {
            return input.get(0) + foo;
        }
    }
    SingleAnnotationTestBean bean = new SingleAnnotationTestBean();
    MethodInvokingMessageGroupProcessor aggregator = new MethodInvokingMessageGroupProcessor(bean);
    SimpleMessageGroup group = new SimpleMessageGroup("FOO");
    group.add(MessageBuilder.withPayload("foo").setHeader("foo", "bar").build());
    group.add(MessageBuilder.withPayload("bar").setHeader("foo", "bar").build());
    assertEquals("foobar", aggregator.aggregatePayloads(group, aggregator.aggregateHeaders(group)));
}

Example 4 with Header

use of org.springframework.messaging.handler.annotation.Header in project spring-integration by spring-projects.

the class GatewayProxyFactoryBean method createGatewayForMethod.

private MethodInvocationGateway createGatewayForMethod(Method method) {
    Gateway gatewayAnnotation = method.getAnnotation(Gateway.class);
    String requestChannelName = null;
    String replyChannelName = null;
    Expression requestTimeout = this.defaultRequestTimeout;
    Expression replyTimeout = this.defaultReplyTimeout;
    String payloadExpression = this.globalMethodMetadata != null ? this.globalMethodMetadata.getPayloadExpression() : null;
    Map<String, Expression> headerExpressions = new HashMap<String, Expression>();
    if (gatewayAnnotation != null) {
        requestChannelName = gatewayAnnotation.requestChannel();
        replyChannelName = gatewayAnnotation.replyChannel();
        /*
			 * INT-2636 Unspecified annotation attributes should not
			 * override the default values supplied by explicit configuration.
			 * There is a small risk that someone has used Long.MIN_VALUE explicitly
			 * to indicate an indefinite timeout on a gateway method and that will
			 * no longer work as expected; they will need to use, say, -1 instead.
			 */
        if (requestTimeout == null || gatewayAnnotation.requestTimeout() != Long.MIN_VALUE) {
            requestTimeout = new ValueExpression<>(gatewayAnnotation.requestTimeout());
        }
        if (StringUtils.hasText(gatewayAnnotation.requestTimeoutExpression())) {
            requestTimeout = ExpressionUtils.longExpression(gatewayAnnotation.requestTimeoutExpression());
        }
        if (replyTimeout == null || gatewayAnnotation.replyTimeout() != Long.MIN_VALUE) {
            replyTimeout = new ValueExpression<>(gatewayAnnotation.replyTimeout());
        }
        if (StringUtils.hasText(gatewayAnnotation.replyTimeoutExpression())) {
            replyTimeout = ExpressionUtils.longExpression(gatewayAnnotation.replyTimeoutExpression());
        }
        if (payloadExpression == null || StringUtils.hasText(gatewayAnnotation.payloadExpression())) {
            payloadExpression = gatewayAnnotation.payloadExpression();
        }
        if (!ObjectUtils.isEmpty(gatewayAnnotation.headers())) {
            for (GatewayHeader gatewayHeader : gatewayAnnotation.headers()) {
                String value = gatewayHeader.value();
                String expression = gatewayHeader.expression();
                String name = gatewayHeader.name();
                boolean hasValue = StringUtils.hasText(value);
                if (hasValue == StringUtils.hasText(expression)) {
                    throw new BeanDefinitionStoreException("exactly one of 'value' or 'expression' " + "is required on a gateway's header.");
                }
                headerExpressions.put(name, hasValue ? new LiteralExpression(value) : EXPRESSION_PARSER.parseExpression(expression));
            }
        }
    } else if (this.methodMetadataMap != null && this.methodMetadataMap.size() > 0) {
        GatewayMethodMetadata methodMetadata = this.methodMetadataMap.get(method.getName());
        if (methodMetadata != null) {
            if (StringUtils.hasText(methodMetadata.getPayloadExpression())) {
                payloadExpression = methodMetadata.getPayloadExpression();
            }
            if (!CollectionUtils.isEmpty(methodMetadata.getHeaderExpressions())) {
                headerExpressions.putAll(methodMetadata.getHeaderExpressions());
            }
            requestChannelName = methodMetadata.getRequestChannelName();
            replyChannelName = methodMetadata.getReplyChannelName();
            String reqTimeout = methodMetadata.getRequestTimeout();
            if (StringUtils.hasText(reqTimeout)) {
                requestTimeout = ExpressionUtils.longExpression(reqTimeout);
            }
            String repTimeout = methodMetadata.getReplyTimeout();
            if (StringUtils.hasText(repTimeout)) {
                replyTimeout = ExpressionUtils.longExpression(repTimeout);
            }
        }
    }
    Map<String, Object> headers = null;
    // We don't want to eagerly resolve the error channel here
    Object errorChannel = this.errorChannel == null ? this.errorChannelName : this.errorChannel;
    if (errorChannel != null && method.getReturnType().equals(void.class)) {
        headers = new HashMap<>();
        headers.put(MessageHeaders.ERROR_CHANNEL, errorChannel);
    }
    if (getMessageBuilderFactory() instanceof DefaultMessageBuilderFactory) {
        Set<String> headerNames = new HashSet<>(headerExpressions.keySet());
        if (this.globalMethodMetadata != null) {
            headerNames.addAll(this.globalMethodMetadata.getHeaderExpressions().keySet());
        }
        List<MethodParameter> methodParameters = GatewayMethodInboundMessageMapper.getMethodParameterList(method);
        for (MethodParameter methodParameter : methodParameters) {
            Header header = methodParameter.getParameterAnnotation(Header.class);
            if (header != null) {
                String headerName = GatewayMethodInboundMessageMapper.determineHeaderName(header, methodParameter);
                headerNames.add(headerName);
            }
        }
        for (String header : headerNames) {
            if ((MessageHeaders.ID.equals(header) || MessageHeaders.TIMESTAMP.equals(header))) {
                throw new BeanInitializationException("Messaging Gateway cannot override 'id' and 'timestamp' read-only headers.\n" + "Wrong headers configuration for " + getComponentName());
            }
        }
    }
    GatewayMethodInboundMessageMapper messageMapper = new GatewayMethodInboundMessageMapper(method, headerExpressions, this.globalMethodMetadata != null ? this.globalMethodMetadata.getHeaderExpressions() : null, headers, this.argsMapper, this.getMessageBuilderFactory());
    if (StringUtils.hasText(payloadExpression)) {
        messageMapper.setPayloadExpression(payloadExpression);
    }
    messageMapper.setBeanFactory(getBeanFactory());
    MethodInvocationGateway gateway = new MethodInvocationGateway(messageMapper);
    if (this.errorChannel != null) {
        gateway.setErrorChannel(this.errorChannel);
    } else if (StringUtils.hasText(this.errorChannelName)) {
        gateway.setErrorChannelName(this.errorChannelName);
    }
    if (this.getTaskScheduler() != null) {
        gateway.setTaskScheduler(this.getTaskScheduler());
    }
    gateway.setBeanName(this.getComponentName());
    if (StringUtils.hasText(requestChannelName)) {
        gateway.setRequestChannelName(requestChannelName);
    } else if (StringUtils.hasText(this.defaultRequestChannelName)) {
        gateway.setRequestChannelName(this.defaultRequestChannelName);
    } else {
        gateway.setRequestChannel(this.defaultRequestChannel);
    }
    if (StringUtils.hasText(replyChannelName)) {
        gateway.setReplyChannelName(replyChannelName);
    } else if (StringUtils.hasText(this.defaultReplyChannelName)) {
        gateway.setReplyChannelName(this.defaultReplyChannelName);
    } else {
        gateway.setReplyChannel(this.defaultReplyChannel);
    }
    if (requestTimeout == null) {
        gateway.setRequestTimeout(-1);
    } else if (requestTimeout instanceof ValueExpression) {
        gateway.setRequestTimeout(requestTimeout.getValue(Long.class));
    } else {
        messageMapper.setSendTimeoutExpression(requestTimeout);
    }
    if (replyTimeout == null) {
        gateway.setReplyTimeout(-1);
    } else if (replyTimeout instanceof ValueExpression) {
        gateway.setReplyTimeout(replyTimeout.getValue(Long.class));
    } else {
        messageMapper.setReplyTimeoutExpression(replyTimeout);
    }
    if (this.getBeanFactory() != null) {
        gateway.setBeanFactory(this.getBeanFactory());
    }
    if (replyTimeout != null) {
        gateway.setReceiveTimeoutExpression(replyTimeout);
    }
    gateway.setShouldTrack(this.shouldTrack);
    gateway.afterPropertiesSet();
    return gateway;
}