How to use

RMQAsyncSendProducer

in

org.apache.rocketmq.test.client.rmq

@Test
public void testSendWithOnlyCallBack() {
  int msgSize = 20;
  RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());
  producer.asyncSend(msgSize);
  producer.waitForResponse(10 * 1000);
  assertThat(producer.getSuccessMsgCount()).isEqualTo(msgSize);
  consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
  assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
    consumer.getListener().getAllMsgBody()))
    .containsExactlyElementsIn(producer.getAllMsgBody());
}

  @Test
  public void testOneWaySendWithOnlyMsgAsParam() {
    int msgSize = 20;
    RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());

    producer.sendOneWay(msgSize);
    producer.waitForResponse(5 * 1000);
    assertThat(producer.getAllMsgBody().size()).isEqualTo(msgSize);

    consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
    assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
      consumer.getListener().getAllMsgBody()))
      .containsExactlyElementsIn(producer.getAllMsgBody());
  }
}

public static RMQAsyncSendProducer getAsyncProducer(String nsAddr, String topic) {
  RMQAsyncSendProducer producer = new RMQAsyncSendProducer(nsAddr, topic);
  if (debug) {
    producer.setDebug();
  }
  mqClients.add(producer);
  return producer;
}

public RMQAsyncSendProducer(String nsAddr, String topic) {
  super(topic);
  this.nsAddr = nsAddr;
  sendCallback = new SendCallback() {
    @Override
    public void onSuccess(SendResult sendResult) {
      successSendResult.add(sendResult);
    }
    @Override
    public void onException(Throwable throwable) {
      exceptionMsgCount.getAndIncrement();
    }
  };
  create();
  start();
}

  @Test
  public void testAsyncSendWithMQ() {
    int msgSize = 20;
    int queueId = 0;
    RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());
    MessageQueue mq = new MessageQueue(topic, broker1Name, queueId);

    producer.sendOneWay(msgSize, mq);
    producer.waitForResponse(5 * 1000);

    consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
    assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
      consumer.getListener().getAllMsgBody()))
      .containsExactlyElementsIn(producer.getAllMsgBody());

    producer.clearMsg();
    consumer.clearMsg();

    mq = new MessageQueue(topic, broker2Name, queueId);
    producer.asyncSend(msgSize, mq);
    producer.waitForResponse(5 * 1000);

    consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
    assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
      consumer.getListener().getAllMsgBody()))
      .containsExactlyElementsIn(producer.getAllMsgBody());
  }
}

RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());
producer.sendOneWay(msgSize, new MessageQueueSelector() {
  @Override
  public MessageQueue select(List<MessageQueue> list, Message message, Object o) {
assertThat(producer.getAllMsgBody().size()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());
producer.clearMsg();
consumer.clearMsg();
producer.sendOneWay(msgSize, new MessageQueueSelector() {
  @Override
  public MessageQueue select(List<MessageQueue> list, Message message, Object o) {
assertThat(producer.getAllMsgBody().size()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());

public void asyncSend(int msgSize, MessageQueue mq) {
  this.msgSize = msgSize;
  for (int i = 0; i < msgSize; i++) {
    Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
    this.asyncSend(msg, mq);
  }
}

  public void sendOneWay(int msgSize, MessageQueueSelector selector) {
    for (int i = 0; i < msgSize; i++) {
      Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
      this.sendOneWay(msg, selector, i);
    }
  }
}

  @Test
  public void testAsyncSendWithMQ() {
    int msgSize = 20;
    int queueId = 0;
    RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());
    MessageQueue mq = new MessageQueue(topic, broker1Name, queueId);

    producer.sendOneWay(msgSize, mq);
    producer.waitForResponse(5 * 1000);

    consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
    assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
      consumer.getListener().getAllMsgBody()))
      .containsExactlyElementsIn(producer.getAllMsgBody());

    producer.clearMsg();
    consumer.clearMsg();

    mq = new MessageQueue(topic, broker2Name, queueId);
    producer.asyncSend(msgSize, mq);
    producer.waitForResponse(5 * 1000);

    consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
    assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
      consumer.getListener().getAllMsgBody()))
      .containsExactlyElementsIn(producer.getAllMsgBody());
  }
}

RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());
producer.sendOneWay(msgSize, new MessageQueueSelector() {
  @Override
  public MessageQueue select(List<MessageQueue> list, Message message, Object o) {
assertThat(producer.getAllMsgBody().size()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());
producer.clearMsg();
consumer.clearMsg();
producer.sendOneWay(msgSize, new MessageQueueSelector() {
  @Override
  public MessageQueue select(List<MessageQueue> list, Message message, Object o) {
assertThat(producer.getAllMsgBody().size()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());

public void asyncSend(int msgSize, MessageQueueSelector selector) {
  this.msgSize = msgSize;
  for (int i = 0; i < msgSize; i++) {
    Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
    this.asyncSend(msg, selector, i);
  }
}

public void sendOneWay(int msgSize) {
  for (int i = 0; i < msgSize; i++) {
    Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
    this.sendOneWay(msg);
  }
}

public RMQAsyncSendProducer(String nsAddr, String topic) {
  super(topic);
  this.nsAddr = nsAddr;
  sendCallback = new SendCallback() {
    public void onSuccess(org.apache.rocketmq.client.producer.SendResult sendResult) {
      successSendResult.add(sendResult);
    }
    public void onException(Throwable throwable) {
      exceptionMsgCount.getAndIncrement();
    }
  };
  create();
  start();
}

MessageQueue mq = new MessageQueue(topic, broker1Name, queueId);
producer.asyncSend(msgSize, mq);
producer.waitForResponse(10 * 1000);
assertThat(producer.getSuccessMsgCount()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());
producer.clearMsg();
consumer.clearMsg();
producer.getSuccessSendResult().clear();
mq = new MessageQueue(topic, broker2Name, queueId);
producer.asyncSend(msgSize, mq);
producer.waitForResponse(10 * 1000);
assertThat(producer.getSuccessMsgCount()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());

  @Test
  public void testOneWaySendWithOnlyMsgAsParam() {
    int msgSize = 20;
    RMQNormalConsumer consumer = getConsumer(nsAddr, topic, "*", new RMQNormalListener());

    producer.sendOneWay(msgSize);
    producer.waitForResponse(5 * 1000);
    assertThat(producer.getAllMsgBody().size()).isEqualTo(msgSize);

    consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
    assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
      consumer.getListener().getAllMsgBody()))
      .containsExactlyElementsIn(producer.getAllMsgBody());
  }
}

public void asyncSend(int msgSize) {
  this.msgSize = msgSize;
  for (int i = 0; i < msgSize; i++) {
    Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
    this.asyncSend(msg);
  }
}

public void sendOneWay(int msgSize, MessageQueue mq) {
  for (int i = 0; i < msgSize; i++) {
    Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
    this.sendOneWay(msg, mq);
  }
}

public static RMQAsyncSendProducer getAsyncProducer(String nsAddr, String topic) {
  RMQAsyncSendProducer producer = new RMQAsyncSendProducer(nsAddr, topic);
  if (debug) {
    producer.setDebug();
  }
  mqClients.add(producer);
  return producer;
}

producer.asyncSend(msgSize, new MessageQueueSelector() {
  @Override
  public MessageQueue select(List<MessageQueue> list, Message message, Object o) {
producer.waitForResponse(5 * 1000);
assertThat(producer.getSuccessMsgCount()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());
producer.clearMsg();
consumer.clearMsg();
producer.getSuccessSendResult().clear();
producer.asyncSend(msgSize, new MessageQueueSelector() {
  @Override
  public MessageQueue select(List<MessageQueue> list, Message message, Object o) {
producer.waitForResponse(5 * 1000);
assertThat(producer.getSuccessMsgCount()).isEqualTo(msgSize);
consumer.getListener().waitForMessageConsume(producer.getAllMsgBody(), consumeTime);
assertThat(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(),
  consumer.getListener().getAllMsgBody()))
  .containsExactlyElementsIn(producer.getAllMsgBody());

public void asyncSend(int msgSize) {
  this.msgSize = msgSize;
  for (int i = 0; i < msgSize; i++) {
    Message msg = new Message(topic, RandomUtil.getStringByUUID().getBytes());
    this.asyncSend(msg);
  }
}