Java 从零开始实现消息队列（MQ-02）：如何实现生产者调用消费者？

后端

2024-01-06 01:03:30

1. 引言

在上一篇文章中，我们介绍了如何启动生产者和消费者。现在，我们将进一步探索如何让生产者调用消费者，从而实现消息的发送和接收。

2. 生产者调用消费者

生产者调用消费者可以通过以下步骤实现：

生产者创建一个 Channel，用于向消费者发送消息。
生产者将要发送的消息写入 Channel。
消费者从 Channel 中读取消息。
消费者处理消息。

3. 代码实现

以下代码展示了生产者如何调用消费者：

public class Producer {

    private Channel channel;

    public Producer(Channel channel) {
        this.channel = channel;
    }

    public void sendMessage(String message) {
        channel.writeAndFlush(message);
    }
}

public class Consumer {

    private Channel channel;

    public Consumer(Channel channel) {
        this.channel = channel;
    }

    public void receiveMessage() {
        channel.read();
    }
}

4. 多线程处理

为了确保生产者和消费者可以同时运行，我们需要使用多线程。以下代码展示了如何使用多线程实现生产者和消费者的并发执行：

public class Main {

    public static void main(String[] args) {
        // 创建 Channel
        Channel channel = NettyServer.createChannel();

        // 创建生产者和消费者
        Producer producer = new Producer(channel);
        Consumer consumer = new Consumer(channel);

        // 启动生产者和消费者
        new Thread(producer).start();
        new Thread(consumer).start();
    }
}