RabbitMQ实现异步及同步RPC

编程语言 2018-05-02 19:43:37 阅读次数: 5

一、同步RPC
客户端：

package com.rabbitmq.synchronization;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.UUID;
import java.util.concurrent.TimeoutException;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.ConsumerCancelledException;
import com.rabbitmq.client.DefaultConsumer;
import com.rabbitmq.client.Envelope;
import com.rabbitmq.client.QueueingConsumer;
import com.rabbitmq.client.ShutdownSignalException;
import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.AMQP.BasicProperties;
import com.rabbitmq.utility.BlockingCell;
import com.xinwei.rabbitmq.RabbitConfig;

public class RPCClient {

    private final Map<String,BlockingCell<Object>> _contibuationMap = new HashMap<String, BlockingCell<Object>>();
    private int correlationId = 0;
    private Connection connection;
    private Channel channel;
    private String requestQueuename = "rpc_queue";
    private String replyQueueName;
    private DefaultConsumer _consumer;
    public RPCClient() throws IOException, TimeoutException{
        ConnectionFactory factory =new ConnectionFactory();
        factory.setHost(RabbitConfig.ip);
        factory.setPort(RabbitConfig.port);
        factory.setUsername(RabbitConfig.username);
        factory.setPassword(RabbitConfig.password);

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

        replyQueueName = channel.queueDeclare().getQueue();
        _consumer = setupConsumer();
    }
    public String call(String message) throws IOException, 
        ShutdownSignalException, ConsumerCancelledException,
        InterruptedException, TimeoutException{
        if(_consumer == null)
            return null;
        BlockingCell<Object> k = new BlockingCell<Object>();
        BasicProperties props;
        synchronized (_contibuationMap) {
            correlationId++;
            String corrId = "" + correlationId;
            props = new BasicProperties.Builder()
                    .correlationId(corrId)
                    .replyTo(replyQueueName)
                    .build();
            _contibuationMap.put(corrId, k);
        }

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

        Object reply = k.uninterruptibleGet(4000);
        if (reply instanceof ShutdownSignalException) {
            ShutdownSignalException sig = (ShutdownSignalException) reply;
            ShutdownSignalException wrapper = new ShutdownSignalException(
                    sig.isHardError(), sig.isInitiatedByApplication(),
                    sig.getReason(), sig.getReference());
            wrapper.initCause(sig);
            throw wrapper;
        } else {
            return (String)reply;
        }
    }
    public void close() throws IOException{
        connection.close();
    }

    protected DefaultConsumer setupConsumer() throws IOException {
        DefaultConsumer consumer = new DefaultConsumer(channel) {
            @Override
            public void handleShutdownSignal(String consumerTag,
                    ShutdownSignalException signal) {
                synchronized (_contibuationMap) {
                    for (Entry<String, BlockingCell<Object>> entry : _contibuationMap
                            .entrySet()) {
                        entry.getValue().set(signal);
                    }
                    _consumer = null;
                }
            }
            @Override
            public void handleDelivery(String consumerTag, Envelope envelope,
                    AMQP.BasicProperties properties, byte[] body)
                    throws IOException {
                synchronized (_contibuationMap) {
                    String replyId = properties.getCorrelationId();
                    BlockingCell<Object> blocker = _contibuationMap
                            .get(replyId);
                    _contibuationMap.remove(replyId);

                    if (null != body) {
                        String replyBoay = new String(body, "UTF-8");
                        blocker.set(replyBoay);
                    } else {
                        blocker.set(body);
                    }
                }
            }
        };
        channel.basicConsume(replyQueueName, true, consumer);
        return consumer;
    }

    public static void main(String[] args) throws Exception {
        RPCClient fibrpc =  new RPCClient();
        System.out.println("request fib(30)");
        String response =  fibrpc.call("30");
        System.out.println("got "+response+",");
        fibrpc.close();
    }
}

服务器端

package com.rabbitmq.synchronization;

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;

public class RPCServer {

    public static final String RPC_QUENE_NAME = "rpc_queue";
    public static void main(String[] args) 
            throws Exception {
        ConnectionFactory factory = new ConnectionFactory();
        factory.setHost(RPCConfig.ip);
        factory.setPort(RPCConfig.port);
        factory.setUsername(RPCConfig.username);
        factory.setPassword(RPCConfig.password);

        Connection connection = factory.newConnection();
        Channel channel = connection.createChannel();
        channel.queueDeclare(RPC_QUENE_NAME, false, false, false, null);
        channel.basicQos(1);
        QueueingConsumer consumer = new QueueingConsumer(channel);
        channel.basicConsume(RPC_QUENE_NAME, false,consumer);
        System.out.println("Awaiting RPCrequests");
        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);
    }
}

