Search in sources :

Example 41 with IndexSample

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

the class BatchInsertTest method shouldRunConstraintPopulationJobAtShutdown.

@ParameterizedTest
@MethodSource("params")
void shouldRunConstraintPopulationJobAtShutdown(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.createDeferredConstraint(label("Hacker")).assertPropertyIsUnique("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, internalUniqueIndex.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 42 with IndexSample

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

the class FullScanNonUniqueIndexSampler method sample.

@Override
public IndexSample sample(CursorContext cursorContext) {
    KEY lowest = layout.newKey();
    lowest.initialize(Long.MIN_VALUE);
    lowest.initValuesAsLowest();
    KEY highest = layout.newKey();
    highest.initialize(Long.MAX_VALUE);
    highest.initValuesAsHighest();
    KEY prev = layout.newKey();
    try (Seeker<KEY, VALUE> seek = gbpTree.seek(lowest, highest, cursorContext)) {
        long sampledValues = 0;
        long uniqueValues = 0;
        // Get the first one so that prev gets initialized
        if (seek.next()) {
            prev = layout.copyKey(seek.key(), prev);
            sampledValues++;
            uniqueValues++;
            // Then do the rest
            while (seek.next()) {
                if (layout.compareValue(prev, seek.key()) != 0) {
                    uniqueValues++;
                    layout.copyKey(seek.key(), prev);
                }
                // else this is a duplicate of the previous one
                sampledValues++;
            }
        }
        return new IndexSample(sampledValues, uniqueValues, sampledValues);
    } catch (IOException e) {
        throw new UncheckedIOException(e);
    }
}
Also used : IndexSample(org.neo4j.kernel.api.index.IndexSample) UncheckedIOException(java.io.UncheckedIOException) UncheckedIOException(java.io.UncheckedIOException) IOException(java.io.IOException)

Example 43 with IndexSample

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

the class NonUniqueDatabaseIndexPopulatorTest method sampleIncludedUpdates.

@Test
void sampleIncludedUpdates() {
    populator = newPopulator();
    List<IndexEntryUpdate<?>> updates = Arrays.asList(add(1, labelSchemaDescriptor, "aaa"), add(2, labelSchemaDescriptor, "bbb"), add(3, labelSchemaDescriptor, "ccc"));
    populator.add(updates, NULL);
    IndexSample sample = populator.sample(NULL);
    assertEquals(new IndexSample(3, 3, 3), sample);
}
Also used : IndexEntryUpdate(org.neo4j.storageengine.api.IndexEntryUpdate) IndexSample(org.neo4j.kernel.api.index.IndexSample) Test(org.junit.jupiter.api.Test)

Example 44 with IndexSample

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

the class NonUniqueLuceneIndexPopulatingUpdaterIT method shouldSampleRemovals.

@Test
void shouldSampleRemovals() throws Exception {
    // Given
    var provider = createIndexProvider();
    var populator = getPopulator(provider, SCHEMA_DESCRIPTOR);
    // When
    try (var updater = populator.newPopulatingUpdater(mock(NodePropertyAccessor.class), NULL)) {
        updater.process(add(1, SCHEMA_DESCRIPTOR, "foo"));
        updater.process(add(2, SCHEMA_DESCRIPTOR, "bar"));
        updater.process(add(3, SCHEMA_DESCRIPTOR, "baz"));
        updater.process(add(4, SCHEMA_DESCRIPTOR, "qux"));
        updater.process(remove(1, SCHEMA_DESCRIPTOR, "foo"));
        updater.process(remove(2, SCHEMA_DESCRIPTOR, "bar"));
        updater.process(remove(4, SCHEMA_DESCRIPTOR, "qux"));
    }
    // Then samples calculated with documents pending merge
    assertThat(populator.sample(NULL)).isEqualTo(new IndexSample(1, 4, 4));
}
Also used : IndexSample(org.neo4j.kernel.api.index.IndexSample) NodePropertyAccessor(org.neo4j.storageengine.api.NodePropertyAccessor) Test(org.junit.jupiter.api.Test)

Example 45 with IndexSample

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

the class NonUniqueLuceneIndexPopulatingUpdaterIT method shouldSampleCompositeIndex.

@Test
void shouldSampleCompositeIndex() throws Exception {
    // Given
    var provider = createIndexProvider();
    var populator = getPopulator(provider, COMPOSITE_SCHEMA_DESCRIPTOR);
    // When
    try (var updater = populator.newPopulatingUpdater(mock(NodePropertyAccessor.class), NULL)) {
        updater.process(add(1, COMPOSITE_SCHEMA_DESCRIPTOR, "bit", "foo"));
        updater.process(add(2, COMPOSITE_SCHEMA_DESCRIPTOR, "bit", "bar"));
        updater.process(add(3, COMPOSITE_SCHEMA_DESCRIPTOR, "bit", "baz"));
        updater.process(add(4, COMPOSITE_SCHEMA_DESCRIPTOR, "bit", "qux"));
        updater.process(remove(1, COMPOSITE_SCHEMA_DESCRIPTOR, "bit", "foo"));
        updater.process(remove(2, COMPOSITE_SCHEMA_DESCRIPTOR, "bit", "bar"));
        updater.process(change(3, COMPOSITE_SCHEMA_DESCRIPTOR, new Object[] { "bit", "baz" }, new Object[] { "foo", "bar" }));
    }
    // Then samples calculated with documents pending merge
    assertThat(populator.sample(NULL)).isEqualTo(new IndexSample(2, 5, 10));
}
Also used : IndexSample(org.neo4j.kernel.api.index.IndexSample) NodePropertyAccessor(org.neo4j.storageengine.api.NodePropertyAccessor) Test(org.junit.jupiter.api.Test)

Aggregations

IndexSample (org.neo4j.kernel.api.index.IndexSample)56 Test (org.junit.jupiter.api.Test)43 IndexDescriptor (org.neo4j.internal.schema.IndexDescriptor)14 CursorContext (org.neo4j.io.pagecache.context.CursorContext)13 IndexPopulator (org.neo4j.kernel.api.index.IndexPopulator)10 IndexAccessor (org.neo4j.kernel.api.index.IndexAccessor)8 IndexEntryUpdate (org.neo4j.storageengine.api.IndexEntryUpdate)8 NodePropertyAccessor (org.neo4j.storageengine.api.NodePropertyAccessor)8 IndexProvider (org.neo4j.kernel.api.index.IndexProvider)7 IndexSampler (org.neo4j.kernel.api.index.IndexSampler)7 ArrayList (java.util.ArrayList)6 ParameterizedTest (org.junit.jupiter.params.ParameterizedTest)6 Config (org.neo4j.configuration.Config)6 Transaction (org.neo4j.graphdb.Transaction)6 IOException (java.io.IOException)5 Path (java.nio.file.Path)4 Arrays.asList (java.util.Arrays.asList)4 HashSet (java.util.HashSet)4 Set (java.util.Set)4 Assertions.assertThat (org.assertj.core.api.Assertions.assertThat)4