Search in sources :

Example 11 with IndexProvider

use of org.neo4j.kernel.api.index.IndexProvider in project neo4j by neo4j.

the class BatchInsertTest method shouldRunIndexPopulationJobAtShutdown.

@ParameterizedTest
@MethodSource("params")
void shouldRunIndexPopulationJobAtShutdown(int denseNodeThreshold) throws Throwable {
    // GIVEN
    IndexPopulator populator = mock(IndexPopulator.class);
    IndexProvider provider = mock(IndexProvider.class);
    IndexAccessor accessor = mock(IndexAccessor.class);
    when(provider.getProviderDescriptor()).thenReturn(DESCRIPTOR);
    when(provider.getPopulator(any(IndexDescriptor.class), any(IndexSamplingConfig.class), any(), any(), any(TokenNameLookup.class))).thenReturn(populator);
    when(populator.sample(any(CursorContext.class))).thenReturn(new IndexSample());
    when(provider.getOnlineAccessor(any(IndexDescriptor.class), any(IndexSamplingConfig.class), any(TokenNameLookup.class))).thenReturn(accessor);
    when(provider.completeConfiguration(any(IndexDescriptor.class))).then(inv -> inv.getArgument(0));
    BatchInserter inserter = newBatchInserterWithIndexProvider(singleInstanceIndexProviderFactory(KEY, provider), provider.getProviderDescriptor(), denseNodeThreshold);
    inserter.createDeferredSchemaIndex(label("Hacker")).on("handle").create();
    long nodeId = inserter.createNode(map("handle", "Jakewins"), label("Hacker"));
    // WHEN
    inserter.shutdown();
    // THEN
    verify(provider).init();
    verify(provider).start();
    verify(provider).getPopulator(any(IndexDescriptor.class), any(IndexSamplingConfig.class), any(), any(), any(TokenNameLookup.class));
    verify(populator).create();
    verify(populator).add(argThat(c -> c.contains(add(nodeId, internalIndex.schema(), Values.of("Jakewins")))), any(CursorContext.class));
    verify(populator).verifyDeferredConstraints(any(NodePropertyAccessor.class));
    verify(populator).close(eq(true), any());
    verify(provider).stop();
    verify(provider).shutdown();
}
Also used : IndexProvider(org.neo4j.kernel.api.index.IndexProvider) Arrays(java.util.Arrays) ResourceIterator(org.neo4j.graphdb.ResourceIterator) DatabaseReadOnlyChecker.readOnly(org.neo4j.configuration.helpers.DatabaseReadOnlyChecker.readOnly) CursorContext(org.neo4j.io.pagecache.context.CursorContext) ArgumentMatchers.eq(org.mockito.ArgumentMatchers.eq) Direction(org.neo4j.graphdb.Direction) TokenNameLookup(org.neo4j.common.TokenNameLookup) IndexPopulator(org.neo4j.kernel.api.index.IndexPopulator) NORMAL(org.neo4j.kernel.impl.store.record.RecordLoad.NORMAL) Config(org.neo4j.configuration.Config) IndexSamplingConfig(org.neo4j.kernel.impl.api.index.IndexSamplingConfig) DEFAULT_DATABASE_NAME(org.neo4j.configuration.GraphDatabaseSettings.DEFAULT_DATABASE_NAME) Arrays.asList(java.util.Arrays.asList) Map(java.util.Map) PageCacheTracer(org.neo4j.io.pagecache.tracing.PageCacheTracer) NULL(org.neo4j.io.pagecache.context.CursorContext.NULL) RecoveryCleanupWorkCollector(org.neo4j.index.internal.gbptree.RecoveryCleanupWorkCollector) TokenIndexProviderFactory(org.neo4j.kernel.impl.index.schema.TokenIndexProviderFactory) SchemaStore(org.neo4j.kernel.impl.store.SchemaStore) Set(java.util.Set) SchemaRecord(org.neo4j.kernel.impl.store.record.SchemaRecord) Arguments(org.junit.jupiter.params.provider.Arguments) GraphDatabaseAPI(org.neo4j.kernel.internal.GraphDatabaseAPI) ConstraintDefinition(org.neo4j.graphdb.schema.ConstraintDefinition) Stream(java.util.stream.Stream) BatchInserter(org.neo4j.batchinsert.BatchInserter) IndexEntryUpdate.add(org.neo4j.storageengine.api.IndexEntryUpdate.add) Assertions.assertTrue(org.junit.jupiter.api.Assertions.assertTrue) RelationshipType(org.neo4j.graphdb.RelationshipType) NodeStore(org.neo4j.kernel.impl.store.NodeStore) IndexDescriptor(org.neo4j.internal.schema.IndexDescriptor) Mockito.mock(org.mockito.Mockito.mock) Assertions.assertThrows(org.junit.jupiter.api.Assertions.assertThrows) Assertions.fail(org.junit.jupiter.api.Assertions.fail) MapUtil.map(org.neo4j.internal.helpers.collection.MapUtil.map) SimpleDateFormat(java.text.SimpleDateFormat) KernelVersion(org.neo4j.kernel.KernelVersion) Node(org.neo4j.graphdb.Node) ArrayList(java.util.ArrayList) OptionalLong(java.util.OptionalLong) TestDatabaseManagementServiceBuilder(org.neo4j.test.TestDatabaseManagementServiceBuilder) GBPTreeCountsStore(org.neo4j.internal.counts.GBPTreeCountsStore) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) MemoryTracker(org.neo4j.memory.MemoryTracker) NodePropertyAccessor(org.neo4j.storageengine.api.NodePropertyAccessor) TestIndexProviderDescriptor(org.neo4j.kernel.impl.api.index.TestIndexProviderDescriptor) IndexSample(org.neo4j.kernel.api.index.IndexSample) Files(java.nio.file.Files) Iterators(org.neo4j.internal.helpers.collection.Iterators) IOException(java.io.IOException) Label.label(org.neo4j.graphdb.Label.label) Assertions.assertArrayEquals(org.junit.jupiter.api.Assertions.assertArrayEquals) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) ExtensionFactory(org.neo4j.kernel.extension.ExtensionFactory) TokenHolders(org.neo4j.token.TokenHolders) ConstraintType(org.neo4j.graphdb.schema.ConstraintType) NodeRecord(org.neo4j.kernel.impl.store.record.NodeRecord) IndexDefinition(org.neo4j.graphdb.schema.IndexDefinition) RecordLoad(org.neo4j.kernel.impl.store.record.RecordLoad) Iterators.asCollection(org.neo4j.internal.helpers.collection.Iterators.asCollection) SchemaIndexTestHelper.singleInstanceIndexProviderFactory(org.neo4j.kernel.impl.api.index.SchemaIndexTestHelper.singleInstanceIndexProviderFactory) SchemaRule(org.neo4j.internal.schema.SchemaRule) SchemaRuleAccess(org.neo4j.internal.recordstorage.SchemaRuleAccess) ArgumentMatchers.argThat(org.mockito.ArgumentMatchers.argThat) MyRelTypes(org.neo4j.kernel.impl.MyRelTypes) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) ConstraintDescriptor(org.neo4j.internal.schema.ConstraintDescriptor) DatabaseLayout(org.neo4j.io.layout.DatabaseLayout) NodeLabelsField(org.neo4j.kernel.impl.store.NodeLabelsField) Assertions.assertFalse(org.junit.jupiter.api.Assertions.assertFalse) Iterables.map(org.neo4j.internal.helpers.collection.Iterables.map) Iterables.single(org.neo4j.internal.helpers.collection.Iterables.single) Transaction(org.neo4j.graphdb.Transaction) IndexProviderDescriptor(org.neo4j.internal.schema.IndexProviderDescriptor) MethodSource(org.junit.jupiter.params.provider.MethodSource) PageCache(org.neo4j.io.pagecache.PageCache) TestDirectory(org.neo4j.test.rule.TestDirectory) RecordStorageEngine(org.neo4j.internal.recordstorage.RecordStorageEngine) Neo4jLayoutExtension(org.neo4j.test.extension.Neo4jLayoutExtension) String.format(java.lang.String.format) Test(org.junit.jupiter.api.Test) List(java.util.List) INSTANCE(org.neo4j.memory.EmptyMemoryTracker.INSTANCE) DatabaseManagementService(org.neo4j.dbms.api.DatabaseManagementService) ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) Label(org.neo4j.graphdb.Label) GraphDatabaseSettings(org.neo4j.configuration.GraphDatabaseSettings) HashMap(java.util.HashMap) IndexAccessor(org.neo4j.kernel.api.index.IndexAccessor) Iterators.iterator(org.neo4j.internal.helpers.collection.Iterators.iterator) IndexProvider(org.neo4j.kernel.api.index.IndexProvider) Values(org.neo4j.values.storable.Values) GraphDatabaseSettings.preallocate_logical_logs(org.neo4j.configuration.GraphDatabaseSettings.preallocate_logical_logs) HashSet(java.util.HashSet) ConstraintViolationException(org.neo4j.graphdb.ConstraintViolationException) NodeLabels(org.neo4j.kernel.impl.store.NodeLabels) GraphDatabaseService(org.neo4j.graphdb.GraphDatabaseService) NeoStores(org.neo4j.kernel.impl.store.NeoStores) Inject(org.neo4j.test.extension.Inject) Iterables(org.neo4j.internal.helpers.collection.Iterables) GraphDatabaseSettings.dense_node_threshold(org.neo4j.configuration.GraphDatabaseSettings.dense_node_threshold) GraphDatabaseInternalSettings(org.neo4j.configuration.GraphDatabaseInternalSettings) Arguments.arguments(org.junit.jupiter.params.provider.Arguments.arguments) TestIndexDescriptorFactory(org.neo4j.kernel.api.schema.index.TestIndexDescriptorFactory) PageCacheExtension(org.neo4j.test.extension.pagecache.PageCacheExtension) Mockito.when(org.mockito.Mockito.when) Mockito.verify(org.mockito.Mockito.verify) TimeUnit(java.util.concurrent.TimeUnit) CountsAccessor(org.neo4j.counts.CountsAccessor) Relationship(org.neo4j.graphdb.Relationship) CountsBuilder(org.neo4j.internal.counts.CountsBuilder) Iterators.asSet(org.neo4j.internal.helpers.collection.Iterators.asSet) Pair(org.neo4j.internal.helpers.collection.Pair) GraphDatabaseSettings.neo4j_home(org.neo4j.configuration.GraphDatabaseSettings.neo4j_home) BatchInserters(org.neo4j.batchinsert.BatchInserters) Collections(java.util.Collections) TransactionIdStore(org.neo4j.storageengine.api.TransactionIdStore) FileSystemAbstraction(org.neo4j.io.fs.FileSystemAbstraction) IndexPopulator(org.neo4j.kernel.api.index.IndexPopulator) IndexSamplingConfig(org.neo4j.kernel.impl.api.index.IndexSamplingConfig) BatchInserter(org.neo4j.batchinsert.BatchInserter) TokenNameLookup(org.neo4j.common.TokenNameLookup) IndexSample(org.neo4j.kernel.api.index.IndexSample) IndexAccessor(org.neo4j.kernel.api.index.IndexAccessor) CursorContext(org.neo4j.io.pagecache.context.CursorContext) IndexDescriptor(org.neo4j.internal.schema.IndexDescriptor) NodePropertyAccessor(org.neo4j.storageengine.api.NodePropertyAccessor) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) MethodSource(org.junit.jupiter.params.provider.MethodSource)

