Examples with TransactionState com.scalar.db.api.TransactionState used on opensource projects

Example 6 with TransactionState

use of com.scalar.db.api.TransactionState in project scalardb by scalar-labs.

the class TwoPhaseCommitTransactionServiceWithTwoPhaseConsensusCommitIntegrationTest method abort_forOngoingTransaction_ShouldAbortCorrectly.

@Test
public void abort_forOngoingTransaction_ShouldAbortCorrectly() throws TransactionException {
    // Arrange
    GrpcTwoPhaseCommitTransaction transaction = manager.start();
    transaction.get(prepareGet(0, 0, TABLE_1));
    transaction.put(preparePut(0, 0, TABLE_1).withValue(BALANCE, 1));
    // Act
    manager.abort(transaction.getId());
    transaction.prepare();
    assertThatCode(transaction::commit).isInstanceOf(CommitException.class);
    transaction.rollback();
    // Assert
    TransactionState state = manager.getState(transaction.getId());
    assertThat(state).isEqualTo(TransactionState.ABORTED);
}

Example 7 with TransactionState

use of com.scalar.db.api.TransactionState in project scalardb by scalar-labs.

the class TwoPhaseCommitTransactionServiceWithTwoPhaseConsensusCommitIntegrationTest method getState_forFailedTransaction_ShouldReturnAbortedState.

@Test
public void getState_forFailedTransaction_ShouldReturnAbortedState() throws TransactionException {
    // Arrange
    GrpcTwoPhaseCommitTransaction transaction1 = manager.start();
    transaction1.get(prepareGet(0, 0, TABLE_1));
    transaction1.put(preparePut(0, 0, TABLE_1).withValue(BALANCE, 1));
    GrpcTwoPhaseCommitTransaction transaction2 = manager.start();
    transaction2.get(prepareGet(0, 0, TABLE_1));
    transaction2.put(preparePut(0, 0, TABLE_1).withValue(BALANCE, 1));
    transaction2.prepare();
    transaction2.commit();
    assertThatCode(transaction1::prepare).isInstanceOf(PreparationException.class);
    transaction1.rollback();
    // Act
    TransactionState state = manager.getState(transaction1.getId());
    // Assert
    assertThat(state).isEqualTo(TransactionState.ABORTED);
}

Example 8 with TransactionState

use of com.scalar.db.api.TransactionState in project scalardb by scalar-labs.

the class TwoPhaseCommitTransactionService method abort.

@Override
public void abort(AbortRequest request, StreamObserver<AbortResponse> responseObserver) {
    execute(() -> {
        TransactionState state = manager.abort(request.getTransactionId());
        responseObserver.onNext(AbortResponse.newBuilder().setState(ProtoUtils.toTransactionState(state)).build());
        responseObserver.onCompleted();
    }, responseObserver, "abort");
}

Example 9 with TransactionState

use of com.scalar.db.api.TransactionState in project scalardb by scalar-labs.

the class TwoPhaseConsensusCommitManagerTest method check_StateReturned_ReturnTheState.

@Test
public void check_StateReturned_ReturnTheState() throws CoordinatorException {
    // Arrange
    TransactionState expected = TransactionState.COMMITTED;
    when(coordinator.getState(ANY_TX_ID)).thenReturn(Optional.of(new State(ANY_TX_ID, expected)));
    // Act
    TransactionState actual = manager.getState(ANY_TX_ID);
    // Assert
    assertThat(actual).isEqualTo(expected);
}

Example 10 with TransactionState

use of com.scalar.db.api.TransactionState in project scalardb by scalar-labs.

the class DistributedTransactionService method getState.

@Override
public void getState(GetTransactionStateRequest request, StreamObserver<GetTransactionStateResponse> responseObserver) {
    execute(() -> {
        TransactionState state = manager.getState(request.getTransactionId());
        responseObserver.onNext(GetTransactionStateResponse.newBuilder().setState(ProtoUtils.toTransactionState(state)).build());
        responseObserver.onCompleted();
    }, responseObserver, "get_state");
}