Examples with LabelScanWriter org.neo4j.kernel.api.labelscan.LabelScanWriter used on opensource projects

Example 6 with LabelScanWriter

use of org.neo4j.kernel.api.labelscan.LabelScanWriter in project neo4j by neo4j.

the class LabelScanStoreTest method prepareIndex.

private void prepareIndex() throws IOException {
    start();
    try (LabelScanWriter labelScanWriter = store.newWriter()) {
        labelScanWriter.write(NodeLabelUpdate.labelChanges(1, new long[] {}, new long[] { 1 }));
    }
    store.shutdown();
}

Example 7 with LabelScanWriter

use of org.neo4j.kernel.api.labelscan.LabelScanWriter in project neo4j by neo4j.

the class NativeLabelScanStoreIT method randomModifications.

private void randomModifications(long[] expected, int count) throws IOException {
    BitSet editedNodes = new BitSet();
    try (LabelScanWriter writer = store.newWriter()) {
        for (int i = 0; i < count; i++) {
            int nodeId = random.nextInt(NODE_COUNT);
            if (editedNodes.get(nodeId)) {
                i--;
                continue;
            }
            int changeSize = random.nextInt(3) + 1;
            long labels = expected[nodeId];
            long[] labelsBefore = getLabels(labels);
            for (int j = 0; j < changeSize; j++) {
                labels = flipRandom(labels, LABEL_COUNT, random.random());
            }
            long[] labelsAfter = getLabels(labels);
            editedNodes.set(nodeId);
            NodeLabelUpdate labelChanges = labelChanges(nodeId, labelsBefore, labelsAfter);
            writer.write(labelChanges);
            expected[nodeId] = labels;
        }
    }
}

Example 8 with LabelScanWriter

use of org.neo4j.kernel.api.labelscan.LabelScanWriter in project neo4j by neo4j.

the class IndexBatchTransactionApplierTest method shouldProvideLabelScanStoreUpdatesSortedByNodeId.

@Test
public void shouldProvideLabelScanStoreUpdatesSortedByNodeId() throws Exception {
    // GIVEN
    IndexingService indexing = mock(IndexingService.class);
    LabelScanWriter writer = new OrderVerifyingLabelScanWriter(10, 15, 20);
    WorkSync<Supplier<LabelScanWriter>, LabelUpdateWork> labelScanSync = spy(new WorkSync<>(singletonProvider(writer)));
    WorkSync<IndexingService, IndexUpdatesWork> indexUpdatesSync = new WorkSync<>(indexing);
    TransactionToApply tx = mock(TransactionToApply.class);
    PropertyStore propertyStore = mock(PropertyStore.class);
    try (IndexBatchTransactionApplier applier = new IndexBatchTransactionApplier(indexing, labelScanSync, indexUpdatesSync, mock(NodeStore.class), mock(PropertyLoader.class), new PropertyPhysicalToLogicalConverter(propertyStore), TransactionApplicationMode.INTERNAL)) {
        try (TransactionApplier txApplier = applier.startTx(tx)) {
            // WHEN
            txApplier.visitNodeCommand(node(15));
            txApplier.visitNodeCommand(node(20));
            txApplier.visitNodeCommand(node(10));
        }
    }
    // THEN all assertions happen inside the LabelScanWriter#write and #close
    verify(labelScanSync).apply(any());
}

Example 9 with LabelScanWriter

use of org.neo4j.kernel.api.labelscan.LabelScanWriter in project neo4j by neo4j.

the class NodeRangeDocumentLabelScanStorageStrategyTest method shouldRemoveLabelFieldsThatDoesNotRepresentAnyNodes.

@Test
public void shouldRemoveLabelFieldsThatDoesNotRepresentAnyNodes() throws Exception {
    // given
    IndexWriter indexWriter = mock(IndexWriter.class);
    Document doc = document(format.rangeField(0), format.labelField(7, 0x1), format.labelField(8, 0x1));
    WritableIndexPartition partition = newIndexPartitionMock(indexWriter, doc);
    WritableDatabaseLabelScanIndex index = buildLuceneIndex(partition);
    LabelScanWriter writer = new PartitionedLuceneLabelScanWriter(index, format);
    // when
    writer.write(labelChanges(0, labels(7, 8), labels(8)));
    writer.close();
    // then
    verify(indexWriter).updateDocument(eq(format.rangeTerm(0)), match(document(format.rangeField(0), format.labelField(8, 0x01), format.labelSearchField(8))));
}

Example 10 with LabelScanWriter

use of org.neo4j.kernel.api.labelscan.LabelScanWriter in project neo4j by neo4j.

the class NodeRangeDocumentLabelScanStorageStrategyTest method shouldStoreAnyNodeIdInRange.

@Test
public void shouldStoreAnyNodeIdInRange() throws Exception {
    for (int i = 0, max = 1 << format.bitmapFormat().shift; i < max; i++) {
        // given
        IndexWriter indexWriter = mock(IndexWriter.class);
        WritableIndexPartition partition = newIndexPartitionMock(indexWriter);
        WritableDatabaseLabelScanIndex index = buildLuceneIndex(partition);
        LabelScanWriter writer = new PartitionedLuceneLabelScanWriter(index, format);
        // when
        writer.write(labelChanges(i, labels(), labels(7)));
        writer.close();
        // then
        Document document = new Document();
        format.addRangeValuesField(document, 0);
        format.addLabelAndSearchFields(document, 7, new Bitmap(1L << i));
        verify(indexWriter).updateDocument(eq(format.rangeTerm(0)), match(document));
    }
}