Examples with TransactionState org.neo4j.kernel.api.txstate.TransactionState used on opensource projects

Example 31 with TransactionState

use of org.neo4j.kernel.api.txstate.TransactionState in project neo4j by neo4j.

the class StateHandlingStatementOperationsTest method shouldConsiderTransactionStateDuringIndexSeekWithIndexQuery.

@Test
public void shouldConsiderTransactionStateDuringIndexSeekWithIndexQuery() throws Exception {
    // Given
    TransactionState txState = mock(TransactionState.class);
    KernelStatement statement = mock(KernelStatement.class);
    when(statement.hasTxStateWithChanges()).thenReturn(true);
    when(statement.txState()).thenReturn(txState);
    when(txState.indexUpdatesForSeek(index, OrderedPropertyValues.ofUndefined("value"))).thenReturn(new DiffSets<>(Collections.singleton(42L), Collections.singleton(44L)));
    when(txState.addedAndRemovedNodes()).thenReturn(new DiffSets<>(Collections.singleton(45L), Collections.singleton(46L)));
    StoreReadLayer storeReadLayer = mock(StoreReadLayer.class);
    IndexReader indexReader = addMockedIndexReader(statement);
    IndexQuery.ExactPredicate query = IndexQuery.exact(index.schema().getPropertyId(), "value");
    when(indexReader.query(query)).thenReturn(PrimitiveLongCollections.resourceIterator(PrimitiveLongCollections.iterator(43L, 44L, 46L), null));
    StateHandlingStatementOperations context = newTxStateOps(storeReadLayer);
    // When
    PrimitiveLongIterator results = context.indexQuery(statement, index, query);
    // Then
    assertEquals(asSet(42L, 43L), PrimitiveLongCollections.toSet(results));
}

Example 32 with TransactionState

use of org.neo4j.kernel.api.txstate.TransactionState in project neo4j by neo4j.

the class StateHandlingStatementOperationsTest method shouldConsiderTransactionStateDuringIndexRangeSeekByPrefixWithIndexQuery.

@Test
public void shouldConsiderTransactionStateDuringIndexRangeSeekByPrefixWithIndexQuery() throws Exception {
    // Given
    TransactionState txState = mock(TransactionState.class);
    KernelStatement statement = mock(KernelStatement.class);
    when(statement.hasTxStateWithChanges()).thenReturn(true);
    when(statement.txState()).thenReturn(txState);
    when(txState.indexUpdatesForRangeSeekByPrefix(index, "prefix")).thenReturn(new DiffSets<>(Collections.singleton(42L), Collections.singleton(44L)));
    when(txState.addedAndRemovedNodes()).thenReturn(new DiffSets<>(Collections.singleton(45L), Collections.singleton(46L)));
    StoreReadLayer storeReadLayer = mock(StoreReadLayer.class);
    IndexReader indexReader = addMockedIndexReader(statement);
    IndexQuery.StringPrefixPredicate indexQuery = IndexQuery.stringPrefix(index.schema().getPropertyId(), "prefix");
    when(indexReader.query(indexQuery)).thenReturn(PrimitiveLongCollections.resourceIterator(PrimitiveLongCollections.iterator(43L, 44L, 46L), null));
    StateHandlingStatementOperations context = newTxStateOps(storeReadLayer);
    // When
    PrimitiveLongIterator results = context.indexQuery(statement, index, indexQuery);
    // Then
    assertEquals(asSet(42L, 43L), PrimitiveLongCollections.toSet(results));
}

Example 33 with TransactionState

use of org.neo4j.kernel.api.txstate.TransactionState in project neo4j by neo4j.

the class StateHandlingStatementOperationsTest method shouldConsiderTransactionStateDuringIndexBetweenRangeSeekByNumberWithIndexQuery.