二、异步RPC
客户端

package com.rabbitmq;

import java.io.IOException;
import java.util.UUID;
import java.util.concurrent.TimeoutException;

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.ConsumerCancelledException;
import com.rabbitmq.client.QueueingConsumer;
import com.rabbitmq.client.ShutdownSignalException;
import com.rabbitmq.utility.BlockingCell;

public class RPCClient {

    private Connection connection;
    private Channel channel;
    private String requestQueuename = "rpc_queue";
    private String replyQueueName;
    private QueueingConsumer consumer;
    public RPCClient() throws IOException, TimeoutException{
        ConnectionFactory factory =new ConnectionFactory();
        factory.setHost(RabbitConfig.ip);
        factory.setPort(RabbitConfig.port);
        factory.setUsername(RabbitConfig.username);
        factory.setPassword(RabbitConfig.password);

        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 IOException, 
        ShutdownSignalException, ConsumerCancelledException,
        InterruptedException, TimeoutException{
        String response = null;
        String corrId = UUID.randomUUID().toString();
        BasicProperties props = new BasicProperties.Builder()
                .correlationId(corrId)
                .replyTo(replyQueueName)
                .build();
        channel.basicPublish("", requestQueuename, props, message.getBytes());

        BlockingCell<Object> k = new BlockingCell<Object>();
        Object reply = k.uninterruptibleGet();
        System.out.println("reply:"+reply.toString());

        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 IOException{
        connection.close();
    }
    public static void main(String[] args) throws Exception {
        RPCClient fibrpc =  new RPCClient();
        System.out.println("request fib(30)");
        String response =  fibrpc.call("30");
        System.out.println("got "+response+",");
        fibrpc.close();

    }
}

服务器端

package com.rabbitmq;

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;

public class RPCServer {

    public static final String RPC_QUENE_NAME = "rpc_queue";
    public static void main(String[] args) 
            throws Exception {
        ConnectionFactory factory = new ConnectionFactory();
        factory.setHost(RabbitConfig.ip);
        factory.setPort(RabbitConfig.port);
        factory.setUsername(RabbitConfig.username);
        factory.setPassword(RabbitConfig.password);

        Connection connection = factory.newConnection();
        Channel channel = connection.createChannel();
        channel.queueDeclare(RPC_QUENE_NAME, false, false, false, null);
        channel.basicQos(1);
        QueueingConsumer consumer = new QueueingConsumer(channel);
        channel.basicConsume(RPC_QUENE_NAME, false,consumer);
        System.out.println("Awaiting RPCrequests");
        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);
    }
}

猜你喜欢

转载自blog.csdn.net/addisonjing/article/details/71249228
今日推荐
Linus “吃狗粮”最积极！
开源日报 | Winamp播放器即将开源；生成式AI之战升级第二轮；Linus“吃狗粮”最积极；AI进入泡沫前期；吴泳铭为阿里云带来了什么？
NetBSD 禁止提交由 AI 生成的代码
Apache Doris 2.0.10 版本正式发布！
开源日报 | 大模型开战；大模型独角兽被曝卖身；周鸿祎建议谷歌开源所有产品；最大开源AI社区提供1000万美元共享GPU
开源日报 | Chrome内置Gemini的意义不在于Gemini；中国AI追随之路的五大误区；ECharts创始人“下海”养鱼；谷歌I/O开发者大会什么都有，只是没有惊喜
微软回应中国区AI团队“打包赴美”传闻
周排行
SVN服务端安装在阿里云
实战 | 相机标定
webpack核心概念
note20——》只要肯低头吃苦，人生就会有救
PAT甲级 1062 Talent and Virtue （25 分）排序
NG Toolset开发笔记--5GNR Resource Grid（26）
如何对待上司
oracle命令
第9章 STL迭代器
logstash使用es映射模板
每日归档
更多
2024-05-20(36)
2024-05-19(0)
2024-05-18(4)
2024-05-17(34)
2024-05-16(6)
2024-05-15(24)
2024-05-14(0)
2024-05-13(18)
2024-05-12(0)
2024-05-11(38)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022