温馨提示×

java多线程共享队列怎么做

小亿
109
2023-12-11 11:43:44
栏目: 编程语言

在Java中,可以使用java.util.concurrent包中的BlockingQueue实现多线程共享队列。

BlockingQueue是一个线程安全的队列,它提供了多种方法来处理并发访问的问题。常用的方法有:

  • put(E e): 将元素添加到队列的尾部,如果队列已满则阻塞等待。
  • take(): 移除并返回队列头部的元素,如果队列为空则阻塞等待。
  • offer(E e): 将元素添加到队列的尾部,如果队列已满则返回false
  • poll(): 移除并返回队列头部的元素,如果队列为空则返回null

以下是一个简单的示例代码:

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;

public class SharedQueueExample {
    public static void main(String[] args) {
        // 创建一个容量为10的阻塞队列
        BlockingQueue<Integer> queue = new LinkedBlockingQueue<>(10);

        // 创建并启动多个生产者线程
        for (int i = 0; i < 5; i++) {
            new Producer(queue).start();
        }

        // 创建并启动一个消费者线程
        new Consumer(queue).start();
    }

    static class Producer extends Thread {
        private final BlockingQueue<Integer> queue;

        public Producer(BlockingQueue<Integer> queue) {
            this.queue = queue;
        }

        @Override
        public void run() {
            try {
                for (int i = 0; i < 10; i++) {
                    queue.put(i); // 阻塞等待直到队列有空闲位置
                    System.out.println("Produced: " + i);
                    Thread.sleep(1000);
                }
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    static class Consumer extends Thread {
        private final BlockingQueue<Integer> queue;

        public Consumer(BlockingQueue<Integer> queue) {
            this.queue = queue;
        }

        @Override
        public void run() {
            try {
                while (true) {
                    int value = queue.take(); // 阻塞等待直到队列有可用元素
                    System.out.println("Consumed: " + value);
                    Thread.sleep(2000);
                }
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

在上面的示例中,我们创建了一个容量为10的阻塞队列queue,并创建了5个生产者线程和1个消费者线程。每个生产者线程会不断向队列中添加元素,而消费者线程会不断从队列中取出元素进行处理。

通过使用BlockingQueue,我们可以避免手动实现线程同步和等待/通知机制,简化了多线程编程的复杂性。

0