use of org.spf4j.base.Either in project spf4j by zolyfarkas.
the class SimpleSmartObjectPool method get.
@Override
@SuppressFBWarnings({ "RV_RETURN_VALUE_IGNORED_BAD_PRACTICE", "PRMC_POSSIBLY_REDUNDANT_METHOD_CALLS" })
public T get(final ObjectBorower borower) throws InterruptedException, TimeoutException, ObjectCreationException {
lock.lock();
try {
if (maxSize <= 0) {
throw new IllegalStateException("Pool closed, noting available for " + borower);
}
int nrAvailable = availableObjects.size();
// trying to be fair here, if others are already waiting, we will not get one.
if (nrAvailable - waitingForReturn > 0) {
Iterator<T> it = availableObjects.iterator();
T object = it.next();
it.remove();
if (!borrowedObjects.put(borower, object)) {
throw new IllegalStateException("Cannot borrow " + object + ", " + borrowedObjects);
}
return object;
} else if (borrowedObjects.size() + nrAvailable < maxSize) {
T object = factory.create();
if (!borrowedObjects.put(borower, object)) {
throw new IllegalStateException("Cannot borrow " + object + ", " + borrowedObjects);
}
return object;
} else {
// try to reclaim object from borrowers
for (ObjectBorower<T> b : borrowedObjects.keySet()) {
if (borower != b) {
T object = b.tryReturnObjectIfNotInUse();
if (object != null) {
if (!borrowedObjects.remove(b, object)) {
throw new IllegalStateException("Returned Object hasn't been borrowed " + object);
}
if (!borrowedObjects.put(borower, object)) {
throw new IllegalStateException("Cannot borrow " + object + ", " + borrowedObjects);
}
return object;
}
}
}
// evrything in use, try to place a return request
boolean requestMade = false;
while (!requestMade) {
boolean hasValidBorowers = false;
for (ObjectBorower<T> b : borrowedObjects.keySet()) {
if (borower != b) {
hasValidBorowers = true;
Either<ObjectBorower.Action, T> objOrPromise = b.tryRequestReturnObject();
if (objOrPromise.isRight()) {
T obj = objOrPromise.getRight();
if (!borrowedObjects.remove(b, obj)) {
throw new IllegalStateException("Returned Object " + obj + " hasn't been borrowed: " + borrowedObjects);
}
if (!borrowedObjects.put(borower, obj)) {
throw new IllegalStateException("Cannot boroow " + obj + ", " + borrowedObjects);
}
return obj;
} else {
requestMade = objOrPromise.getLeft() == Action.REQUEST_MADE;
if (requestMade) {
break;
}
}
}
}
if (!hasValidBorowers) {
throw new IllegalStateException("Borrower asks for more than possible " + borower);
}
if (!requestMade) {
// probably was unable to acquire the locks
do {
available.await(1, TimeUnit.MILLISECONDS);
long millisToDeadline = ExecutionContexts.getTimeRelativeToDeadline(TimeUnit.MILLISECONDS);
if (millisToDeadline < 0) {
throw new TimeoutException("Object wait timeout expired by " + (-millisToDeadline));
}
} while (borrowedObjects.isEmpty());
}
}
waitingForReturn++;
while (availableObjects.isEmpty()) {
long waitTime = ExecutionContexts.getTimeToDeadline(TimeUnit.MILLISECONDS);
if (!available.await(waitTime, TimeUnit.MILLISECONDS)) {
throw new TimeoutException("Object wait timeout expired, waitTime = " + waitTime);
}
}
waitingForReturn--;
Iterator<T> it = availableObjects.iterator();
T objectT = it.next();
it.remove();
if (!borrowedObjects.put(borower, objectT)) {
throw new IllegalStateException("Cannot borrow " + objectT + ", " + borrowedObjects);
}
return objectT;
}
} finally {
lock.unlock();
}
}
use of org.spf4j.base.Either in project spf4j by zolyfarkas.
the class TSDBReaderTest method testTailing.
@Test
public void testTailing() throws IOException, InterruptedException, ExecutionException, TimeoutException {
File testFile = File.createTempFile("test", ".tsdb2");
try (TSDBWriter writer = new TSDBWriter(testFile, 4, "test", true);
TSDBReader reader = new TSDBReader(testFile, 1024)) {
writer.flush();
final BlockingQueue<Either<TableDef, DataBlock>> queue = new ArrayBlockingQueue<>(100);
Future<Void> bgWatch = reader.bgWatch((Either<TableDef, DataBlock> object, long deadline) -> {
queue.put(object);
}, TSDBReader.EventSensitivity.HIGH);
long tableId = writer.writeTableDef(tableDef);
writer.flush();
final long time = System.currentTimeMillis();
writer.writeDataRow(tableId, time, 0, 1, 2);
writer.writeDataRow(tableId, time + 10, 1, 1, 2);
writer.flush();
Either<TableDef, DataBlock> td = queue.take();
Assert.assertEquals(tableDef, td.getLeft());
Either<TableDef, DataBlock> take = queue.take();
Assert.assertEquals(2, take.getRight().getValues().size());
writer.writeDataRow(tableId, time + 20, 2, 1, 2);
writer.writeDataRow(tableId, time + 30, 3, 1, 2);
writer.writeDataRow(tableId, time + 40, 4, 1, 2);
writer.flush();
Assert.assertEquals(3, queue.take().getRight().getValues().size());
reader.stopWatching();
bgWatch.get(10000, TimeUnit.MILLISECONDS);
}
}
Aggregations