18 |
|
|
19 |
|
static final Phaser phaser = new Phaser(CONSUMERS + 1); |
20 |
|
|
21 |
< |
static class Sub implements Flow.Subscriber<Boolean> { |
21 |
> |
static final class Sub implements Flow.Subscriber<Boolean> { |
22 |
|
Flow.Subscription sn; |
23 |
|
int count; |
24 |
< |
public void onSubscribe(Flow.Subscription s) { |
25 |
< |
(sn = s).request(CAP); |
24 |
> |
public void onSubscribe(Flow.Subscription s) { |
25 |
> |
(sn = s).request(CAP); |
26 |
|
} |
27 |
< |
public void onNext(Boolean t) { |
27 |
> |
public void onNext(Boolean t) { |
28 |
|
if ((++count & (CAP - 1)) == (CAP >>> 1)) |
29 |
|
sn.request(CAP); |
30 |
|
} |
31 |
|
public void onError(Throwable t) { t.printStackTrace(); } |
32 |
< |
public void onComplete() { |
32 |
> |
public void onComplete() { |
33 |
|
if (count != ITEMS) |
34 |
|
System.out.println("Error: remaining " + (ITEMS - count)); |
35 |
< |
phaser.arrive(); |
35 |
> |
phaser.arrive(); |
36 |
|
} |
37 |
|
} |
38 |
|
|
42 |
|
int reps = REPS; |
43 |
|
if (args.length > 0) |
44 |
|
reps = Integer.parseInt(args[0]); |
45 |
< |
|
46 |
< |
System.out.println("ITEMS: " + ITEMS + |
45 |
> |
|
46 |
> |
System.out.println("ITEMS: " + ITEMS + |
47 |
|
" CONSUMERS: " + CONSUMERS + |
48 |
|
" CAP: " + CAP); |
49 |
|
ExecutorService exec = ForkJoinPool.commonPool(); |
50 |
< |
for (int rep = 0; rep < reps; ++reps) { |
51 |
< |
long startTime = System.nanoTime(); |
52 |
< |
final SubmissionPublisher<Boolean> pub = |
53 |
< |
new SubmissionPublisher<Boolean>(exec, CAP); |
54 |
< |
for (int i = 0; i < CONSUMERS; ++i) |
55 |
< |
pub.subscribe(new Sub()); |
56 |
< |
for (int i = 0; i < ITEMS; ++i) { |
57 |
< |
pub.submit(Boolean.TRUE); |
58 |
< |
} |
59 |
< |
pub.close(); |
60 |
< |
phaser.arriveAndAwaitAdvance(); |
61 |
< |
long elapsed = System.nanoTime() - startTime; |
62 |
< |
double secs = ((double)elapsed) / NPS; |
63 |
< |
System.out.printf("\tTime: %7.3f\n", secs); |
64 |
< |
// System.out.println(exec); |
50 |
> |
for (int rep = 0; rep < reps; ++rep) { |
51 |
> |
oneRun(exec); |
52 |
> |
System.out.println(exec); |
53 |
|
Thread.sleep(1000); |
54 |
|
} |
55 |
|
if (exec != ForkJoinPool.commonPool()) |
56 |
|
exec.shutdown(); |
57 |
|
} |
58 |
+ |
|
59 |
+ |
static void oneRun(ExecutorService exec) throws Exception { |
60 |
+ |
long startTime = System.nanoTime(); |
61 |
+ |
final SubmissionPublisher<Boolean> pub = |
62 |
+ |
new SubmissionPublisher<Boolean>(exec, CAP); |
63 |
+ |
for (int i = 0; i < CONSUMERS; ++i) |
64 |
+ |
pub.subscribe(new Sub()); |
65 |
+ |
for (int i = 0; i < ITEMS; ++i) { |
66 |
+ |
pub.submit(Boolean.TRUE); |
67 |
+ |
} |
68 |
+ |
pub.close(); |
69 |
+ |
phaser.arriveAndAwaitAdvance(); |
70 |
+ |
long elapsed = System.nanoTime() - startTime; |
71 |
+ |
double secs = ((double)elapsed) / NPS; |
72 |
+ |
System.out.printf("\tTime: %7.3f\n", secs); |
73 |
+ |
} |
74 |
|
} |