rabbitmq rpc java

http://www.tuicool.com/articles/iIreEvE

这次试着用RabbitMQ进行RPC。

其实用RabbitMQ搞RPC也没什么特别的。

只是我们需要在请求中再加入一个callback queue。

比如这样：

callbackQueueName = channel.queueDeclare().getQueue();

BasicProperties props = new BasicProperties
.Builder()
.replyTo(callbackQueueName)
.build();

channel.basicPublish("", "rpc_queue", props, message.getBytes());

剩下的工作就是等待对方处理完成再从callback队列中读取响应消息。

上面用到了BasicProperties。

(注意：是 com.rabbitmq.client.AMQP.BasicProperties 不是 com.rabbitmq.client.BasicProperties )

关于Message properties，AMQP协议为消息预定义了14种属性。

private String contentType;
private String contentEncoding;
private Map<String,Object> headers;
private Integer deliveryMode;
private Integer priority;
private String correlationId;
private String replyTo;
private String expiration;
private String messageId;
private Date timestamp;
private String type;
private String userId;
private String appId;
private String clusterId;

通常我们只需要使用其中一小部分：

・deliveryMode: 将消息设置为持久或者临时，2为持久，其余为临时。

・contentType: 指定mime-type，比如要使用JSON就是application/json

・replyTo: 指定callback queue的名字

・correlationId: 用来关联RPC请求和响应的标识。

上面那段代码中就是用到了correlationId。

另外需要说明这个 correlationId 。

其实在上面的代码中我们为每一个RPC请求都创建了一个回调队列。

但这样明显不效率，我们可以为每一个客户端只创建一个回调队列。

但这样我们又需要考虑另一个问题：<当我们将收到的消息放到队列时，如何确定该消息是属于哪个请求？>

这时我们可以使用correlationId解决这个问题。

我们可以用它来为每一个请求加上标识，获取信息时对比这个标识，以对应请求和响应。

如果我们收到了无法识别的correlationId，即该响应不与任何请求匹配，那么这个消息将会废除。



好了，代码比较简单。

RPCServer：

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("localhost");

Connection connection = factory.newConnection();
Channel channel = connection.createChannel();

channel.queueDeclare(RPC_QUEUE_NAME, false, false, false, null);

channel.basicQos(1);

QueueingConsumer consumer = new QueueingConsumer(channel);
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());
int n = Integer.parseInt(message);

System.out.println(" [.] fib(" + message + ")");
String response = "" + fib(n);

channel.basicPublish( "", props.getReplyTo(), replyProps, response.getBytes());

channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
}

}

private static int fib(int n) throws Exception {
if (n == 0) return 0;
if (n == 1) return 1;
return fib(n-1) + fib(n-2);
}
}

由于是共享队列，这里我们就不用exchange和routing了。

另外，有时我们可能需要运行多个服务，为了让多个服务端负载均衡，我们可以使用prefetchCount。

这个属性在之前任务队列的例子里也用过，也就是

workerChannel.basicQos(1);

即让多个worker一次获取一个任务。

用basicConsume方法进入队列后循环等待请求，发现有请求到达时根据队列和CorrelationId对相应请求作出响应。

另外需要注意的一点，server中basicConsume的第二个参数是false。

其意义为是否自动作出回应，即：

true if the server should consider messages acknowledged once delivered; false if the server should expect explicit acknowledgements

于是循环时需要显示调用basicAck进行回应。

RPCClient：

class RPCClient{

private Connection connection;
private Channel channel;
private String requestQueueName = "rpc_queue";
private String replyQueueName;
private QueueingConsumer consumer;

public RPCClient() throws Exception {
ConnectionFactory factory = new ConnectionFactory();
factory.setHost("localhost");
connection = factory.newConnection();
channel = connection.createChannel();

replyQueueName = channel.queueDeclare().getQueue();
consumer = new QueueingConsumer(channel);
channel.basicConsume(replyQueueName, true, consumer);
}

public String call(String message) throws Exception {
String response = null;
String corrId = java.util.UUID.randomUUID().toString();

BasicProperties props = new BasicProperties
.Builder()
.correlationId(corrId)
.replyTo(replyQueueName)
.build();

channel.basicPublish("", requestQueueName, props, message.getBytes());

while (true) {
QueueingConsumer.Delivery delivery = consumer.nextDelivery();
if (delivery.getProperties().getCorrelationId().equals(corrId)) {
response = new String(delivery.getBody());
break;
}
}

return response;
}

public void close() throws Exception {
connection.close();
}
}

callback队列只是一个匿名队列，但切记需要将其设置到BasicProperties中。

corrId的生成方法有很多种，在这里使用UUID。

call方法中通过调用basicPublish进行RPC请求，参数中带着BasicProperties。