背景
最近在改造一个IM系统后台,用到RabbitMQ。在处理消费线程的时候遇到点小麻烦,网上似乎也很少有提到如何多线程处理消息,所以记录一下。
业务场景
模块A向消息队列发送消息,模块B消费其中的消息,因为处理每条消息耗时略有不同,所以希望B能多线程处理。其中模块B建立20个Connection到MQ Broker,并且每个Connection创建一个Channel,用于消费消息。
实现方式
大概按照下面这样的方式,设置了一个40到160的线程池给ConnectionFactory。
ConnectionFactory factory = new ConnectionFactory();
factory.setSharedExecutor(someThreadPool); // someThreadPool为40到160的线程池
Connection mqConn = factory.newConnection(new ListAddressResolver(mqAddress));
someChannel = mqConn.createChannel();
someChannel.basicQos(1);
someChannel.queueDeclare(queueName, false, false, false, null);
String consumerTag = generateConsumerTag();
Consumer consumer = new SomeConsumer(someChannel;
someChannel.basicConsume(outputQueue, true, consumerTag, consumer);