Examples with MessageSelector org.apache.rocketmq.client.consumer.MessageSelector used on opensource projects

Example 1 with MessageSelector

use of org.apache.rocketmq.client.consumer.MessageSelector in project rocketmq-rocketmq-all-4.1.0-incubating by lirenzuo.

the class SqlFilterIT method testFilterConsumer.

@Test
public void testFilterConsumer() throws Exception {
    int msgSize = 16;
    String group = initConsumerGroup();
    MessageSelector selector = MessageSelector.bySql("(TAGS is not null and TAGS in ('TagA', 'TagB'))");
    RMQSqlConsumer consumer = ConsumerFactory.getRMQSqlConsumer(nsAddr, group, topic, selector, new RMQNormalListner(group + "_1"));
    Thread.sleep(3000);
    producer.send("TagA", msgSize);
    producer.send("TagB", msgSize);
    producer.send("TagC", msgSize);
    Assert.assertEquals("Not all sent succeeded", msgSize * 3, producer.getAllUndupMsgBody().size());
    consumer.getListner().waitForMessageConsume(msgSize * 2, consumeTime);
    assertThat(producer.getAllMsgBody()).containsAllIn(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(), consumer.getListner().getAllMsgBody()));
    assertThat(consumer.getListner().getAllMsgBody().size()).isEqualTo(msgSize * 2);
}

Example 2 with MessageSelector

use of org.apache.rocketmq.client.consumer.MessageSelector in project rocketmq by apache.

the class SqlFilterIT method testFilterConsumer.

@Test
public void testFilterConsumer() throws Exception {
    int msgSize = 16;
    String group = initConsumerGroup();
    MessageSelector selector = MessageSelector.bySql("(TAGS is not null and TAGS in ('TagA', 'TagB'))");
    RMQSqlConsumer consumer = ConsumerFactory.getRMQSqlConsumer(nsAddr, group, topic, selector, new RMQNormalListener(group + "_1"));
    Thread.sleep(3000);
    producer.send("TagA", msgSize);
    producer.send("TagB", msgSize);
    producer.send("TagC", msgSize);
    Assert.assertEquals("Not all sent succeeded", msgSize * 3, producer.getAllUndupMsgBody().size());
    consumer.getListener().waitForMessageConsume(msgSize * 2, consumeTime);
    assertThat(producer.getAllMsgBody()).containsAllIn(VerifyUtils.getFilterdMessage(producer.getAllMsgBody(), consumer.getListener().getAllMsgBody()));
    assertThat(consumer.getListener().getAllMsgBody().size()).isEqualTo(msgSize * 2);
}