use of org.mule.runtime.api.util.concurrent.Latch in project mule by mulesoft.
the class AbstractTransactionQueueManagerTestCase method testTakePut.
@Test
public void testTakePut() throws Exception {
final QueueManager mgr = createQueueManager();
mgr.start();
final Latch latch = new Latch();
Thread t = new Thread() {
@Override
public void run() {
try {
latch.countDown();
Thread.sleep(200);
QueueSession s = mgr.getQueueSession();
Queue q = s.getQueue("queue1");
assertEquals("Queue size", 0, q.size());
q.put("String1");
} catch (Exception e) {
// ignore, let test fail
}
}
};
t.start();
latch.await();
long t0 = System.currentTimeMillis();
QueueSession s = mgr.getQueueSession();
Queue q = s.getQueue("queue1");
assertEquals("Queue size", 0, q.size());
Object o = q.take();
long t1 = System.currentTimeMillis();
t.join();
assertNotNull(o);
assertEquals("Queue content", "String1", o);
assertEquals("Queue size", 0, q.size());
assertTrue(t1 - t0 > 100);
mgr.stop();
}
use of org.mule.runtime.api.util.concurrent.Latch in project mule by mulesoft.
the class AbstractTransactionQueueManagerTestCase method testTakePutOverCapacity.
@Test
public void testTakePutOverCapacity() throws Exception {
final QueueManager mgr = createQueueManager();
mgr.start();
mgr.setDefaultQueueConfiguration(new DefaultQueueConfiguration(2, false));
final Latch latch = new Latch();
Thread t = new Thread() {
@Override
public void run() {
try {
latch.await();
Thread.sleep(200);
QueueSession s = mgr.getQueueSession();
Queue q = s.getQueue("queue1");
Object o = q.take();
assertEquals("Queue content", "String1", o);
} catch (Exception e) {
// ignore, let test fail
}
}
};
t.start();
QueueSession s = mgr.getQueueSession();
Queue q = s.getQueue("queue1");
assertEquals("Queue size", 0, q.size());
q.put("String1");
q.put("String2");
latch.countDown();
long t0 = System.currentTimeMillis();
q.put("String3");
long t1 = System.currentTimeMillis();
t.join();
assertEquals("Queue size", 2, q.size());
assertTrue(t1 - t0 > 100);
mgr.stop();
}
use of org.mule.runtime.api.util.concurrent.Latch in project mule by mulesoft.
the class AbstractTransactionQueueManagerTestCase method testOffer.
@Test
public void testOffer() throws Exception {
final QueueManager mgr = createQueueManager();
mgr.setDefaultQueueConfiguration(new DefaultQueueConfiguration(1, false));
try {
mgr.start();
QueueSession s = mgr.getQueueSession();
Queue q = s.getQueue("queue1");
assertThat("Queue size", q.size(), is(0));
assertThat(q.offer("String1", 0L), is(true));
assertThat("Queue size", q.size(), is(1));
assertThat(q.offer("String2", 1000), is(false));
assertThat("Queue size", q.size(), is(1));
final Latch takeExecutionLatch = new Latch();
final Thread takeExecutionThread = new Thread(() -> {
try {
takeExecutionLatch.release();
QueueSession s1 = mgr.getQueueSession();
Queue q1 = s1.getQueue("queue1");
assertThat("Queue content", q1.take(), is("String1"));
} catch (Exception e) {
// unlikely to happen. But if it does lets show it in the test logs.
logger.warn("Error using queue session", e);
}
});
takeExecutionThread.start();
if (!takeExecutionLatch.await(THREAD_EXECUTION_TIMEOUT, TimeUnit.MILLISECONDS)) {
fail("Thread executing put over queue was not executed");
}
assertThat(q.offer("String2", 1000), is(true));
takeExecutionThread.join(THREAD_EXECUTION_TIMEOUT);
assertThat("Queue size", q.size(), is(1));
} finally {
mgr.stop();
}
}
use of org.mule.runtime.api.util.concurrent.Latch in project mule by mulesoft.
the class AbstractTransactionQueueManagerTestCase method testPoll.
@Test
public void testPoll() throws Exception {
final QueueManager mgr = createQueueManager();
try {
mgr.start();
QueueSession s = mgr.getQueueSession();
Queue q = s.getQueue("queue1");
assertEquals("Queue size", 0, q.size());
Object o = q.poll(0);
assertEquals("Queue size", 0, q.size());
assertNull(o);
o = q.poll(1000);
assertEquals("Queue size", 0, q.size());
assertNull(o);
q.put("String1");
assertEquals("Queue size", 1, q.size());
o = q.poll(0);
assertEquals("Queue size", 0, q.size());
assertEquals("Queue content", "String1", o);
final Latch putExecutionLatch = new Latch();
Thread putExecutionThread = new Thread(() -> {
try {
QueueSession s1 = mgr.getQueueSession();
Queue q1 = s1.getQueue("queue1");
putExecutionLatch.release();
q1.put("String1");
} catch (Exception e) {
// unlikely to happen. But if it does lets show it in the test logs.
logger.warn("Error using queue session", e);
}
});
putExecutionThread.start();
if (!putExecutionLatch.await(THREAD_EXECUTION_TIMEOUT, TimeUnit.MILLISECONDS)) {
fail("Thread executing put over queue was not executed");
}
o = q.poll(RECEIVE_TIMEOUT);
putExecutionThread.join(THREAD_EXECUTION_TIMEOUT);
assertEquals("Queue size", q.size(), 0);
assertEquals("Queue content", "String1", o);
} finally {
mgr.stop();
}
}
use of org.mule.runtime.api.util.concurrent.Latch in project mule by mulesoft.
the class OnceTestCase method concurrentConsume.
@Test
public void concurrentConsume() {
Latch controlLath = new Latch();
Latch testLath = new Latch();
CountingConsumer consumer = new CountingConsumer();
ConsumeOnce<String> once = Once.of(consumer);
new Thread(() -> {
await(controlLath);
once.consumeOnce("s");
}).start();
new Thread(() -> {
controlLath.release();
once.consumeOnce("s");
testLath.release();
}).start();
await(testLath);
assertThat(consumer.getInvokationCount(), is(1));
}
Aggregations