Examples with ITConnection com.google.cloud.spanner.connection.ITAbstractSpannerTest.ITConnection used on opensource projects

Example 6 with ITConnection

use of com.google.cloud.spanner.connection.ITAbstractSpannerTest.ITConnection in project java-spanner by googleapis.

the class AbortedTest method testAbortedDuringRetryOfFailedUpdateAsFirstStatement.

@Test
public void testAbortedDuringRetryOfFailedUpdateAsFirstStatement() {
    final Statement invalidStatement = Statement.of("INSERT INTO FOO");
    StatusRuntimeException notFound = Status.NOT_FOUND.withDescription("Table not found").asRuntimeException();
    mockSpanner.putStatementResult(StatementResult.exception(invalidStatement, notFound));
    try (ITConnection connection = createConnection(createAbortRetryListener(2, invalidStatement, notFound))) {
        try {
            connection.execute(invalidStatement);
            fail("missing expected exception");
        } catch (SpannerException e) {
            assertThat(e.getErrorCode()).isEqualTo(ErrorCode.NOT_FOUND);
        }
        connection.execute(INSERT_STATEMENT);
        // Force an abort and retry.
        mockSpanner.abortNextStatement();
        connection.commit();
    }
    assertThat(mockSpanner.countRequestsOfType(CommitRequest.class)).isEqualTo(2);
    assertThat(mockSpanner.countRequestsOfType(ExecuteSqlRequest.class)).isEqualTo(8);
}

Example 7 with ITConnection

use of com.google.cloud.spanner.connection.ITAbstractSpannerTest.ITConnection in project java-spanner by googleapis.

the class AbortedTest method createConnection.

ITConnection createConnection(TransactionRetryListener listener) {
    ITConnection connection = super.createConnection(ImmutableList.of(), ImmutableList.of(listener));
    connection.setAutocommit(false);
    return connection;
}

Example 8 with ITConnection

use of com.google.cloud.spanner.connection.ITAbstractSpannerTest.ITConnection in project java-spanner by googleapis.

the class AbortedTest method testAbortedDuringRetryOfFailedQuery.

@Test
public void testAbortedDuringRetryOfFailedQuery() {
    final Statement invalidStatement = Statement.of("SELECT * FROM FOO");
    StatusRuntimeException notFound = Status.NOT_FOUND.withDescription("Table not found").asRuntimeException();
    mockSpanner.putStatementResult(StatementResult.exception(invalidStatement, notFound));
    try (ITConnection connection = createConnection(createAbortFirstRetryListener(invalidStatement, notFound))) {
        connection.execute(INSERT_STATEMENT);
        try (ResultSet rs = connection.executeQuery(invalidStatement)) {
            fail("missing expected exception");
        } catch (SpannerException e) {
            assertThat(e.getErrorCode()).isEqualTo(ErrorCode.NOT_FOUND);
        }
        // Force an abort and retry.
        mockSpanner.abortNextStatement();
        connection.commit();
    }
    assertThat(mockSpanner.countRequestsOfType(CommitRequest.class)).isEqualTo(2);
    // The transaction will be executed 3 times, which means that there will be 6
    // ExecuteSqlRequests:
    // 1. The initial attempt.
    // 2. The first retry attempt. This will fail on the invalid statement as it is aborted.
    // 3. the second retry attempt. This will succeed.
    assertThat(mockSpanner.countRequestsOfType(ExecuteSqlRequest.class)).isEqualTo(6);
}

Example 9 with ITConnection

use of com.google.cloud.spanner.connection.ITAbstractSpannerTest.ITConnection in project java-spanner by googleapis.

the class AbortedTest method testRetryUsesTags.

@Test
public void testRetryUsesTags() {
    mockSpanner.putStatementResult(StatementResult.query(SELECT_COUNT_STATEMENT, SELECT_COUNT_RESULTSET_BEFORE_INSERT));
    mockSpanner.putStatementResult(StatementResult.update(INSERT_STATEMENT, UPDATE_COUNT));
    try (ITConnection connection = createConnection()) {
        connection.setTransactionTag("transaction-tag");
        connection.setStatementTag("statement-tag");
        connection.executeUpdate(INSERT_STATEMENT);
        connection.setStatementTag("statement-tag");
        connection.executeBatchUpdate(Collections.singleton(INSERT_STATEMENT));
        connection.setStatementTag("statement-tag");
        connection.executeQuery(SELECT_COUNT_STATEMENT);
        mockSpanner.abortNextStatement();
        connection.commit();
    }
    long executeSqlRequestCount = mockSpanner.getRequestsOfType(ExecuteSqlRequest.class).stream().filter(request -> request.getRequestOptions().getRequestTag().equals("statement-tag") && request.getRequestOptions().getTransactionTag().equals("transaction-tag")).count();
    assertEquals(4L, executeSqlRequestCount);
    long executeBatchSqlRequestCount = mockSpanner.getRequestsOfType(ExecuteBatchDmlRequest.class).stream().filter(request -> request.getRequestOptions().getRequestTag().equals("statement-tag") && request.getRequestOptions().getTransactionTag().equals("transaction-tag")).count();
    assertEquals(2L, executeBatchSqlRequestCount);
    long commitRequestCount = mockSpanner.getRequestsOfType(CommitRequest.class).stream().filter(request -> request.getRequestOptions().getRequestTag().equals("") && request.getRequestOptions().getTransactionTag().equals("transaction-tag")).count();
    assertEquals(2L, commitRequestCount);
}

Example 10 with ITConnection

use of com.google.cloud.spanner.connection.ITAbstractSpannerTest.ITConnection in project java-spanner by googleapis.

the class AbortedTest method testAbortedDuringRetryOfFailedQueryAsFirstStatement.

@Test
public void testAbortedDuringRetryOfFailedQueryAsFirstStatement() {
    final Statement invalidStatement = Statement.of("SELECT * FROM FOO");
    StatusRuntimeException notFound = Status.NOT_FOUND.withDescription("Table not found").asRuntimeException();
    mockSpanner.putStatementResult(StatementResult.exception(invalidStatement, notFound));
    try (ITConnection connection = createConnection(createAbortRetryListener(2, invalidStatement, notFound))) {
        try (ResultSet rs = connection.executeQuery(invalidStatement)) {
            fail("missing expected exception");
        } catch (SpannerException e) {
            assertThat(e.getErrorCode()).isEqualTo(ErrorCode.NOT_FOUND);
        }
        connection.executeUpdate(INSERT_STATEMENT);
        // Force an abort and retry.
        mockSpanner.abortNextStatement();
        connection.commit();
    }
    assertThat(mockSpanner.countRequestsOfType(CommitRequest.class)).isEqualTo(2);
    // 6 times invalid query + 2 times INSERT.
    assertThat(mockSpanner.countRequestsOfType(ExecuteSqlRequest.class)).isEqualTo(8);
}