--- jsr166/src/test/loops/TimeoutProducerConsumerLoops.java 2009/10/23 19:57:07 1.3 +++ jsr166/src/test/loops/TimeoutProducerConsumerLoops.java 2015/01/15 18:34:19 1.12 @@ -1,11 +1,10 @@ /* * Written by Doug Lea with assistance from members of JCP JSR-166 * Expert Group and released to the public domain, as explained at - * http://creativecommons.org/licenses/publicdomain + * http://creativecommons.org/publicdomain/zero/1.0/ */ import java.util.concurrent.*; -//import jsr166y.*; public class TimeoutProducerConsumerLoops { static final int NCPUS = Runtime.getRuntime().availableProcessors(); @@ -17,10 +16,14 @@ public class TimeoutProducerConsumerLoop static final int POOL_MASK = POOL_SIZE-1; static final Integer[] intPool = new Integer[POOL_SIZE]; static { - for (int i = 0; i < POOL_SIZE; ++i) + for (int i = 0; i < POOL_SIZE; ++i) intPool[i] = Integer.valueOf(i); } + // max lag between a producer and consumer to avoid + // this becoming a GC test rather than queue test. + // Used only per-pair to lessen impact on queue sync + static final int LAG_MASK = (1 << 12) - 1; static boolean print = false; static int producerSum; @@ -43,24 +46,23 @@ public class TimeoutProducerConsumerLoop int maxPairs = NCPUS * 3 / 2; int iters = 1000000; - if (args.length > 0) + if (args.length > 0) maxPairs = Integer.parseInt(args[0]); print = true; - int k = 1; - for (int i = 1; i <= maxPairs;) { + for (int k = 1, i = 1; i <= maxPairs;) { System.out.println("Pairs:" + i); oneTest(i, iters); Thread.sleep(100); if (i == k) { k = i << 1; i = i + (i >>> 1); - } - else + } + else i = k; } pool.shutdown(); - } + } static void oneTest(int n, int iters) throws Exception { if (print) @@ -104,21 +106,23 @@ public class TimeoutProducerConsumerLoop oneRun(new ArrayBlockingQueue(POOL_SIZE, true), n, iters/16); } - - static abstract class Stage implements Runnable { + + abstract static class Stage implements Runnable { final int iters; final BlockingQueue queue; final CyclicBarrier barrier; - Stage (BlockingQueue q, CyclicBarrier b, int iters) { - queue = q; + final Phaser lagPhaser; + Stage(BlockingQueue q, CyclicBarrier b, Phaser s, int iters) { + queue = q; barrier = b; + lagPhaser = s; this.iters = iters; } } static class Producer extends Stage { - Producer(BlockingQueue q, CyclicBarrier b, int iters) { - super(q, b, iters); + Producer(BlockingQueue q, CyclicBarrier b, Phaser s, int iters) { + super(q, b, s, iters); } public void run() { @@ -136,6 +140,8 @@ public class TimeoutProducerConsumerLoop ++i; if (timeout > 1) timeout--; + if ((i & LAG_MASK) == LAG_MASK) + lagPhaser.arriveAndAwaitAdvance(); } else timeout++; @@ -143,16 +149,16 @@ public class TimeoutProducerConsumerLoop addProducerSum(s); barrier.await(); } - catch (Exception ie) { - ie.printStackTrace(); - return; + catch (Exception ie) { + ie.printStackTrace(); + return; } } } static class Consumer extends Stage { - Consumer(BlockingQueue q, CyclicBarrier b, int iters) { - super(q, b, iters); + Consumer(BlockingQueue q, CyclicBarrier b, Phaser s, int iters) { + super(q, b, s, iters); } public void run() { @@ -163,7 +169,7 @@ public class TimeoutProducerConsumerLoop int i = 0; long timeout = 1000; while (i < iters) { - Integer e = queue.poll(timeout, + Integer e = queue.poll(timeout, TimeUnit.NANOSECONDS); if (e != null) { l = LoopHelpers.compute4(e.intValue()); @@ -171,6 +177,8 @@ public class TimeoutProducerConsumerLoop ++i; if (timeout > 1) --timeout; + if ((i & LAG_MASK) == LAG_MASK) + lagPhaser.arriveAndAwaitAdvance(); } else ++timeout; @@ -178,9 +186,9 @@ public class TimeoutProducerConsumerLoop addConsumerSum(s); barrier.await(); } - catch (Exception ie) { - ie.printStackTrace(); - return; + catch (Exception ie) { + ie.printStackTrace(); + return; } } @@ -190,8 +198,9 @@ public class TimeoutProducerConsumerLoop LoopHelpers.BarrierTimer timer = new LoopHelpers.BarrierTimer(); CyclicBarrier barrier = new CyclicBarrier(npairs * 2 + 1, timer); for (int i = 0; i < npairs; ++i) { - pool.execute(new Producer(q, barrier, iters)); - pool.execute(new Consumer(q, barrier, iters)); + Phaser s = new Phaser(2); + pool.execute(new Producer(q, barrier, s, iters)); + pool.execute(new Consumer(q, barrier, s, iters)); } barrier.await(); barrier.await(); @@ -205,12 +214,12 @@ public class TimeoutProducerConsumerLoop static final class LTQasSQ extends LinkedTransferQueue { LTQasSQ() { super(); } public void put(T x) { - try { super.transfer(x); + try { super.transfer(x); } catch (InterruptedException ex) { throw new Error(); } } public boolean offer(T x, long timeout, TimeUnit unit) { - return super.offer(x, timeout, unit); + return super.offer(x, timeout, unit); } }