Example 12 with IndexProvider

use of org.neo4j.kernel.api.index.IndexProvider in project neo4j by neo4j.

the class BatchInsertTest method dbWithIndexAndSingleIndexedNode.

private long dbWithIndexAndSingleIndexedNode(int denseNodeThreshold) throws Exception {
    IndexPopulator populator = mock(IndexPopulator.class);
    IndexProvider provider = mock(IndexProvider.class);
    when(provider.getProviderDescriptor()).thenReturn(DESCRIPTOR);
    when(provider.getPopulator(any(IndexDescriptor.class), any(IndexSamplingConfig.class), any(), any(), any(TokenNameLookup.class))).thenReturn(populator);
    when(provider.completeConfiguration(any(IndexDescriptor.class))).then(inv -> inv.getArgument(0));
    BatchInserter inserter = newBatchInserterWithIndexProvider(singleInstanceIndexProviderFactory(KEY, provider), provider.getProviderDescriptor(), denseNodeThreshold);
    inserter.createDeferredSchemaIndex(label("Hacker")).on("handle").create();
    long nodeId = inserter.createNode(map("handle", "Jakewins"), label("Hacker"));
    inserter.shutdown();
    return nodeId;
}
Also used : IndexProvider(org.neo4j.kernel.api.index.IndexProvider) IndexPopulator(org.neo4j.kernel.api.index.IndexPopulator) IndexSamplingConfig(org.neo4j.kernel.impl.api.index.IndexSamplingConfig) BatchInserter(org.neo4j.batchinsert.BatchInserter) TokenNameLookup(org.neo4j.common.TokenNameLookup) IndexDescriptor(org.neo4j.internal.schema.IndexDescriptor)

