Examples with StandardIDPool com.thinkaurelius.titan.graphdb.database.idassigner.StandardIDPool used on opensource projects

Example 1 with StandardIDPool

use of com.thinkaurelius.titan.graphdb.database.idassigner.StandardIDPool in project titan by thinkaurelius.

the class IDPoolTest method testStandardIDPool1.

@Test
public void testStandardIDPool1() throws InterruptedException {
    final MockIDAuthority idauth = new MockIDAuthority(200);
    testIDPoolWith(new IDPoolFactory() {

        @Override
        public StandardIDPool get(int partitionID) {
            return new StandardIDPool(idauth, partitionID, partitionID, Integer.MAX_VALUE, Duration.ofMillis(2000L), 0.2);
        }
    }, 1000, 6, 100000);
}

Example 2 with StandardIDPool

use of com.thinkaurelius.titan.graphdb.database.idassigner.StandardIDPool in project titan by thinkaurelius.

the class IDPoolTest method testAllocationTimeout.

@Test
public void testAllocationTimeout() {
    final MockIDAuthority idauth = new MockIDAuthority(10000, Integer.MAX_VALUE, 5000);
    StandardIDPool pool = new StandardIDPool(idauth, 1, 1, Integer.MAX_VALUE, Duration.ofMillis(4000), 0.1);
    try {
        pool.nextID();
        fail();
    } catch (TitanException e) {
    }
}

Example 3 with StandardIDPool

use of com.thinkaurelius.titan.graphdb.database.idassigner.StandardIDPool in project titan by thinkaurelius.

the class IDPoolTest method testIDPoolWith.

private void testIDPoolWith(IDPoolFactory poolFactory, final int numPartitions, final int numThreads, final int attemptsPerThread) throws InterruptedException {
    final Random random = new Random();
    final IntSet[] ids = new IntSet[numPartitions];
    final StandardIDPool[] idPools = new StandardIDPool[numPartitions];
    for (int i = 0; i < numPartitions; i++) {
        ids[i] = new IntHashSet(attemptsPerThread * numThreads / numPartitions);
        int partition = i * 100;
        idPools[i] = poolFactory.get(partition);
    }
    Thread[] threads = new Thread[numThreads];
    for (int i = 0; i < numThreads; i++) {
        threads[i] = new Thread(new Runnable() {

            @Override
            public void run() {
                for (int attempt = 0; attempt < attemptsPerThread; attempt++) {
                    int offset = random.nextInt(numPartitions);
                    long id = idPools[offset].nextID();
                    assertTrue(id < Integer.MAX_VALUE);
                    IntSet idset = ids[offset];
                    synchronized (idset) {
                        assertFalse(idset.contains((int) id));
                        idset.add((int) id);
                    }
                }
            }
        });
        threads[i].start();
    }
    for (int i = 0; i < numThreads; i++) threads[i].join();
    for (int i = 0; i < idPools.length; i++) idPools[i].close();
    //Verify consecutive id assignment
    for (int i = 0; i < ids.length; i++) {
        IntSet set = ids[i];
        int max = 0;
        int[] all = set.getAll();
        for (int j = 0; j < all.length; j++) if (all[j] > max)
            max = all[j];
        for (int j = 1; j <= max; j++) assertTrue(i + " contains: " + j, set.contains(j));
    }
}

Example 4 with StandardIDPool

use of com.thinkaurelius.titan.graphdb.database.idassigner.StandardIDPool in project titan by thinkaurelius.

the class IDPoolTest method testStandardIDPool3.

@Test
public void testStandardIDPool3() throws InterruptedException {
    final MockIDAuthority idauth = new MockIDAuthority(200);
    testIDPoolWith(new IDPoolFactory() {

        @Override
        public StandardIDPool get(int partitionID) {
            return new StandardIDPool(idauth, partitionID, partitionID, Integer.MAX_VALUE, Duration.ofMillis(2000), 0.2);
        }
    }, 10, 20, 100000);
}

Example 5 with StandardIDPool

use of com.thinkaurelius.titan.graphdb.database.idassigner.StandardIDPool in project titan by thinkaurelius.

the class IDPoolTest method testPoolExhaustion1.

@Test
public void testPoolExhaustion1() {
    MockIDAuthority idauth = new MockIDAuthority(200);
    int idUpper = 10000;
    StandardIDPool pool = new StandardIDPool(idauth, 0, 1, idUpper, Duration.ofMillis(2000), 0.2);
    for (int i = 1; i < idUpper * 2; i++) {
        try {
            long id = pool.nextID();
            assertTrue(id < idUpper);
        } catch (IDPoolExhaustedException e) {
            assertEquals(idUpper, i);
            break;
        }
    }
}