Search in sources :

Example 1 with Either

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();
    }
}
Also used : ObjectBorower(org.spf4j.recyclable.ObjectBorower) Iterator(java.util.Iterator) Either(org.spf4j.base.Either) TimeoutException(java.util.concurrent.TimeoutException) SuppressFBWarnings(edu.umd.cs.findbugs.annotations.SuppressFBWarnings)

Example 2 with Either

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);
    }
}
Also used : ArrayBlockingQueue(java.util.concurrent.ArrayBlockingQueue) Either(org.spf4j.base.Either) File(java.io.File) TableDef(org.spf4j.tsdb2.avro.TableDef) DataBlock(org.spf4j.tsdb2.avro.DataBlock) Test(org.junit.Test)

Aggregations

Either (org.spf4j.base.Either)2 SuppressFBWarnings (edu.umd.cs.findbugs.annotations.SuppressFBWarnings)1 File (java.io.File)1 Iterator (java.util.Iterator)1 ArrayBlockingQueue (java.util.concurrent.ArrayBlockingQueue)1 TimeoutException (java.util.concurrent.TimeoutException)1 Test (org.junit.Test)1 ObjectBorower (org.spf4j.recyclable.ObjectBorower)1 DataBlock (org.spf4j.tsdb2.avro.DataBlock)1 TableDef (org.spf4j.tsdb2.avro.TableDef)1