Example 13 with IndexProvider

use of org.neo4j.kernel.api.index.IndexProvider in project neo4j by neo4j.

the class SchemaLoggingIT method shouldLogUserReadableLabelAndPropertyNames.

@Test
void shouldLogUserReadableLabelAndPropertyNames() {
    String labelName = "User";
    String property = "name";
    // when
    long indexId = createIndex(db, labelName, property);
    // then
    IndexProvider defaultProvider = indexProviderMap.getDefaultProvider();
    IndexProviderDescriptor providerDescriptor = defaultProvider.getProviderDescriptor();
    String indexName = "Index( id=" + indexId + ", name='index_a908f819', type='GENERAL BTREE', " + "schema=(:User {name}), indexProvider='" + providerDescriptor.name() + "' )";
    assertThat(logProvider).forLevel(INFO).containsMessageWithArguments("Index population started: [%s]", indexName).containsMessageWithArguments(CREATION_FINISHED, indexName);
}
Also used : IndexProvider(org.neo4j.kernel.api.index.IndexProvider) IndexProviderDescriptor(org.neo4j.internal.schema.IndexProviderDescriptor) Test(org.junit.jupiter.api.Test)

Example 14 with IndexProvider

use of org.neo4j.kernel.api.index.IndexProvider in project neo4j by neo4j.

