Question

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

示例：

class POJO_Message{

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

}

Answer 1

您可以使用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);
        }

    }

}

Answer 2

您可以使用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();
            }
        }
    }
}

这种方法的性能不好。

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

2 个答案: