RocketMQ基本API使用

RocketMQ基本API使用

基于原生客户端

<dependency>
            <groupId>org.apache.rocketmq</groupId>
            <artifactId>rocketmq-client</artifactId>
            <version>4.2.0</version>
 </dependency>

简单案例——同步发送Producer

/**
 * 同步发送消息：需要Broker返回确认信息
 */
public class SyncProducer {

    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("group_SyncProducer");
        producer.setNamesrvAddr("localhost:9876");
        producer.start();
        for (int i = 0; i < 100; i++) {
            Message msg = new Message("topic_SyncProducer", "tag_SyncProducer", ("Hello RocketMQ " + i).getBytes(RemotingHelper.DEFAULT_CHARSET));
            final SendResult sendResult = producer.send(msg);
            System.out.printf("%s%n", sendResult);
        }
        producer.shutdown();
    }
}

简单案例——异步发送Producer

/**
 * 异步发送消息：需要Broker返回确认信息
 */
public class AsyncProducer {

    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("group_AsyncProducer");
        producer.setNamesrvAddr("localhost:9876");
        producer.start();
        producer.setRetryTimesWhenSendAsyncFailed(0);

        int messageCount = 100;
        final CountDownLatch countDownLatch = new CountDownLatch(messageCount);
        for (int i = 0; i < messageCount; i++) {
            try {
                final int index = i;
                Message msg = new Message(
                        "topic_AsyncProducer",
                        "tag_AsyncProducer",
                        "Hello Word".getBytes(RemotingHelper.DEFAULT_CHARSET)
                );
                producer.send(msg, new SendCallback() {
                    @Override
                    public void onSuccess(SendResult sendResult) {
                        countDownLatch.countDown();
                        System.out.printf("%-10d OK %s %n", index, sendResult.getMsgId());
                    }

                    @Override
                    public void onException(Throwable e) {
                        countDownLatch.countDown();
                        System.out.printf("%-10d Exception %s %n", index, e);
                        e.printStackTrace();
                    }
                });
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        countDownLatch.await(5, TimeUnit.SECONDS);
        producer.shutdown();
    }
}

简单案例——单向发送Producer

/**
 * One-way模式发送消息即单向模式：不需要Broker返回确认信息
 */
public class OneWayProducer {

    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("group_OneWayProducer");
        producer.setNamesrvAddr("localhost:9876");
        producer.start();
        for (int i = 0; i < 100; i++) {
            Message msg = new Message(
                    "topic_OneWayProducer",
                    "tag_OneWayProducer",
                    ("Hello RocketMQ " + i).getBytes(RemotingHelper.DEFAULT_CHARSET)
            );
            producer.sendOneway(msg);
        }
        TimeUnit.SECONDS.sleep(5);
        producer.shutdown();
    }
}

简单案例——Consumer

public class Consumer {

    public static void main(String[] args) throws InterruptedException {
        Thread syncThread = new Thread(() -> {
            DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("group_SyncProducer");
            consumer.setNamesrvAddr("localhost:9876");
            try {
                consumer.subscribe("topic_SyncProducer", "*");
            } catch (MQClientException e) {
                e.printStackTrace();
            }
            consumer.registerMessageListener(new MessageListenerConcurrently() {
                @Override
                public ConsumeConcurrentlyStatus consumeMessage(
                        List<MessageExt> msgs,
                        ConsumeConcurrentlyContext context) {
                    System.out.printf("%s Receive New Messages: %s %n", Thread.currentThread().getName(), msgs);
                    return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
                }
            });

            try {
                consumer.start();
            } catch (MQClientException e) {
                e.printStackTrace();
            }
            System.out.printf("Consumer Started.%n");
        });

        Thread asyncThread = new Thread(() -> {
            DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("group_AsyncProducer");
            consumer.setNamesrvAddr("localhost:9876");
            try {
                consumer.subscribe("topic_AsyncProducer", "*");
            } catch (MQClientException e) {
                e.printStackTrace();
            }
            consumer.registerMessageListener(new MessageListenerConcurrently() {
                @Override
                public ConsumeConcurrentlyStatus consumeMessage(
                        List<MessageExt> msgs,
                        ConsumeConcurrentlyContext context) {
                    System.out.printf("%s Receive New Messages: %s %n", Thread.currentThread().getName(), msgs);
                    return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
                }
            });

            try {
                consumer.start();
            } catch (MQClientException e) {
                e.printStackTrace();
            }
            System.out.printf("Consumer Started.%n");
        });

        Thread oneWayThread = new Thread(() -> {
            DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("group_OneWayProducer");
            consumer.setNamesrvAddr("localhost:9876");
            try {
                consumer.subscribe("topic_OneWayProducer", "*");
            } catch (MQClientException e) {
                e.printStackTrace();
            }
            consumer.registerMessageListener(new MessageListenerConcurrently() {
                @Override
                public ConsumeConcurrentlyStatus consumeMessage(
                        List<MessageExt> msgs,
                        ConsumeConcurrentlyContext context) {
                    System.out.printf("%s Receive New Messages: %s %n", Thread.currentThread().getName(), msgs);
                    return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
                }
            });

            try {
                consumer.start();
            } catch (MQClientException e) {
                e.printStackTrace();
            }
            System.out.printf("Consumer Started.%n");
        });

        syncThread.start();
        asyncThread.start();
        oneWayThread.start();
        TimeUnit.MINUTES.sleep(10);
    }
}

