http://blog.csdn.net/zhu_tianwei/article/details/40887885
在一般使用RabbitMQ做RPC很容易。客户端发送一个请求消息然后服务器回复一个响应消息。为了收到一个响应,我们需要发送一个'回调'的请求的队列地址。我们可以使用默认队列(在Java客户端除外)。
AMQP协议给消息定义了14个属性。大部分的属性很少使用,除了下面几个:
deliveryMode: 将消息标记为持久(值为2)或瞬态(任何其他值)。你可能记得在第二个教程中使用了这个属性。
contentType:用来设置mime类型。例如经常使用的JSON格式数据,就需要将此属性设置为:application/json。
replyTo: 通常用来命名一个回调队列.
correlationId: 用来关联RPC请求的响应.
RPC工作流程:
1)、客户端启动时,创建了一个匿名的回调队列。
2)、在一个RPC请求中,客户端发送一个消息,它有两个属性:1.REPLYTO,用来设置回调队列名;2.correlationId,对于每个请求都被设置成唯一的值。
3)、请求被发送到rpc_queue队列.
4)、RPC工作者(又名:服务器)等待接收该队列的请求。当收到一个请求,它就会处理并把结果发送给客户端,使用的队列是replyTo字段指定的。
5)、客户端等待接收回调队列中的数据。当接到一个消息,它会检查它的correlationId属性。如果它和设置的相匹配,就会把响应返回给应用程序。
1、RPC服务器的RPCServer.java,接收消息调用rpc并返回结果
- package cn.slimsmart.rabbitmq.demo.rpc;
- import java.security.MessageDigest;
- import com.rabbitmq.client.AMQP;
- 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;
- //RPC调用服务端
- public class RPCServer {
- private static final String RPC_QUEUE_NAME = "rpc_queue";
- public static void main(String[] args) throws Exception {
- //• 先建立连接、通道,并声明队列
- ConnectionFactory factory = new ConnectionFactory();
- factory.setHost("192.168.36.217");
- factory.setUsername("admin");
- factory.setPassword("admin");
- factory.setPort(AMQP.PROTOCOL.PORT);
- Connection connection = factory.newConnection();
- Channel channel = connection.createChannel();
- channel.queueDeclare(RPC_QUEUE_NAME, false, false, false, null);
- //•可以运行多个服务器进程。通过channel.basicQos设置prefetchCount属性可将负载平均分配到多台服务器上。
- channel.basicQos(1);
- QueueingConsumer consumer = new QueueingConsumer(channel);
- //打开应答机制autoAck=false
- channel.basicConsume(RPC_QUEUE_NAME, false, consumer);
- System.out.println(" [x] Awaiting RPC requests");
- while (true) {
- QueueingConsumer.Delivery delivery = consumer.nextDelivery();
- BasicProperties props = delivery.getProperties();
- BasicProperties replyProps = new BasicProperties.Builder()
- .correlationId(props.getCorrelationId()).build();
- String message = new String(delivery.getBody());
- System.out.println(" [.] getMd5String(" + message + ")");
- String response = getMd5String(message);
- //返回处理结果队列
- channel.basicPublish("", props.getReplyTo(), replyProps,
- response.getBytes());
- //发送应答
- channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
- }
- }
- // 模拟RPC方法 获取MD5字符串
- public static String getMd5String(String str) {
- MessageDigest md5 = null;
- try {
- md5 = MessageDigest.getInstance("MD5");
- } catch (Exception e) {
- System.out.println(e.toString());
- e.printStackTrace();
- return "";
- }
- char[] charArray = str.toCharArray();
- byte[] byteArray = new byte[charArray.length];
- for (int i &#61; 0; i < charArray.length; i&#43;&#43;)
- byteArray[i] &#61; (byte) charArray[i];
- byte[] md5Bytes &#61; md5.digest(byteArray);
- StringBuffer hexValue &#61; new StringBuffer();
- for (int i &#61; 0; i < md5Bytes.length; i&#43;&#43;) {
- int val &#61; ((int) md5Bytes[i]) & 0xff;
- if (val < 16)
- hexValue.append("0");
- hexValue.append(Integer.toHexString(val));
- }
- return hexValue.toString();
- }
- }
2.客户端RPCClient.java&#xff0c;发送rpc调用消息&#xff0c;接收结果
- package cn.slimsmart.rabbitmq.demo.rpc;
- import com.rabbitmq.client.AMQP;
- import com.rabbitmq.client.Channel;
- import com.rabbitmq.client.Connection;
- import com.rabbitmq.client.ConnectionFactory;
- import com.rabbitmq.client.QueueingConsumer;
- import com.rabbitmq.client.AMQP.BasicProperties;
- //RPC调用客户端
- public class RPCClient {
- private Connection connection;
- private Channel channel;
- private String requestQueueName &#61; "rpc_queue";
- private String replyQueueName;
- private QueueingConsumer consumer;
- public RPCClient() throws Exception {
- //• 先建立一个连接和一个通道&#xff0c;并为回调声明一个唯一的&#39;回调&#39;队列
- ConnectionFactory factory &#61; new ConnectionFactory();
- factory.setHost("192.168.36.217");
- factory.setUsername("admin");
- factory.setPassword("admin");
- factory.setPort(AMQP.PROTOCOL.PORT);
- connection &#61; factory.newConnection();
- channel &#61; connection.createChannel();
- //• 注册&#39;回调&#39;队列&#xff0c;这样就可以收到RPC响应
- replyQueueName &#61; channel.queueDeclare().getQueue();
- consumer &#61; new QueueingConsumer(channel);
- channel.basicConsume(replyQueueName, true, consumer);
- }
- //发送RPC请求
- public String call(String message) throws Exception {
- String response &#61; null;
- String corrId &#61; java.util.UUID.randomUUID().toString();
- //发送请求消息&#xff0c;消息使用了两个属性&#xff1a;replyto和correlationId
- BasicProperties props &#61; new BasicProperties.Builder()
- .correlationId(corrId).replyTo(replyQueueName).build();
- channel.basicPublish("", requestQueueName, props, message.getBytes());
- //等待接收结果
- while (true) {
- QueueingConsumer.Delivery delivery &#61; consumer.nextDelivery();
- //检查它的correlationId是否是我们所要找的那个
- if (delivery.getProperties().getCorrelationId().equals(corrId)) {
- response &#61; new String(delivery.getBody());
- break;
- }
- }
- return response;
- }
- public void close() throws Exception {
- connection.close();
- }
- }
3、运行client主函数RPCMain.java
- package cn.slimsmart.rabbitmq.demo.rpc;
- public class RPCMain {
- public static void main(String[] args) throws Exception {
- RPCClient rpcClient &#61; new RPCClient();
- System.out.println(" [x] Requesting getMd5String(abc)");
- String response &#61; rpcClient.call("abc");
- System.out.println(" [.] Got &#39;" &#43; response &#43; "&#39;");
- rpcClient.close();
- }
- }
先运行服务端&#xff0c;再运行RPCMain&#xff0c;发送消息调用RPC。
这里介绍的是该设计不是实现RPC服务的唯一可能&#xff0c;但它有一些重要的优点:
1&#xff09;如果RPC服务器速度太慢&#xff0c;你可以通过运行多个RPC服务器。尝试在一个新的控制台上运行第二RPCServer。
2&#xff09;RPC客户端只发送和接收一个消息。不需要queueDeclare那样要求同步调用。因此&#xff0c;RPC客户端只需要在一个网络上发送和接收为一个单一的RPC请求。