RocketMq消息 demo

参考 https://blog.csdn.net/asdf08442a/article/details/54882769 整理出来的测试 demo

1、produce 生产者

package com.bwdz.sp.comm.util.test;

import org.apache.rocketmq.client.exception.MQBrokerException;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.client.producer.SendStatus;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.remoting.exception.RemotingException;

import java.util.UUID;

/**
 * Created by xy on 2018/11/16.
 */
public class SyncProducer {
    private static DefaultMQProducer producer = null;

    public static void main(String[] args) {
        System.out.print("[----------]Start\n");
        int pro_count = 1;
        if (args.length > 0) {
            pro_count = Integer.parseInt(args[0]);
        }
        boolean result = false;
        try {
            ProducerStart();
            for (int i = 1; i < pro_count; i++) {
                String msg = "hello rocketmq "+ i+"".toString();
                SendMessage("qch_20170706",              //topic
                        "Tag"+i,                           //tag
                        "Key"+i,                           //key
                         msg);                                  //body
                System.out.print(msg + "\n");
            }
        }finally {
            producer.shutdown();
        }
        System.out.print("[----------]Succeed\n");
    }

    private static boolean ProducerStart() {
        producer = new DefaultMQProducer("pro_qch_test");
        producer.setNamesrvAddr("192.168.69.173:9876");
        producer.setInstanceName(UUID.randomUUID().toString());
        try {
            producer.start();
        } catch(MQClientException e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }

    private static boolean SendMessage(String topic,String tag,String key, String str) {
        Message msg = new Message(topic,tag,key,str.getBytes());
        try {
            SendResult result = producer.send(msg);
            SendStatus status = result.getSendStatus();
            System.out.println("___________________________SendMessage: "+status.name());
        } catch (MQClientException | RemotingException | MQBrokerException | InterruptedException e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }
}

2、consumer 消费者

package com.bwdz.sp.comm.util.test;

import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyContext;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;

import java.util.List;
import java.util.UUID;

/**
 * Created by xy on 2018/11/16.
 */
public class ConsumerTest {
    public static void main(String[] args) {
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("con_qch_test");
        consumer.setInstanceName(UUID.randomUUID().toString());
        consumer.setConsumeMessageBatchMaxSize(32);
        consumer.setNamesrvAddr("192.168.69.173:9876");
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        consumer.registerMessageListener(new MessageListenerConcurrently() {
            @Override
            public ConsumeConcurrentlyStatus consumeMessage(
                    List<MessageExt> list, ConsumeConcurrentlyContext consumeConcurrentlyContext) {
                for(MessageExt me : list) {
                    if("Tag1".equals(me.getTags())){
                        System.out.println("处理 Tag1 业务");
                        System.out.println(new String(me.getBody()) + "消费成功" + "\n");
                    }else if("Tag2".equals(me.getTags())){
                        System.out.println("处理 Tag2 业务");
                        System.out.println(new String(me.getBody()) + "消费成功" + "\n");
                    }else if("Tag3".equals(me.getTags())){
                        System.out.println("处理 Tag3 业务");
                        System.out.println(new String(me.getBody()) + "消费失败" + "\n");
                        return ConsumeConcurrentlyStatus.RECONSUME_LATER;
                    }else{
                        //consumer.subscribe("qch_20170706", "Tag1||Tag2||Tag3");
                        System.out.println("过滤掉的业务"+ me.getKeys());
                    }
                }
                return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
            }
        });
        try {
            consumer.subscribe("qch_20170706", "Tag1||Tag2||Tag3");
            consumer.start();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

 

先运行produce，控制台输出结果：

[----------]Start
___________________________SendMessage: SEND_OK
hello rocketmq 1
___________________________SendMessage: SEND_OK
hello rocketmq 2
___________________________SendMessage: SEND_OK
hello rocketmq 3
___________________________SendMessage: SEND_OK
hello rocketmq 4
[----------]Succeed

再运行consumer，控制台输出结果：

注：消息 ”hello rocketmq 4“ 被consumer里47行代码过滤掉了，所以不会被消费；消息 “hello rocket 3” 在消费的时候被指定失败ConsumeConcurrentlyStatus.RECONSUME_LATER，表示消费失败，如果被指定失败，表明此消息下次还可以继续发送到consumer被继续消费处理，其他消息则不会被再一次消费

处理 Tag2 业务
hello rocketmq 2消费成功

处理 Tag3 业务
hello rocketmq 3消费失败

处理 Tag1 业务
hello rocketmq 1消费成功

consumer再次运行，控制台输出结果(直到被指定成功ConsumeConcurrentlyStatus.CONSUME_SUCCESS，Broker服务才不会继续发送消息)：

处理 Tag3 业务
hello rocketmq 3消费失败