Làm cách nào để ngắt một BlockingQueue đang chặn trên take ()?

Tôi có một lớp lấy các đối tượng từ a BlockingQueuevà xử lý chúng bằng cách gọi take()trong một vòng lặp liên tục. Tại một thời điểm nào đó, tôi biết rằng sẽ không có thêm đối tượng nào được thêm vào hàng đợi. Làm cách nào để ngắt take()phương thức để nó ngừng chặn?

Đây là lớp xử lý các đối tượng:

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();
    }
  }
}

Và đây là phương thức sử dụng lớp này để xử lý các đối tượng:

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?
}

Nếu ngắt luồng không phải là một tùy chọn, thì một tùy chọn khác là đặt một đối tượng "điểm đánh dấu" hoặc "lệnh" trên hàng đợi sẽ được MyObjHandler nhận dạng như vậy và thoát ra khỏi vòng lặp.

BlockingQueue<MyObj> queue = new ArrayBlockingQueue<MyObj>(100);
MyObjectHandler handler = new MyObjectHandler(queue);
Thread thread = new Thread(handler);
thread.start();
for (Iterator<MyObj> i = getMyObjIterator(); i.hasNext(); ) {
  queue.put(i.next());
}
thread.interrupt();

Tuy nhiên, nếu bạn làm điều này, chuỗi có thể bị gián đoạn trong khi vẫn còn các mục trong hàng đợi để xử lý. Bạn có thể muốn xem xét sử dụng pollthay vì take, điều này sẽ cho phép luồng xử lý hết thời gian chờ và kết thúc khi nó đã đợi một lúc mà không có đầu vào mới.

Rất muộn nhưng Hy vọng điều này cũng sẽ giúp ích cho những người khác khi tôi gặp phải vấn đề tương tự và sử dụng pollcách tiếp cận được đề xuất bởi erickson ở trên với một số thay đổi nhỏ,

class MyObjHandler implements Runnable 
{
    private final BlockingQueue<MyObj> queue;
    public volatile boolean Finished;  //VOLATILE GUARANTEES UPDATED VALUE VISIBLE TO ALL
    public MyObjHandler(BlockingQueue queue) 
    {
        this.queue = queue;
        Finished = false;
    }
    @Override
    public void run() 
    {        
        while (true) 
        {
            try 
            {
                MyObj obj = queue.poll(100, TimeUnit.MILLISECONDS);
                if(obj!= null)//Checking if job is to be processed then processing it first and then checking for return
                {
                    // process obj here
                    // ...
                }
                if(Finished && queue.isEmpty())
                    return;

            } 
            catch (InterruptedException e) 
            {                   
                return;
            }
        }
    }
}

public void testHandler() 
{
    BlockingQueue<MyObj> queue = new ArrayBlockingQueue<MyObj>(100); 

    MyObjHandler  handler = new MyObjHandler(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?
    handler.Finished = true; //THIS TELLS HIM
    //If you need you can wait for the termination otherwise remove join
    myThread.join();
}

Điều này giải quyết được cả hai vấn đề

1. Gắn cờ BlockingQueueđể biết rằng nó không phải đợi thêm các phần tử
2. Không bị gián đoạn ở giữa để các khối xử lý chỉ kết thúc khi tất cả các mục trong hàng đợi được xử lý và không còn mục nào để thêm

Làm gián đoạn chuỗi:

thread.interrupt()

Hoặc đừng làm gián đoạn, nó khó chịu.

    public class MyQueue<T> extends ArrayBlockingQueue<T> {

        private static final long serialVersionUID = 1L;
        private boolean done = false;

        public ParserQueue(int capacity) {  super(capacity); }

        public void done() { done = true; }

        public boolean isDone() { return done; }

        /**
         * May return null if producer ends the production after consumer 
         * has entered the element-await state.
         */
        public T take() throws InterruptedException {
            T el;
            while ((el = super.poll()) == null && !done) {
                synchronized (this) {
                    wait();
                }
            }

            return el;
        }
    }

1. khi nhà sản xuất đặt đối tượng vào hàng đợi, hãy gọi queue.notify(), nếu nó kết thúc, hãy gọiqueue.done()
2. vòng lặp while (! queue.isDone () ||! queue.isEmpty ())
3. test take () trả về giá trị cho null

Còn về một

queue.add(new MyObj())

trong một số chuỗi nhà sản xuất, nơi cờ dừng báo hiệu chuỗi người tiêu dùng kết thúc vòng lặp trong khi?