the class DefaultIndexProviderMap method put.

private IndexProvider put(IndexProviderDescriptor providerDescriptor, IndexProvider provider) {
    IndexProvider existing = indexProvidersByDescriptor.putIfAbsent(providerDescriptor, provider);
    indexProvidersByName.putIfAbsent(providerDescriptor.name(), provider);
    return existing;
}
Also used : IndexProvider(org.neo4j.kernel.api.index.IndexProvider) TokenIndexProvider(org.neo4j.kernel.impl.index.schema.TokenIndexProvider)

Example 15 with IndexProvider

use of org.neo4j.kernel.api.index.IndexProvider in project neo4j by neo4j.

the class DefaultIndexProviderMap method initDefaultProviders.

private void initDefaultProviders() {
    String providerName = config.get(GraphDatabaseSettings.default_schema_provider);
    IndexProvider configuredDefaultProvider = indexProvidersByName.get(providerName);
    requireNonNull(configuredDefaultProvider, () -> format("Configured default provider: `%s` not found. Available index providers: %s.", providerName, indexProvidersByName.keySet()));
    defaultIndexProvider = configuredDefaultProvider;
    String fulltextProviderName = config.get(GraphDatabaseInternalSettings.default_fulltext_provider);
    fulltextIndexProvider = indexProvidersByName.get(fulltextProviderName);
    if (fulltextIndexProvider == null) {
        // Not all environments have the full-text index provider available.
        fulltextIndexProvider = IndexProvider.EMPTY;
    }
    var configuredTokenIndexProvider = indexProvidersByName.get(TOKEN_INDEX_PROVIDER_NAME);
    requireNonNull(configuredTokenIndexProvider, () -> format("Token index provider: `%s` not found. Available index providers: %s.", TOKEN_INDEX_PROVIDER_NAME, indexProvidersByName.keySet().toString()));
    tokenIndexProvider = configuredTokenIndexProvider;
}
Also used : IndexProvider(org.neo4j.kernel.api.index.IndexProvider) TokenIndexProvider(org.neo4j.kernel.impl.index.schema.TokenIndexProvider)

Aggregations

IndexProvider (org.neo4j.kernel.api.index.IndexProvider)40 Test (org.junit.jupiter.api.Test)23 IndexDescriptor (org.neo4j.internal.schema.IndexDescriptor)17 IndexProviderDescriptor (org.neo4j.internal.schema.IndexProviderDescriptor)14 ArrayList (java.util.ArrayList)9 Config (org.neo4j.configuration.Config)9 GenericNativeIndexProvider (org.neo4j.kernel.impl.index.schema.GenericNativeIndexProvider)7 DefaultIndexProviderMap (org.neo4j.kernel.impl.transaction.state.DefaultIndexProviderMap)7 CursorContext (org.neo4j.io.pagecache.context.CursorContext)6 IndexSample (org.neo4j.kernel.api.index.IndexSample)6 TokenNameLookup (org.neo4j.common.TokenNameLookup)5 IndexAccessor (org.neo4j.kernel.api.index.IndexAccessor)5 TokenIndexProvider (org.neo4j.kernel.impl.index.schema.TokenIndexProvider)5 JobScheduler (org.neo4j.scheduler.JobScheduler)5 IOException (java.io.IOException)4 String.format (java.lang.String.format)4 Arrays (java.util.Arrays)4 HashMap (java.util.HashMap)4 List (java.util.List)4 Map (java.util.Map)4