6

我有几个使用 ArrayBlockingQueue 的工人。

每个工作人员从队列中取出一个对象，对其进行处理，结果可以获得多个对象，这些对象将被放入队列中以进行进一步处理。所以，工人=生产者+消费者。

工人：

public class Worker implements Runnable
{
    private BlockingQueue<String> processQueue = null;

    public Worker(BlockingQueue<String> processQueue)
    {
        this.processQueue = processQueue;
    }

    public void run()
    {
        try
        {
            do
            {
                String item = this.processQueue.take();
                ArrayList<String> resultItems = this.processItem(item);

                for(String resultItem : resultItems)
                {
                    this.processQueue.put(resultItem);
                }
            }
            while(true);
        }
        catch(Exception)
        {
            ...
        }
    }

    private ArrayList<String> processItem(String item) throws Exception
    {
        ...
    }
}

主要的：

public class Test
{
    public static void main(String[] args) throws Exception
    {
        new Test().run();
    }

    private void run() throws Exception
    {
        BlockingQueue<String> processQueue = new ArrayBlockingQueue<>(10000);
        processQueue.put("lalala");

        Executor service = Executors.newFixedThreadPool(100);
        for(int i=0; i<100; ++i)
        {
            service.execute(new Worker(processQueue));
        }
    }
}

当没有更多工作时，停止工人的最佳方法是什么？

首先，我的想法是定期检查队列中有多少项目以及当前正在处理的项目数量。如果两者都为零，则在 ExecutorService 上执行类似“shutdownNow()”的操作。但我不确定这是最好的方法。

4

3 回答 3

2

如果没有更多工作要做，请在队列中输入一条消息，这样说，并让工作人员在方便时自行关闭。这是防止数据损坏的好方法。

如果您需要通知另一个线程所有工作人员都已回家，您可以使用 aCountDownLatch来执行此操作。

于 2012-04-02T21:08:27.200 回答
1

听起来您有自己的解决方案——使用单独的进行中队列，其大小将是当前正在处理的项目数。如果您使用访问任一队列的约定是在synchronized(theArrayBlockingQueue)块中，那么一切都应该很好。特别是，当将项目移动到处理状态时，将其从 ArrayBlockingQueue 中移除，并将其添加到同一同步块内的 processingQueue 中。

于 2012-04-02T21:16:42.620 回答
0

我稍微修改了您的代码，不确定这是否是您所期望的，但至少它终止了！如果你使用shutdownNow而不是shutdown你的工人将被打断，除非你让他们重新开始工作，否则将退出而不保证队列是空的。

public class Test {

    public static void main(String[] args) throws Exception {
        new Test().run();
    }

    private void run() throws Exception {
        BlockingQueue<String> processQueue = new ArrayBlockingQueue<>(10000);
        processQueue.put("lalalalalalalalalalalalala"); //a little longer to make sure there is enough to process

        ExecutorService service = Executors.newFixedThreadPool(100);
        for (int i = 0; i < 100; ++i) {
            service.execute(new Worker(processQueue));
        }
        service.shutdown(); //orderly shutdown = lets the tasks terminate what they are doing
        service.awaitTermination(1, TimeUnit.SECONDS); //blocks until all tasks have finished or throws TimeOutException if timeout is reached
    }

    public static class Worker implements Runnable {

        private BlockingQueue<String> processQueue = null;
        private int count = 0;

        public Worker(BlockingQueue<String> processQueue) {
            this.processQueue = processQueue;
        }

        @Override
        public void run() {
            try {
                do {
                    //tries to get something from the queue for 100ms and returns null if it could not get anything
                    String item = this.processQueue.poll(100, TimeUnit.MILLISECONDS);
                    if (item == null) break; //Ends the job because the queue was empty
                    count++;
                    List<String> resultItems = this.processItem(item);

                    for (String resultItem : resultItems) {
                        this.processQueue.put(resultItem);
                    }
                } while (true);
            } catch (InterruptedException e) {
                System.out.println("Interrupted");
                Thread.currentThread().interrupt();
            }
            if (count != 0) System.out.println(Thread.currentThread() + ": processed " + count + " entries");
        }

        private List<String> processItem(String item) { //let's put the string back less final character
            if (item.isEmpty()) {
                return Collections.<String> emptyList();
            } else {
                return Arrays.asList(item.substring(0, item.length() - 1));
            }
        }
    }
}
于 2012-04-04T08:55:33.393 回答