RabbitMQ RPC

RPCClient

package com.gch.rabbit.demo;

import com.rabbitmq.client.AMQP.BasicProperties;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.QueueingConsumer;
import java.io.IOException;
import java.util.concurrent.TimeoutException;

public class TestRPCClient {

    private static final String RPC_QUEUE_NAME = "rpc_queue";

    public static void main(String[] args) throws IOException, TimeoutException, InterruptedException {

        ConnectionFactory factory = new ConnectionFactory();//创建工厂
        factory.setHost("localhost");//设置ip
        Connection connection = factory.newConnection();//创建连接
        Channel channel = connection.createChannel();//创建信道

        channel.queueDeclare(RPC_QUEUE_NAME,false, false, false, null);//声明队列
        String replyQueueName = channel.queueDeclare().getQueue();//声明一个临时的回调队列

        QueueingConsumer consumer = new QueueingConsumer(channel);//为这个信道创建一个消费者，用于监听信道中的回调队列，获取结果

        channel.basicConsume(replyQueueName, true, consumer);//把临时的回调队列和消费者绑定

        String corrId = java.util.UUID.randomUUID().toString();//创建一个唯一id

        BasicProperties props = new BasicProperties
            .Builder()
            .correlationId(corrId)
            .replyTo(replyQueueName)
            .build(); //设置replyTo和correlationId属性值

        String message = "发送的请求";
        //发送消息到rpc_queue队列
        channel.basicPublish("", RPC_QUEUE_NAME, props, message.getBytes());//发送消息并声名回调方法

        String result;//用于接收返回的结果
        while (true) {
            QueueingConsumer.Delivery delivery = consumer.nextDelivery();
            if (delivery.getProperties().getCorrelationId().equals(corrId)) {
                result = new String(delivery.getBody(),"UTF-8");
                break;
            }
        }
        System.out.println( "result：" + result);

    }

}

RPCServer

package com.gch.rabbit.demo;

import com.rabbitmq.client.AMQP.BasicProperties;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.QueueingConsumer;
import java.io.IOException;
import java.util.concurrent.TimeoutException;

public class TestRPCServer {

    private static final String RPC_QUEUE_NAME = "rpc_queue";

    public static void main (String[] args) throws IOException, TimeoutException, InterruptedException {

        ConnectionFactory factory = new ConnectionFactory();//创建工厂
        factory.setHost("localhost");//设置ip
        Connection connection = factory.newConnection();//创建连接
        Channel channel = connection.createChannel();//创建信道

        channel.queueDeclare(RPC_QUEUE_NAME,false, false, false, null);//声明队列

        channel.basicQos(1);//限制：每次最多给一个消费者发送1条消息

        QueueingConsumer consumer = new QueueingConsumer(channel);//为信道创建消费者
        channel.basicConsume(RPC_QUEUE_NAME, false, consumer);//把消费者和队列绑定

        System.out.println("等待请求...");
        while (true) {
            QueueingConsumer.Delivery delivery = consumer.nextDelivery();

            //获取请求中的correlationId属性值，并将其设置到结果消息的correlationId属性中
            BasicProperties props = delivery.getProperties();
            BasicProperties replyProps = new BasicProperties
                .Builder()
                .correlationId(props.getCorrelationId())
                .build();

            String callQueueName = props.getReplyTo();//获取回调队列名字
            String message = new String(delivery.getBody(),"UTF-8");//获取接收到的请求内容
            System.out.println("接收到的请求：" + message);

            String response = "处理的结果";//设置返回结果

            channel.basicPublish("", callQueueName, replyProps,response.getBytes());//发送处理后的结果

            channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);//后手动发送消息反馈
        }

    }

}