Search in sources :

Example 6 with DirectoryLayer

use of com.apple.foundationdb.directory.DirectoryLayer in project fdb-record-layer by FoundationDB.

the class ScopedDirectoryLayerTest method testDefaultDirectoryResolver.

@Test
public void testDefaultDirectoryResolver() {
    LocatableResolver resolver = globalScope;
    try (FDBRecordContext context = database.openContext()) {
        Long value = resolver.resolve(context.getTimer(), "foo").join();
        DirectoryLayer directoryLayer = DirectoryLayer.getDefault();
        validate(context, resolver, directoryLayer, "foo", value);
    }
}
Also used : DirectoryLayer(com.apple.foundationdb.directory.DirectoryLayer) FDBRecordContext(com.apple.foundationdb.record.provider.foundationdb.FDBRecordContext) Test(org.junit.jupiter.api.Test)

Example 7 with DirectoryLayer

use of com.apple.foundationdb.directory.DirectoryLayer in project fdb-record-layer by FoundationDB.

the class ScopedDirectoryLayerTest method testLocatableDirectoryResolver.

@Test
public void testLocatableDirectoryResolver() {
    KeySpace keySpace = new KeySpace(new KeySpaceDirectory("path", KeyType.STRING, "path").addSubdirectory(new KeySpaceDirectory("to", KeyType.STRING, "to").addSubdirectory(new KeySpaceDirectory("dirLayer", KeyType.STRING, "dirLayer"))));
    ResolvedKeySpacePath path;
    try (FDBRecordContext context = database.openContext()) {
        path = keySpace.resolveFromKey(context, Tuple.from("path", "to", "dirLayer"));
    }
    LocatableResolver resolver = resolverFactory.create(path);
    Long value = resolver.resolve("foo").join();
    DirectoryLayer directoryLayer = new DirectoryLayer(new Subspace(Bytes.concat(path.toTuple().pack(), DirectoryLayer.DEFAULT_NODE_SUBSPACE.getKey())), path.toSubspace());
    try (FDBRecordContext context = database.openContext()) {
        validate(context, resolver, directoryLayer, "foo", value);
        DirectoryLayer defaultDirectoryLayer = DirectoryLayer.getDefault();
        List<String> defaultDirectories = defaultDirectoryLayer.list(context.ensureActive()).join();
        assertThat("entry is not in the default directory layer", defaultDirectories, not(hasItem("foo")));
    }
}
Also used : DirectoryLayer(com.apple.foundationdb.directory.DirectoryLayer) FDBRecordContext(com.apple.foundationdb.record.provider.foundationdb.FDBRecordContext) Subspace(com.apple.foundationdb.subspace.Subspace) Test(org.junit.jupiter.api.Test)

Example 8 with DirectoryLayer

use of com.apple.foundationdb.directory.DirectoryLayer in project fdb-record-layer by FoundationDB.

the class ExtendedDirectoryLayerTest method testParallelAllocation.

