Java多线程通信问题：生产者与消费者案例

原创左手的ㄟ右手 2024-12-11 01:03 11阅读 0赞

在Java中，我们可以通过`java.util.concurrent`包中的工具来实现生产者和消费者的多线程通信。

下面是一个简单的生产者-消费者模型的例子：

1. 生产者类（Producer.java）：
```java
import java.util.concurrent.*;

public class Producer {
    private final Queue<String> queue;
    private final ExecutorService executor;

public Producer(Queue<String> queue, int threadNum) {
        this.queue = queue;
        this.executor = Executors.newFixedThreadPool(threadNum);
    }

public void produce() throws InterruptedException {
        for (int i = 0; i < 10; i++) { // 生产10个元素
            String element = "Element-" + i; // 构造元素
            executor.submit(() -> queue.put(element))); // 提交任务到线程池
        }
    }

public void shutdown() {
        executor.shutdown();
    }
}
```

2. 消费者类（Consumer.java）：
```java
import java.util.concurrent.*;

public class Consumer {
    private final Queue<String> queue;
    private final ExecutorService executor;

public Consumer(Queue<String> queue, int threadNum) {
        this.queue = queue;
        this.executor = Executors.newFixedThreadPool(threadNum);
    }

public void consume() throws InterruptedException {
        while (!queue.isEmpty()) { // 循环直到队列为空
            String element = queue.poll(); // 取出并移除队列中的元素
            executor.submit(() -> System.out.println("Consumed Element: " + element))); // 提交任务到线程池，打印消费的元素
            // 这里可以根据实际需求添加其他处理逻辑
        }
    }

public void shutdown() {
        executor.shutdown();
    }
}
```

3. 主程序（Main.java）：
```java
import java.util.concurrent.TimeUnit;

public class Main {
    public static void main(String[] args) {
        Queue<String> queue = new LinkedList<>();
        ExecutorService executor = Executors.newFixedThreadPool(2);

// 生产者
        Producer producer = new Producer(queue, 2);
        producer.produce();

// 消费者
        Consumer consumer = new Consumer(queue, 2);
        consumer.consume();

// 线程池关闭
        executor.shutdown();
        try {
            // 等待所有线程完成，阻塞在此
            executor.awaitTermination(10, TimeUnit.SECONDS));
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}
```

以上就是一个简单的生产者-消费者模型的实现。你可以根据实际需求进行扩展和优化。