/**
* @param pid PID of the other party.
* @param size Size of the space.
* @return Token pair.
*/
private IgnitePair<String> inOutToken(int pid, int size) {
while (true) {
long idx = tokIdxGen.get();
if (tokIdxGen.compareAndSet(idx, idx + 2))
return F.pair(
new File(tokDir, TOKEN_FILE_NAME + idx + "-" + pid + "-" + size).getAbsolutePath(),
new File(tokDir, TOKEN_FILE_NAME + (idx + 1) + "-" + pid + "-" + size)
.getAbsolutePath());
}
}
private void processResponse(DataInputStream in) throws IOException {
long requestId = in.readLong();
Request r = requests.remove(requestId);
if (r == null) {
throw new IllegalStateException(
"Request " + requestId + " is unknown (last request generated was " + nextRequest.get());
}
Object o = null;
if (in.readBoolean()) {
o = serializerFor(classForName(in.readUTF()), r.getResultDeclaredType()).deserialize(in);
}
r.set(o);
}
/** Should request -1 for infinite */
@Test
public void testRequestFromFinalSubscribeWithoutRequestValue() {
TestSubscriber<String> s = new TestSubscriber<>();
final AtomicLong r = new AtomicLong();
s.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
r.set(n);
}
@Override
public void cancel() {}
});
assertEquals(Long.MAX_VALUE, r.get());
}
@Test
public void testRequestFromChainedOperator() {
TestSubscriber<String> s = new TestSubscriber<>();
Operator<String, String> o =
s1 ->
new Subscriber<String>() {
@Override
public void onSubscribe(Subscription a) {
s1.onSubscribe(a);
}
@Override
public void onComplete() {}
@Override
public void onError(Throwable e) {}
@Override
public void onNext(String t) {}
};
s.request(10);
Subscriber<? super String> ns = o.apply(s);
final AtomicLong r = new AtomicLong();
// set set the producer at the top of the chain (ns) and it should flow through the operator to
// the (s) subscriber
// and then it should request up with the value set on the final Subscriber (s)
ns.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
r.set(n);
}
@Override
public void cancel() {}
});
assertEquals(10, r.get());
}
@Test
public void testRequestToObservable() {
TestSubscriber<Integer> ts = new TestSubscriber<>();
ts.request(3);
final AtomicLong requested = new AtomicLong();
Observable.<Integer>create(
s ->
s.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
requested.set(n);
}
@Override
public void cancel() {}
}))
.subscribe(ts);
assertEquals(3, requested.get());
}
@Test
public void testRequestThroughTakeWhereRequestIsSmallerThanTake() {
TestSubscriber<Integer> ts = new TestSubscriber<>((Long) null);
ts.request(3);
final AtomicLong requested = new AtomicLong();
Observable.<Integer>create(
s ->
s.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
requested.set(n);
}
@Override
public void cancel() {}
}))
.take(10)
.subscribe(ts);
assertEquals(3, requested.get());
}
@Test
public void testRequestThroughTakeThatReducesRequest() {
TestSubscriber<Integer> ts = new TestSubscriber<>((Long) null);
ts.request(3);
final AtomicLong requested = new AtomicLong();
Observable.<Integer>create(
s ->
s.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
requested.set(n);
}
@Override
public void cancel() {}
}))
.take(2)
.subscribe(ts);
// FIXME the take now requests Long.MAX_PATH if downstream requests at least the limit
assertEquals(Long.MAX_VALUE, requested.get());
}
public AtomicLong combine(AtomicLong x, AtomicLong y) {
// Not clear whether this is meaningful:
return new AtomicLong(x.addAndGet(y.get()));
}
@Test
public void testRequestFromDecoupledOperatorThatRequestsN() {
TestSubscriber<String> s = new TestSubscriber<>();
final AtomicLong innerR = new AtomicLong();
Operator<String, String> o =
child -> {
// we want to decouple the chain so set our own Producer on the child instead of it coming
// from the parent
child.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
innerR.set(n);
}
@Override
public void cancel() {}
});
AsyncObserver<String> as =
new AsyncObserver<String>() {
@Override
protected void onStart() {
// we request 99 up to the parent
request(99);
}
@Override
public void onComplete() {}
@Override
public void onError(Throwable e) {}
@Override
public void onNext(String t) {}
};
return as;
};
s.request(10);
Subscriber<? super String> ns = o.apply(s);
final AtomicLong r = new AtomicLong();
// set set the producer at the top of the chain (ns) and it should flow through the operator to
// the (s) subscriber
// and then it should request up with the value set on the final Subscriber (s)
ns.onSubscribe(
new Subscription() {
@Override
public void request(long n) {
r.set(n);
}
@Override
public void cancel() {}
});
assertEquals(99, r.get());
assertEquals(10, innerR.get());
}
/**
* @param time Test time.
* @param writers Number of writers threads.
* @throws Exception If failed.
*/
public void testQueue(long time, int writers) throws Exception {
System.out.println("Start test [writers=" + writers + ", time=" + time + "]");
Thread rdr =
new Thread() {
@SuppressWarnings({"InfiniteLoopStatement", "unchecked"})
@Override
public void run() {
try {
while (true) {
Future fut;
while ((fut = queue.poll()) != null) {
qSize.decrementAndGet();
fut.onDone(true);
}
long size = qSize.get();
if (size == 0) {
synchronized (mux) {
while (queue.isEmpty()) {
mux.wait();
}
}
}
}
} catch (InterruptedException ignore) {
}
}
};
rdr.start();
List<Thread> putThreads = new ArrayList<>();
for (int i = 0; i < writers; i++) {
putThreads.add(
new Thread() {
@SuppressWarnings("CallToNotifyInsteadOfNotifyAll")
@Override
public void run() {
try {
while (!stop) {
long id = cnt.incrementAndGet();
Future<Boolean> fut = new Future<Boolean>(new Message(id));
queue.offer(fut);
long size = qSize.incrementAndGet();
if (size == 1) {
synchronized (mux) {
mux.notify();
}
}
Boolean res = fut.get();
if (!res) {
System.out.println("Error");
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
});
}
for (Thread t : putThreads) t.start();
Thread.sleep(time);
stop = true;
for (Thread t : putThreads) t.join();
rdr.interrupt();
rdr.join();
System.out.println("Total: " + cnt.get());
System.gc();
System.gc();
System.gc();
}