use of com.questdb.ex.RetryLockException in project questdb by bluestreak01.
the class CachingReaderFactoryTest method testLockUnlock.
@Test
public void testLockUnlock() throws Exception {
// create journals
final JournalMetadata<?> m1 = new JournalStructure("x").$date("ts").$().build();
((WriterFactory) getFactory()).writer(m1).close();
final JournalMetadata<?> m2 = new JournalStructure("y").$date("ts").$().build();
((WriterFactory) getFactory()).writer(m2).close();
Journal x, y;
try (final CachingReaderFactory rf = new CachingReaderFactory(factoryContainer.getConfiguration(), 1000, 2)) {
x = rf.reader(m1);
Assert.assertNotNull(x);
y = rf.reader(m2);
Assert.assertNotNull(y);
// expect lock to fail because we have "x" open
try {
rf.lock(m1.getName());
Assert.fail();
} catch (RetryLockException ignore) {
}
x.close();
// expect lock to succeed after we closed "x"
rf.lock(m1.getName());
// expect "x" to be physically closed
Assert.assertFalse(x.isOpen());
// "x" is locked, expect this to fail
try {
Assert.assertNull(rf.reader(m1));
} catch (JournalLockedException ignored) {
}
rf.unlock(m1.getName());
x = rf.reader(m1);
Assert.assertNotNull(x);
x.close();
Assert.assertTrue(x.isOpen());
}
Assert.assertTrue(y.isOpen());
y.close();
Assert.assertFalse(y.isOpen());
// "x" was not busy and should be closed by factory
Assert.assertFalse(x.isOpen());
}
use of com.questdb.ex.RetryLockException in project questdb by bluestreak01.
the class CachingReaderFactoryTest method testLockUnlockMultiple.
@Test
public void testLockUnlockMultiple() throws Exception {
final JournalMetadata<?> m1 = new JournalStructure("x").$date("ts").$().build();
getFactory().writer(m1).close();
try (final CachingReaderFactory rf = new CachingReaderFactory(factoryContainer.getConfiguration(), 1000, 2)) {
Journal r1 = rf.reader(m1);
Journal r2 = rf.reader(m1);
r1.close();
try {
rf.lock(m1.getName());
} catch (RetryLockException e) {
e.printStackTrace();
}
r2.close();
rf.lock(m1.getName());
rf.unlock(m1.getName());
}
}
use of com.questdb.ex.RetryLockException in project questdb by bluestreak01.
the class CachingReaderFactoryTest method testLockBusyReader.
@Test
public void testLockBusyReader() throws Exception {
final int readerCount = 5;
int threadCount = 2;
final int iterations = 10000;
// create journals to read
final JournalMetadata<?>[] meta = new JournalMetadata[readerCount];
for (int i = 0; i < readerCount; i++) {
final JournalMetadata<?> m = new JournalStructure("x" + i).$date("ts").$().build();
((WriterFactory) getFactory()).writer(m).close();
meta[i] = m;
}
try {
try (final CachingReaderFactory rf = new CachingReaderFactory(factoryContainer.getConfiguration(), 1000, 2)) {
final CyclicBarrier barrier = new CyclicBarrier(threadCount);
final CountDownLatch halt = new CountDownLatch(threadCount);
final AtomicInteger errors = new AtomicInteger();
final LongList lockTimes = new LongList();
final LongList workerTimes = new LongList();
new Thread(() -> {
Rnd rnd = new Rnd();
try {
barrier.await();
String name = null;
for (int i = 0; i < iterations; i++) {
if (name == null) {
name = meta[rnd.nextPositiveInt() % readerCount].getName();
}
while (true) {
try {
rf.lock(name);
lockTimes.add(System.currentTimeMillis());
LockSupport.parkNanos(100L);
rf.unlock(name);
name = null;
break;
} catch (JournalException e) {
if (!(e instanceof RetryLockException)) {
e.printStackTrace();
errors.incrementAndGet();
break;
}
}
}
}
} catch (Exception e) {
e.printStackTrace();
errors.incrementAndGet();
}
halt.countDown();
}).start();
new Thread(() -> {
Rnd rnd = new Rnd();
workerTimes.add(System.currentTimeMillis());
for (int i = 0; i < iterations; i++) {
JournalMetadata<?> metadata = meta[rnd.nextPositiveInt() % readerCount];
try (Journal<?> ignored = rf.reader(metadata)) {
if (metadata == meta[readerCount - 1] && barrier.getNumberWaiting() > 0) {
barrier.await();
}
LockSupport.parkNanos(10L);
} catch (JournalLockedException ignored) {
} catch (Exception e) {
e.printStackTrace();
errors.incrementAndGet();
}
}
workerTimes.add(System.currentTimeMillis());
halt.countDown();
}).start();
halt.await();
Assert.assertEquals(0, errors.get());
// check that there are lock times between worker times
int count = 0;
// ensure that we have worker times
Assert.assertEquals(2, workerTimes.size());
long lo = workerTimes.get(0);
long hi = workerTimes.get(1);
Assert.assertTrue(lockTimes.size() > 0);
for (int i = 0, n = lockTimes.size(); i < n; i++) {
long t = lockTimes.getQuick(i);
if (t > lo && t < hi) {
count++;
}
}
Assert.assertTrue(count > 0);
}
} catch (Throwable e) {
e.printStackTrace();
}
}
Aggregations