java - 如何创建消费者生产者队列

Question

我有一个生产者，它产生一个带有属性类型的 POJO。只能有两种类型，“A”和“B”。我有一个消费者线程池。每当我从 Producer 收到类型为“B”的消息时，在继续执行之前，我需要确保池中的所有其他线程都已完成执行（现在是默认的 Thread.sleep）。然后消费者线程应该获取“B”类型的消息并运行它。在此线程运行之前，无法从队列中弹出任何消息。

例子：

class POJO_Message{

String type; //This will contain the type of message "A" or "B"

}

Answer 1

你可以使用ReadWriteLock来做你的工作。当消息类型为'B'时，尝试获取写锁，其他类型的消息获取读锁。这样一个简单的代码。

public class ConsumerProducerQueue {
    ExecutorService executor = Executors.newFixedThreadPool(10);
    private ReadWriteLock lock = new ReentrantReadWriteLock();

    public void addMessage(Message message) {
        if ("B".equals(message.getType())) {
            lock.writeLock().lock();
            Future<?> result = executor.submit(new Task(message));
            try {
                result.get();
            } catch (Exception e) {

            } finally {
                lock.writeLock().unlock();
            }
        } else {
            lock.readLock().lock();
            Future<?> result = executor.submit(new Task(message));
            try {
                result.get();
            } catch (Exception e) {

            } finally {
                lock.readLock().unlock();
            }
        }
    }
}

这种方法的性能不好。

Answer 2

您可以使用LinkedBlockingDeque。一个例子：

public class ProducerConsumer {

    public static void main(String[] args) {

        final LinkedBlockingDeque<Message> queue = new LinkedBlockingDeque<>(10);

        final AtomicLong id = new AtomicLong(0);
        final Timer producer = new Timer(true);
        producer.scheduleAtFixedRate(new TimerTask() {
            @Override
            public void run() {
               queue.add(new Message(  String.format("msg: %s"  , id.incrementAndGet() ) ) );
            }
        }, 10, 10);

        // consume
        for(;;) {
            try {
                Message msg  = queue.take();
                System.out.println( msg );
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }
        }

    }

    private static class Message {
        private final String id;

        public Message(String id) {
            this.id = id;
        }

        public String getId() {
            return id;
        }

        @Override
        public String toString() {
            return String.format("Message [id=%s]", id);
        }

    }

}