顺序消息——Producer

public class OrderedProducer {

    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("group_OrderedProducer");
        producer.setNamesrvAddr("localhost:9876");
        producer.start();

        String[] tags = new String[]{"TagA", "TagB", "TagC", "TagD", "TagE"};
        for (int i = 0; i < 100; i++) {
            int orderId = i % 10;
            Message msg = new Message(
                    "topic_OrderedProducer",
                    tags[i % tags.length],
                    "KEY" + i,
                    ("Hello RocketMQ " + i).getBytes(RemotingHelper.DEFAULT_CHARSET)
            );
            SendResult sendResult = producer.send(msg, new MessageQueueSelector() {
                @Override
                public MessageQueue select(List<MessageQueue> mqs, Message msg, Object arg) {
                    Integer id = (Integer) arg;
                    int index = id % mqs.size();
                    return mqs.get(index);
                }
            }, orderId);

            System.out.printf("%s%n", sendResult);
        }
        producer.shutdown();
    }
}

顺序消息——Consumer

public class OrderedConsumer {

    public static void main(String[] args) throws Exception {
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("group_OrderedProducer");
        consumer.setNamesrvAddr("localhost:9876");
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        consumer.subscribe("topic_OrderedProducer", "TagA || TagC || TagD");
        consumer.registerMessageListener(new MessageListenerOrderly() {
            AtomicLong consumeTimes = new AtomicLong(0);

            @Override
            public ConsumeOrderlyStatus consumeMessage(List<MessageExt> msgs, ConsumeOrderlyContext context) {
                context.setAutoCommit(false);
                System.out.printf(Thread.currentThread().getName() + " Receive New Messages: " + msgs + "%n");
                this.consumeTimes.incrementAndGet();
                if ((this.consumeTimes.get() % 2) == 0) {
                    return ConsumeOrderlyStatus.SUCCESS;
                } else if ((this.consumeTimes.get() % 3) == 0) {
                    return ConsumeOrderlyStatus.ROLLBACK;
                } else if ((this.consumeTimes.get() % 4) == 0) {
                    return ConsumeOrderlyStatus.COMMIT;
                } else if ((this.consumeTimes.get() % 5) == 0) {
                    context.setSuspendCurrentQueueTimeMillis(3000);
                    return ConsumeOrderlyStatus.SUSPEND_CURRENT_QUEUE_A_MOMENT;
                }
                return ConsumeOrderlyStatus.SUCCESS;
            }
        });

        consumer.start();
        System.out.printf("Consumer Started.%n");
    }
}

延时消息——Producer

public class ScheduledMessageProducer {
    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("group_Scheduled");
        producer.setNamesrvAddr("localhost:9876");
        producer.start();
        int totalMessagesToSend = 100;
        for (int i = 0; i < totalMessagesToSend; i++) {
            Message message = new Message("topic_Scheduled", ("Hello scheduled message " + i).getBytes());
            //messageDelayLevel=1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h 共18个等级段
            message.setDelayTimeLevel(3);
            producer.send(message);
        }

        // Shutdown producer after use.
        producer.shutdown();
    }
}

延时消息——Consumer

public class ScheduledMessageProducer {
    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("group_Scheduled");
        producer.setNamesrvAddr("localhost:9876");
        producer.start();
        int totalMessagesToSend = 100;
        for (int i = 0; i < totalMessagesToSend; i++) {
            Message message = new Message("topic_Scheduled", ("Hello scheduled message " + i).getBytes());
            //messageDelayLevel=1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h 共18个等级段
            message.setDelayTimeLevel(3);
            producer.send(message);
        }

        // Shutdown producer after use.
        producer.shutdown();
    }
}

参考资料：
《RocketMQ英文官网》