RabbitMQ + SpringAMPQ：在SimpleMessageListenerContainer中接收多个队列

Question

我目前正在学习有关RabbitMQ + SpringAMQ的一些知识，我试图使SimpleMessageListenerContainer能够读取传递到两个队列的消息，但是只有一个正在接收它。如果您有关于代码的其他提示，或者有任何更好的方法，请让我知道这会很好。

编辑：问题是该代码未将消息仅发送到两个队列。

基本上，我从两个队列中接收，一个在数据库中插入一些内容，另一个将收到验证请求。我希望以后再发送对路由键的响应并进行交换。

这是我的Application.java：

import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import org.springframework.amqp.rabbit.listener.adapter.MessageListenerAdapter;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;

@SpringBootApplication
public class ChallengeApplication {

    public static void main(String[] args) {
        SpringApplication.run(ChallengeApplication.class, args);
    }

    static final String responseExchange = "response.exchange";
    static final String routingKey = "response.routing.key";
    static final String queueInsertion = "insertion.queue";
    static final String queueValidation = "validation.queue";

    @Bean
    public ConnectionFactory connectionFactory() {
        CachingConnectionFactory connectionFactory = new CachingConnectionFactory();
        connectionFactory.setHost("127.0.0.1");
        connectionFactory.setVirtualHost("/");
        connectionFactory.setUsername("guest");
        connectionFactory.setPassword("guest");
        return connectionFactory;
    }

    @Bean
    Queue queueInsertion() {
        return new Queue(queueInsertion, true);
    }

    @Bean
    Queue queueValidation() {
        return new Queue(queueValidation, true);
    }

    @Bean
    DirectExchange exchange() {
        return new DirectExchange(responseExchange);
    }

    @Bean
    Binding bindingInsertion(@Qualifier("queueInsertion") Queue queue, DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(queueInsertion);
    }

    @Bean
    Binding queueValidation(@Qualifier("queueValidation") Queue queue, DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(queueValidation);
    }

    @Bean
    SimpleMessageListenerContainer containerValidation(ConnectionFactory connectionFactory,
            MessageListenerAdapter listenerAdapter) {
        SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
        container.setConnectionFactory(connectionFactory);
        container.setConcurrentConsumers(3);
        container.setQueueNames(queueValidation, queueInsertion);
        container.setMessageListener(listenerAdapter);
        return container;
    }


    @Bean
    MessageListenerAdapter listenerAdapter(Receiver receiver) {
        return new MessageListenerAdapter(receiver, "receiveMessage");
    }

}

您可以看到我的SimpleMessageListenerContainer在setQueueNames中同时包含了我要监听的两个队列。然后，我的MessageListenerAdapter将其发送到方法ReceiveMessage的Receiver.java中。另一个问题是如何从接收消息的队列中读取信息？可能是这种方式，还是应该使用@RabbitListener指向每个位置？

Receiver.java

import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.regex.Pattern;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataAccessException;
import org.springframework.stereotype.Component;

import com.axur.challenge.DAO.WhitelistDAO;
import com.axur.challenge.formatters.InputData;
import com.axur.challenge.model.Whitelist;
import com.google.gson.Gson;
import com.google.gson.JsonParseException;

@Component
public class Receiver {

    @Autowired
    private WhitelistDAO whitelistDAO;

    private CountDownLatch latch = new CountDownLatch(1);

    public void receiveMessage(String message) {
        try {
            InputData inputData = new Gson().fromJson(message, InputData.class);
            String url = inputData.getUrl();
            if (url == null) {
                receivedInsertion(inputData);
            } else {
                receivedValidation(inputData);
            }
        } catch (JsonParseException e) {
            System.out.println("It was not possible to read the input");
        }
        latch.countDown();
    }

    public void receivedInsertion(InputData inputData) {        
        Whitelist whitelist = new Whitelist();
        String client = inputData.getClient();

        if (client == null) {
            whitelist.setClient("global");
        } else {
            whitelist.setClient(inputData.getClient());
        }

        whitelist.setRegex(inputData.getRegex());

        System.out.println("Whitelist: " + whitelist);

        try {
            Whitelist selectedWhitelist = whitelistDAO.getSpecificWhitelist(whitelist.getClient(), whitelist.getRegex());
            System.out.println("selectedWhitelist: " + selectedWhitelist);
            if (selectedWhitelist == null || !selectedWhitelist.equals(whitelist)) {
                whitelistDAO.insertWhitelist(whitelist);
                System.out.println("Whitelist Added");
            } else {
                System.out.println("Already added before");
            }

        } catch (DataAccessException dae) {
            System.out.println("Whitelist NOT Added");
            System.err.println(dae);
        }

    }

    public void receivedValidation(InputData inputData) {
        //getting all regex from this client
        List<Whitelist> listWhitelist = whitelistDAO.getWhitelist(inputData.getClient(), inputData.getRegex());
        boolean match = false;
        //verify if the regex works for the url provided
        if (listWhitelist != null) {
            int index = 0;
            while (match == false && index < listWhitelist.size()) {
                match = checkRegex(listWhitelist.get(index).getRegex(), inputData.getUrl());
                index++;
            }
        }

        if (match == true) {

        } else {

        }
    }

    public boolean checkRegex(String regex, String url) {
        return Pattern.compile(regex).matcher(url).find();
    }

    public CountDownLatch getLatch() {
        return latch;
    }

}

这是Runner.java，我将消息发送到每个队列进行测试：

import java.util.concurrent.TimeUnit;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

@Component
public class Runner implements CommandLineRunner {

    private final RabbitTemplate rabbitTemplate;
    private final Receiver receiver;

    public Runner(Receiver receiver, RabbitTemplate rabbitTemplate) {
        this.receiver = receiver;
        this.rabbitTemplate = rabbitTemplate;
    }

    @Override
    public void run(String... args) throws Exception {
            System.out.println("Sending message...");
            rabbitTemplate.convertAndSend(ChallengeApplication.responseExchange, ChallengeApplication.queueInsertion, "{'client':null, 'regex':'[a-z]'}");
            rabbitTemplate.convertAndSend(ChallengeApplication.responseExchange, ChallengeApplication.queueValidation, "{'client':null, 'regex':'[a-z]'}");
            //rabbitTemplate.convertAndSend(ChallengeApplication.responseExchange, ChallengeApplication.routingKey, "{'client':null, 'regex':'[a-z]'}");
            receiver.getLatch().await(3, TimeUnit.SECONDS);
    }
}

谢谢您的时间！

Answer 1

在您的情况下，我认为您应该使用RabbitListner并将队列名称作为这样的参数传递：

 @RabbitListener(queues = RabbitTopicConfig.QUEUE_NAME)
    public void consumerDefaultMessage(final Message message){
        LOGGER.info("Recive Message with default configuration {}", message.toString());
    }

应该将配置模板中的RabbitTemplate作为bean注入，在这种情况下，由于存在队列，您可以使用许多列表器：

    @Bean
    public RabbitTemplate rabbitTemplate(final ConnectionFactory connectionFactory){
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        rabbitTemplate.setMessageConverter(producerMessageConverter());
        return rabbitTemplate;
    }