Examples with InternalTransaction org.neo4j.kernel.impl.coreapi.InternalTransaction used on opensource projects

Example 21 with InternalTransaction

use of org.neo4j.kernel.impl.coreapi.InternalTransaction in project neo4j by neo4j.

the class CountsComputerTest method getRelTypeIdsFrom.

private int[] getRelTypeIdsFrom(Transaction tx, RelationshipType... types) throws KernelException {
    KernelTransaction ktx = ((InternalTransaction) tx).kernelTransaction();
    TokenWrite tokenWrite = ktx.tokenWrite();
    int[] ids = new int[types.length];
    tokenWrite.relationshipTypeGetOrCreateForNames(Arrays.stream(types).map(RelationshipType::name).toArray(String[]::new), ids);
    return ids;
}

Example 22 with InternalTransaction

use of org.neo4j.kernel.impl.coreapi.InternalTransaction in project neo4j by neo4j.

the class CountsComputerTest method getLabelIdsFrom.

private int[] getLabelIdsFrom(Transaction tx, Label... labels) throws KernelException {
    KernelTransaction ktx = ((InternalTransaction) tx).kernelTransaction();
    TokenWrite tokenWrite = ktx.tokenWrite();
    int[] ids = new int[labels.length];
    tokenWrite.relationshipTypeGetOrCreateForNames(Arrays.stream(labels).map(Label::name).toArray(String[]::new), ids);
    return ids;
}

Example 23 with InternalTransaction

use of org.neo4j.kernel.impl.coreapi.InternalTransaction in project neo4j by neo4j.

the class StartOldDbOnCurrentVersionAndCreateFusionIndexIT method verifyExpectedProvider.

private static void verifyExpectedProvider(GraphDatabaseAPI db, Label label, IndexProviderDescriptor expectedDescriptor) throws TransactionFailureException {
    try (InternalTransaction tx = (InternalTransaction) db.beginTx();
        KernelTransaction kernelTransaction = tx.kernelTransaction()) {
        TokenRead tokenRead = kernelTransaction.tokenRead();
        SchemaRead schemaRead = kernelTransaction.schemaRead();
        int labelId = tokenRead.nodeLabel(label.name());
        int key1Id = tokenRead.propertyKey(KEY1);
        int key2Id = tokenRead.propertyKey(KEY2);
        IndexDescriptor index = single(schemaRead.index(SchemaDescriptor.forLabel(labelId, key1Id)));
        assertIndexHasExpectedProvider(expectedDescriptor, index);
        index = single(schemaRead.index(SchemaDescriptor.forLabel(labelId, key1Id, key2Id)));
        assertIndexHasExpectedProvider(expectedDescriptor, index);
        tx.commit();
    }
}

Example 24 with InternalTransaction

use of org.neo4j.kernel.impl.coreapi.InternalTransaction in project neo4j by neo4j.

the class ConstraintIndexConcurrencyTest method shouldNotAllowConcurrentViolationOfConstraint.

@Test
void shouldNotAllowConcurrentViolationOfConstraint() throws Exception {
    // Given
    Label label = label("Foo");
    String propertyKey = "bar";
    String conflictingValue = "baz";
    String constraintName = "MyConstraint";
    // a constraint
    try (Transaction tx = db.beginTx()) {
        tx.schema().constraintFor(label).assertPropertyIsUnique(propertyKey).withName(constraintName).create();
        tx.commit();
    }
    // When
    try (Transaction tx = db.beginTx()) {
        KernelTransaction ktx = ((InternalTransaction) tx).kernelTransaction();
        int labelId = ktx.tokenRead().nodeLabel(label.name());
        int propertyKeyId = ktx.tokenRead().propertyKey(propertyKey);
        Read read = ktx.dataRead();
        try (NodeValueIndexCursor cursor = ktx.cursors().allocateNodeValueIndexCursor(ktx.cursorContext(), ktx.memoryTracker())) {
            IndexDescriptor index = ktx.schemaRead().indexGetForName(constraintName);
            IndexReadSession indexSession = ktx.dataRead().indexReadSession(index);
            read.nodeIndexSeek(indexSession, cursor, unconstrained(), PropertyIndexQuery.exact(propertyKeyId, "The value is irrelevant, we just want to perform some sort of lookup against this " + "index"));
        }
        // then let another thread come in and create a node
        threads.execute(db -> {
            try (Transaction transaction = db.beginTx()) {
                transaction.createNode(label).setProperty(propertyKey, conflictingValue);
                transaction.commit();
            }
            return null;
        }, db).get();
        // before we create a node with the same property ourselves - using the same statement that we have
        // already used for lookup against that very same index
        long node = ktx.dataWrite().nodeCreate();
        ktx.dataWrite().nodeAddLabel(node, labelId);
        var e = assertThrows(UniquePropertyValueValidationException.class, () -> ktx.dataWrite().nodeSetProperty(node, propertyKeyId, Values.of(conflictingValue)));
        assertEquals(ConstraintDescriptorFactory.uniqueForLabel(labelId, propertyKeyId), e.constraint());
        IndexEntryConflictException conflict = Iterators.single(e.conflicts().iterator());
        assertEquals(Values.stringValue(conflictingValue), conflict.getSinglePropertyValue());
        tx.commit();
    }
}

Example 25 with InternalTransaction

use of org.neo4j.kernel.impl.coreapi.InternalTransaction in project neo4j by neo4j.

the class NonUniqueIndexTest method concurrentIndexPopulationAndInsertsShouldNotProduceDuplicates.

@Test
void concurrentIndexPopulationAndInsertsShouldNotProduceDuplicates() throws Exception {
    // Given
    GraphDatabaseService db = newEmbeddedGraphDatabaseWithSlowJobScheduler();
    try {
        // When
        try (Transaction tx = db.beginTx()) {
            tx.schema().indexFor(label(LABEL)).on(KEY).withName(INDEX_NAME).create();
            tx.commit();
        }
        Node node;
        try (Transaction tx = db.beginTx()) {
            node = tx.createNode(label(LABEL));
            node.setProperty(KEY, VALUE);
            tx.commit();
        }
        try (Transaction tx = db.beginTx()) {
            tx.schema().awaitIndexesOnline(2, MINUTES);
            tx.commit();
        }
        // Then
        try (Transaction tx = db.beginTx()) {
            KernelTransaction ktx = ((InternalTransaction) tx).kernelTransaction();
            IndexDescriptor index = ktx.schemaRead().indexGetForName(INDEX_NAME);
            IndexReadSession indexSession = ktx.dataRead().indexReadSession(index);
            try (NodeValueIndexCursor cursor = ktx.cursors().allocateNodeValueIndexCursor(ktx.cursorContext(), ktx.memoryTracker())) {
                ktx.dataRead().nodeIndexSeek(indexSession, cursor, unconstrained(), PropertyIndexQuery.exact(1, VALUE));
                assertTrue(cursor.next());
                assertEquals(node.getId(), cursor.nodeReference());
                assertFalse(cursor.next());
            }
            tx.commit();
        }
    } finally {
        managementService.shutdown();
    }
}