如何中断在 take() 上阻塞的 BlockingQueue? [英] How to interrupt a BlockingQueue which is blocking on take()?

问题描述

我有一个类从 BlockingQueue 获取对象，并通过在连续循环中调用 take() 来处理它们.在某些时候，我知道不会有更多的对象被添加到队列中.如何中断 take() 方法以使其停止阻塞?

I have a class that takes objects from a BlockingQueue and processes them by calling take() in a continuous loop. At some point I know that no more objects will be added to the queue. How do I interrupt the take() method so that it stops blocking?

这是处理对象的类:

public class MyObjHandler implements Runnable {

  private final BlockingQueue<MyObj> queue;

  public class MyObjHandler(BlockingQueue queue) {
    this.queue = queue;
  }

  public void run() {
    try {
      while (true) {
        MyObj obj = queue.take();
        // process obj here
        // ...
      }
    } catch (InterruptedException e) {
      Thread.currentThread().interrupt();
    }
  }
}

这里是使用这个类来处理对象的方法:

And here's the method that uses this class to process objects:

public void testHandler() {

  BlockingQueue<MyObj> queue = new ArrayBlockingQueue<MyObj>(100);  

  MyObjectHandler  handler = new MyObjectHandler(queue);
  new Thread(handler).start();

  // get objects for handler to process
  for (Iterator<MyObj> i = getMyObjIterator(); i.hasNext(); ) {
    queue.put(i.next());
  }

  // what code should go here to tell the handler
  // to stop waiting for more objects?
}

推荐答案

如果中断线程不是一种选择，另一种方法是在队列中放置一个标记"或命令"对象，以便 MyObjHandler 识别它们并跳出循环.

If interrupting the thread is not an option, another is to place a "marker" or "command" object on the queue that would be recognized as such by MyObjHandler and break out of the loop.

这篇关于如何中断在 take() 上阻塞的 BlockingQueue?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！