private static void testParallelAllocation(boolean checkDirectoryLayer, FDBDatabase database, Function<String, CompletableFuture<Long>> resolveCall1, Function<String, CompletableFuture<Long>> resolveCall2, LocatableResolver resolver1, LocatableResolver resolver2) {
    Map<String, Long> mappings1 = new ConcurrentHashMap<>();
    Map<String, Long> mappings2 = new ConcurrentHashMap<>();
    List<CompletableFuture<?>> operations = new ArrayList<>();
    for (int i = 0; i < 100; i++) {
        String key = "key-" + i;
        operations.add(resolveCall1.apply(key).thenAccept(value -> mappings1.put(key, value)));
        operations.add(resolveCall2.apply(key).thenAccept(value -> mappings2.put(key, value)));
    }
    CompletableFuture.allOf(operations.toArray(new CompletableFuture<?>[0])).join();
    for (Map.Entry<String, Long> entry : mappings1.entrySet()) {
        assertThat(String.format("the mappings for %s are identical", entry.getKey()), mappings2.get(entry.getKey()), is(entry.getValue()));
        try (FDBRecordContext context = database.openContext()) {
            if (checkDirectoryLayer) {
                // only applies when the scope is global
                final DirectoryLayer directoryLayer = DirectoryLayer.getDefault();
                Long value = directoryLayer.open(context.ensureActive(), Collections.singletonList(entry.getKey())).thenApply(subspace -> Tuple.fromBytes(subspace.pack()).getLong(0)).join();
                assertThat("the FDB directory layer sees the mapping", value, is(entry.getValue()));
            }
            assertThat(resolver1.getClass().getName() + " sees the mapping", resolver1.mustResolve(context, entry.getKey()).join(), is(entry.getValue()));
            assertThat(resolver2.getClass().getName() + " sees the mapping", resolver2.mustResolve(context, entry.getKey()).join(), is(entry.getValue()));
            checkMappingInReverseCache(resolver1, entry.getKey(), entry.getValue());
            checkMappingInReverseCache(resolver2, entry.getKey(), entry.getValue());
        }
    }
}
Also used : Assertions.assertThrows(org.junit.jupiter.api.Assertions.assertThrows) MetadataHook(com.apple.foundationdb.record.provider.foundationdb.keyspace.ResolverCreateHooks.MetadataHook) FDBRecordContext(com.apple.foundationdb.record.provider.foundationdb.FDBRecordContext) HashMap(java.util.HashMap) CompletableFuture(java.util.concurrent.CompletableFuture) AsyncUtil(com.apple.foundationdb.async.AsyncUtil) Function(java.util.function.Function) ArrayList(java.util.ArrayList) ImmutableBiMap(com.google.common.collect.ImmutableBiMap) Tuple(com.apple.foundationdb.tuple.Tuple) STRING(com.apple.foundationdb.record.provider.foundationdb.keyspace.KeySpaceDirectory.KeyType.STRING) Map(java.util.Map) Matchers.nullValue(org.hamcrest.Matchers.nullValue) MatcherAssert.assertThat(org.hamcrest.MatcherAssert.assertThat) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) BiMap(com.google.common.collect.BiMap) FDBStoreTimer(com.apple.foundationdb.record.provider.foundationdb.FDBStoreTimer) ConcurrentHashMap(java.util.concurrent.ConcurrentHashMap) DirectoryLayer(com.apple.foundationdb.directory.DirectoryLayer) FDBDatabase(com.apple.foundationdb.record.provider.foundationdb.FDBDatabase) Test(org.junit.jupiter.api.Test) Assertions.assertArrayEquals(org.junit.jupiter.api.Assertions.assertArrayEquals) AtomicLong(java.util.concurrent.atomic.AtomicLong) List(java.util.List) Matchers.is(org.hamcrest.Matchers.is) Collections(java.util.Collections) DirectoryLayer(com.apple.foundationdb.directory.DirectoryLayer) ArrayList(java.util.ArrayList) CompletableFuture(java.util.concurrent.CompletableFuture) FDBRecordContext(com.apple.foundationdb.record.provider.foundationdb.FDBRecordContext) AtomicLong(java.util.concurrent.atomic.AtomicLong) ConcurrentHashMap(java.util.concurrent.ConcurrentHashMap) HashMap(java.util.HashMap) ImmutableBiMap(com.google.common.collect.ImmutableBiMap) Map(java.util.Map) BiMap(com.google.common.collect.BiMap) ConcurrentHashMap(java.util.concurrent.ConcurrentHashMap)

Aggregations

DirectoryLayer (com.apple.foundationdb.directory.DirectoryLayer)8 Test (org.junit.jupiter.api.Test)7 FDBRecordContext (com.apple.foundationdb.record.provider.foundationdb.FDBRecordContext)6 MetadataHook (com.apple.foundationdb.record.provider.foundationdb.keyspace.ResolverCreateHooks.MetadataHook)4 Tuple (com.apple.foundationdb.tuple.Tuple)4 HashMap (java.util.HashMap)4 List (java.util.List)4 Map (java.util.Map)4 MatcherAssert.assertThat (org.hamcrest.MatcherAssert.assertThat)4 Matchers.nullValue (org.hamcrest.Matchers.nullValue)4 Assertions.assertThrows (org.junit.jupiter.api.Assertions.assertThrows)4 AsyncUtil (com.apple.foundationdb.async.AsyncUtil)3 FDBDatabase (com.apple.foundationdb.record.provider.foundationdb.FDBDatabase)3 FDBStoreTimer (com.apple.foundationdb.record.provider.foundationdb.FDBStoreTimer)3 STRING (com.apple.foundationdb.record.provider.foundationdb.keyspace.KeySpaceDirectory.KeyType.STRING)3 BiMap (com.google.common.collect.BiMap)3 ImmutableBiMap (com.google.common.collect.ImmutableBiMap)3 ArrayList (java.util.ArrayList)3 Collections (java.util.Collections)3 CompletableFuture (java.util.concurrent.CompletableFuture)3