1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
| public abstract class TaskQueue<T> {
ExecutorService executor = null; ArrayBlockingQueue<T> queue; private volatile boolean running = false; protected abstract void doWork(T item);
public TaskQueue(ExecutorService executor, ArrayBlockingQueue<T> queue){ this.executor = executor; this.queue = queue; start(); }
public void start() { running = true; executor.submit(new Runnable() { @Override public void run() { processTask(); } });
Runtime.getRuntime().addShutdownHook(new Thread(new Runnable() { @Override public void run() { stop(); } })); } private void processTask() { try { T item = null; while(running){ item = queue.poll(5, TimeUnit.SECONDS); if(item != null){ doWork(item); } } while((item = queue.poll() ) != null){ doWork(item); } } catch (InterruptedException e) { Thread.currentThread().interrupt(); } }
public void stop() { try { running = false;
executor.shutdown(); executor.awaitTermination(30, TimeUnit.SECONDS);
executor = null; } catch (InterruptedException e) { Thread.currentThread().interrupt(); } }
public void put(T item) { try {
queue.put(item); } catch (InterruptedException e) { e.printStackTrace(); } }
}
|