@SuppressWarnings("unchecked")
@Test
public void shouldConsiderTransactionStateDuringIndexBetweenRangeSeekByNumberWithIndexQuery() throws Exception {
    // Given
    final int propertyKey = 2;
    final int inRange = 15;
    int lower = 10;
    int upper = 20;
    TransactionState txState = mock(TransactionState.class);
    KernelStatement statement = mock(KernelStatement.class);
    when(statement.hasTxStateWithChanges()).thenReturn(true);
    when(statement.txState()).thenReturn(txState);
    StorageStatement storageStatement = mock(StorageStatement.class);
    when(statement.getStoreStatement()).thenReturn(storageStatement);
    when(txState.indexUpdatesForRangeSeekByNumber(index, lower, true, upper, false)).thenReturn(new DiffSets<>(Collections.singleton(42L), Collections.singleton(44L)));
    when(txState.addedAndRemovedNodes()).thenReturn(new DiffSets<>(Collections.singleton(45L), Collections.singleton(46L)));
    StoreReadLayer storeReadLayer = mock(StoreReadLayer.class);
    when(txState.augmentSingleNodeCursor(any(Cursor.class), anyLong())).thenAnswer(invocationOnMock -> {
        long nodeId = (long) invocationOnMock.getArguments()[1];
        when(txState.augmentSinglePropertyCursor(any(Cursor.class), any(PropertyContainerState.class), eq(propertyKey))).thenReturn(asPropertyCursor(intProperty(propertyKey, inRange)));
        return asNodeCursor(nodeId, nodeId + 20000);
    });
    IndexReader indexReader = addMockedIndexReader(storageStatement);
    IndexQuery.NumberRangePredicate indexQuery = IndexQuery.range(index.schema().getPropertyId(), lower, true, upper, false);
    when(indexReader.query(indexQuery)).thenReturn(PrimitiveLongCollections.resourceIterator(PrimitiveLongCollections.iterator(43L, 44L, 46L), null));
    when(storageStatement.acquireSingleNodeCursor(anyLong())).thenAnswer(invocationOnMock -> {
        long nodeId = (long) invocationOnMock.getArguments()[0];
        when(storeReadLayer.nodeGetProperty(eq(storageStatement), any(NodeItem.class), eq(propertyKey))).thenReturn(asPropertyCursor(intProperty(propertyKey, inRange)));
        return asNodeCursor(nodeId, nodeId + 20000);
    });
    StateHandlingStatementOperations context = newTxStateOps(storeReadLayer);
    // When
    PrimitiveLongIterator results = context.indexQuery(statement, index, indexQuery);
    // Then
    assertEquals(asSet(42L, 43L), PrimitiveLongCollections.toSet(results));
}

Example 34 with TransactionState

use of org.neo4j.kernel.api.txstate.TransactionState in project ongdb by graphfoundation.

the class DefaultRelationshipTraversalCursor method next.

@Override
public boolean next() {
    boolean hasChanges;
    TransactionState txs;
    if (filterState == FilterState.NOT_INITIALIZED) {
        // Initialize filtering:
        // - Read first record
        // - Remember type and direction
        // - Return first relationship if it's not deleted
        // Subsequent relationships need to have same type and direction
        // <- will setup filter state if needed
        hasChanges = hasChanges();
        txs = hasChanges ? read.txState() : null;
        if (filterState == FilterState.NOT_INITIALIZED && filterStore) {
            read.relationshipFull(this, next, pageCursor);
            setupFilterState();
        }
        if (filterState != FilterState.NOT_INITIALIZED && (!hasChanges || !txs.relationshipIsDeletedInThisTx(getId()))) {
            return true;
        }
    } else {
        hasChanges = hasChanges();
        txs = hasChanges ? read.txState() : null;
    }
    // tx-state relationships
    if (hasChanges && addedRelationships.hasNext()) {
        loadFromTxState(addedRelationships.next());
        return true;
    }
    if (hasBufferedData()) {
        // We have buffered data, iterate the chain of buffered records
        return nextBuffered(txs);
    }
    do {
        if (traversingDenseNode()) {
            traverseDenseNode();
        }
        if (next == NO_ID) {
            reset();
            return false;
        }
        read.relationshipFull(this, next, pageCursor);
        computeNext();
    } while ((filterStore && !correctTypeAndDirection()) || (hasChanges && txs.relationshipIsDeletedInThisTx(getId())));
    return true;
}

Example 35 with TransactionState

use of org.neo4j.kernel.api.txstate.TransactionState in project ongdb by graphfoundation.

the class ConstraintIndexCreatorTest method shouldCreateConstraintIndexForSpecifiedProvider.

@Test
public void shouldCreateConstraintIndexForSpecifiedProvider() throws Exception {
    // given
    IndexingService indexingService = mock(IndexingService.class);
    StubKernel kernel = new StubKernel();
    long constraintIndexId = 111;
    when(schemaRead.indexGetCommittedId(indexReference)).thenReturn(constraintIndexId);
    IndexProxy indexProxy = mock(IndexProxy.class);
    when(indexingService.getIndexProxy(constraintIndexId)).thenReturn(indexProxy);
    when(indexingService.getIndexProxy(descriptor)).thenReturn(indexProxy);
    PropertyAccessor propertyAccessor = mock(PropertyAccessor.class);
    ConstraintIndexCreator creator = new ConstraintIndexCreator(() -> kernel, indexingService, propertyAccessor, logProvider);
    IndexProvider.Descriptor providerDescriptor = new IndexProvider.Descriptor("Groovy", "1.2");
    // when
    KernelTransactionImplementation transaction = createTransaction();
    creator.createUniquenessConstraintIndex(transaction, descriptor, providerDescriptor);
    // then
    assertEquals(1, kernel.transactions.size());
    TransactionState transactionState = kernel.transactions.get(0).txState();
    verify(transactionState).indexRuleDoAdd(SchemaIndexDescriptorFactory.uniqueForSchema(descriptor), providerDescriptor);
    verify(schemaRead).index(LABEL_ID, PROPERTY_KEY_ID);
    verify(schemaRead).indexGetCommittedId(any(IndexReference.class));
    verifyNoMoreInteractions(schemaRead);
}