How to use

org.springframework.amqp.rabbit.AsyncRabbitTemplate$RabbitMessageFuture

constructor

@Test
public void testAsyncDelayExpression() {
  ConnectionFactory connectionFactory = mock(ConnectionFactory.class);
  AsyncRabbitTemplate amqpTemplate = spy(new AsyncRabbitTemplate(new RabbitTemplate(connectionFactory),
      new SimpleMessageListenerContainer(connectionFactory), "replyTo"));
  amqpTemplate.setTaskScheduler(mock(TaskScheduler.class));
  AsyncAmqpOutboundGateway gateway = new AsyncAmqpOutboundGateway(amqpTemplate);
  willAnswer(
      invocation -> amqpTemplate.new RabbitMessageFuture("foo", invocation.getArgument(2)))
        .given(amqpTemplate).sendAndReceive(anyString(), anyString(), any(Message.class));
  gateway.setExchangeName("foo");
  gateway.setRoutingKey("bar");
  gateway.setDelayExpressionString("42");
  gateway.setBeanFactory(mock(BeanFactory.class));
  gateway.setOutputChannel(new NullChannel());
  gateway.afterPropertiesSet();
  gateway.start();
  ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
  gateway.handleMessage(new GenericMessage<>("foo"));
  verify(amqpTemplate).sendAndReceive(eq("foo"), eq("bar"), captor.capture());
  assertThat(captor.getValue().getMessageProperties().getDelay(), equalTo(42));
}

RabbitMessageFuture future = asyncTemplate.new RabbitMessageFuture(null, null);
willReturn(future).given(asyncTemplate).sendAndReceive(anyString(), anyString(),
    any(org.springframework.amqp.core.Message.class));

@Override
public RabbitMessageFuture sendAndReceive(String exchange, String routingKey, Message message) {
  String correlationId = getOrSetCorrelationIdAndSetReplyTo(message);
  RabbitMessageFuture future = new RabbitMessageFuture(correlationId, message);
  CorrelationData correlationData = null;
  if (this.enableConfirms) {
    correlationData = new CorrelationData(correlationId);
    future.setConfirm(new SettableListenableFuture<>());
  }
  this.pending.put(correlationId, future);
  if (this.container != null) {
    this.template.send(exchange, routingKey, message, correlationData);
  }
  else {
    ChannelHolder channelHolder = this.directReplyToContainer.getChannelHolder();
    future.setChannelHolder(channelHolder);
    sendDirect(channelHolder.getChannel(), exchange, routingKey, message, correlationData);
  }
  future.startTimer();
  return future;
}

@Override
public RabbitMessageFuture sendAndReceive(String exchange, String routingKey, Message message) {
  String correlationId = getOrSetCorrelationIdAndSetReplyTo(message);
  RabbitMessageFuture future = new RabbitMessageFuture(correlationId, message);
  CorrelationData correlationData = null;
  if (this.enableConfirms) {
    correlationData = new CorrelationData(correlationId);
    future.setConfirm(new SettableListenableFuture<>());
  }
  this.pending.put(correlationId, future);
  if (this.container != null) {
    this.template.send(exchange, routingKey, message, correlationData);
  }
  else {
    ChannelHolder channelHolder = this.directReplyToContainer.getChannelHolder();
    future.setChannelHolder(channelHolder);
    sendDirect(channelHolder.getChannel(), exchange, routingKey, message, correlationData);
  }
  future.startTimer();
  return future;
}