RabbitMQ实现RPC(java)

如果我们需要在远程计算机上运行一个函数并等待结果,这种模式通常被称为远程过程调用或RPC。

在本教程中,我们将使用RabbitMQ构建一个RPC系统:一个客户端和一个RPC服务器。我们将创建一个返回斐波那契数字的模拟RPC服务。

整个过程示意图如下:
这里写图片描述
客户端将请求发送至rpc_queue(我们定义的消息队列),然后等待响应;服务端获取请求,并处理请求,然后将请求结果返回给队列,客户端得知请求被响应后获取结果。

在结果被响应之前,客户端是被阻塞的,主线程会等待RPC响应

如果每个RPC请求都创建一个回调队列。这是非常低效,我们创建一个单一的客户端回调队列。

这引发了一个新的问题,在该队列中收到回复时,不清楚回复属于哪个请求。这就需要用到 correlationId属性。我们为没有请求设置唯一的correlationId值。然后,当我们在回调队列中收到一条消息时,我们将获取这个值,将响应与请求的进行correlationId匹配。如果我们一致就是我们需要的结果,否则就不是。

客户端代RPCClient 码如下:

package com.adtec.rabbitmq;

import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.DefaultConsumer;
import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.Envelope;

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

public class RPCClient {
   

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

    public RPCClient() throws IOException, TimeoutException {
        //建立一个连接和一个通道,并为回调声明一个唯一的'回调'队列
        ConnectionFactory factory = new ConnectionFactory();
        factory.setHost("localhost");

        connection = factory.newConnection();
        channel = connection.createChannel();
        //定义一个临时变量的接受队列名    
        replyQueueName = channel.queueDeclare().getQueue();
    }
    //发送RPC请求  
    public String call
  • 2
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
以下是通过 RabbitMQ 调用 Celery 任务的 Java 代码示例: ```java import java.io.IOException; import com.rabbitmq.client.Channel; import com.rabbitmq.client.Connection; import com.rabbitmq.client.ConnectionFactory; import com.rabbitmq.client.QueueingConsumer; public class RabbitMQClient { private final static String QUEUE_NAME = "celery"; public static void main(String[] argv) throws Exception { ConnectionFactory factory = new ConnectionFactory(); factory.setHost("localhost"); Connection connection = factory.newConnection(); Channel channel = connection.createChannel(); channel.queueDeclare(QUEUE_NAME, false, false, false, null); System.out.println(" [*] Waiting for messages. To exit press CTRL+C"); QueueingConsumer consumer = new QueueingConsumer(channel); channel.basicConsume(QUEUE_NAME, true, consumer); while (true) { QueueingConsumer.Delivery delivery = consumer.nextDelivery(); String message = new String(delivery.getBody()); System.out.println(" [x] Received '" + message + "'"); // 调用 Celery 任务 CeleryClient.runTask(message); } } } ``` 在这个示例中,我们使用 RabbitMQJava 客户端库创建了一个消费者,等待来自名为 "celery" 的队列的消息。当收到消息时,我们将其传递给 Celery 客户端的 runTask() 方法进行处理。 以下是 Celery 客户端的示例代码: ```java import io.github.cdimascio.dotenv.Dotenv; import io.github.cdimascio.dotenv.DotenvBuilder; import org.json.JSONObject; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.util.HashMap; import java.util.Map; import java.util.UUID; import com.rabbitmq.client.Channel; import com.rabbitmq.client.Connection; import com.rabbitmq.client.ConnectionFactory; public class CeleryClient { private static final Logger logger = LoggerFactory.getLogger(CeleryClient.class); private static final String TASK_QUEUE_NAME = "celery"; private static final String APP_NAME = "myapp"; private static final String DEFAULT_EXCHANGE = "celery"; private static final String DEFAULT_ROUTING_KEY = "celery"; private static final String DEFAULT_BACKEND = "rpc://"; private static final String DEFAULT_RESULT_EXPIRES = "3600"; private static final String DEFAULT_TASK_TRACK_STARTED = "true"; private static final String DEFAULT_TASK_TIME_LIMIT = "600"; private static final String DEFAULT_TASK_SOFT_TIME_LIMIT = "600"; private static final Dotenv dotenv = new DotenvBuilder().ignoreIfMissing().load(); private static final String CELERY_BROKER_URL = dotenv.get("CELERY_BROKER_URL"); private static final String CELERY_RESULT_BACKEND = dotenv.get("CELERY_RESULT_BACKEND"); public static void runTask(String message) { try { JSONObject json = new JSONObject(message); String taskName = json.getString("task"); JSONObject args = json.getJSONObject("args"); ConnectionFactory factory = new ConnectionFactory(); factory.setUri(CELERY_BROKER_URL); try (Connection connection = factory.newConnection()) { try (Channel channel = connection.createChannel()) { String taskId = UUID.randomUUID().toString(); Map<String, Object> headers = new HashMap<>(); headers.put("task", taskName); headers.put("id", taskId); headers.put("app", APP_NAME); headers.put("lang", "java"); headers.put("taskset", "celery"); headers.put("shadow", null); headers.put("eta", null); headers.put("expires", DEFAULT_RESULT_EXPIRES); headers.put("group", null); headers.put("retries", null); headers.put("timelimit", new int[]{Integer.parseInt(DEFAULT_TASK_TIME_LIMIT), Integer.parseInt(DEFAULT_TASK_SOFT_TIME_LIMIT)}); headers.put("callbacks", null); headers.put("errbacks", null); headers.put("reply_to", null); headers.put("correlation_id", null); headers.put("delivery_info", null); headers.put("kwargsrepr", "{}"); headers.put("utc", true); channel.basicPublish(DEFAULT_EXCHANGE, DEFAULT_ROUTING_KEY, null, args.toString().getBytes("UTF-8"), headers); logger.info("Task sent: " + taskName + " - " + taskId); } } } catch (Exception e) { logger.error("Error while sending task to Celery", e); } } } ``` 在这个示例中,我们使用 Celery Java 客户端库创建了一个方法,接收 RabbitMQ 消息并将其作为 Celery 任务发送。我们使用 dotenv 库从环境变量中读取 Celery 的 URL 和结果后端 URL,并使用这些值创建 Celery 连接。 注意,这里的 runTask() 方法只是将消息发送到 Celery,而实际任务的执行是在 Celery 服务器上完成的。因此,需要在 Celery 服务器上配置任务并启动 Celery 工作人员。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值