Examples with Connection io.r2dbc.spi.Connection used on opensource projects

Example 6 with Connection

use of io.r2dbc.spi.Connection in project spring-framework by spring-projects.

the class SingleConnectionFactoryUnitTests method shouldNotSuppressClose.

@Test
void shouldNotSuppressClose() {
    SingleConnectionFactory factory = new SingleConnectionFactory("r2dbc:h2:mem:///foo", false);
    Connection connection = factory.create().block();
    StepVerifier.create(connection.close()).verifyComplete();
    StepVerifier.create(connection.setTransactionIsolationLevel(IsolationLevel.READ_COMMITTED)).verifyError(R2dbcNonTransientResourceException.class);
    factory.destroy();
}

Example 7 with Connection

use of io.r2dbc.spi.Connection in project spring-framework by spring-projects.

the class R2dbcTransactionManagerUnitTests method testRollback.

@Test
void testRollback() {
    AtomicInteger commits = new AtomicInteger();
    when(connectionMock.commitTransaction()).thenReturn(Mono.fromRunnable(commits::incrementAndGet));
    AtomicInteger rollbacks = new AtomicInteger();
    when(connectionMock.rollbackTransaction()).thenReturn(Mono.fromRunnable(rollbacks::incrementAndGet));
    TransactionalOperator operator = TransactionalOperator.create(tm);
    ConnectionFactoryUtils.getConnection(connectionFactoryMock).doOnNext(connection -> {
        throw new IllegalStateException();
    }).as(operator::transactional).as(StepVerifier::create).verifyError(IllegalStateException.class);
    assertThat(commits).hasValue(0);
    assertThat(rollbacks).hasValue(1);
    verify(connectionMock).isAutoCommit();
    verify(connectionMock).beginTransaction();
    verify(connectionMock).rollbackTransaction();
    verify(connectionMock).close();
    verifyNoMoreInteractions(connectionMock);
}

Example 8 with Connection

use of io.r2dbc.spi.Connection in project spring-framework by spring-projects.

the class ScriptUtils method executeSqlScript.

/**
 * Execute the given SQL script.
 * <p>Statement separators and comments will be removed before executing
 * individual statements within the supplied script.
 * <p><strong>Warning</strong>: this method does <em>not</em> release the
 * provided {@link Connection}.
 * @param connection the R2DBC connection to use to execute the script; already
 * configured and ready to use
 * @param resource the resource (potentially associated with a specific encoding)
 * to load the SQL script from
 * @param dataBufferFactory the factory to create data buffers with
 * @param continueOnError whether or not to continue without throwing an exception
 * in the event of an error
 * @param ignoreFailedDrops whether or not to continue in the event of specifically
 * an error on a {@code DROP} statement
 * @param commentPrefixes the prefixes that identify single-line comments in the
 * SQL script (typically "--")
 * @param separator the script statement separator; defaults to
 * {@value #DEFAULT_STATEMENT_SEPARATOR} if not specified and falls back to
 * {@value #FALLBACK_STATEMENT_SEPARATOR} as a last resort; may be set to
 * {@value #EOF_STATEMENT_SEPARATOR} to signal that the script contains a
 * single statement without a separator
 * @param blockCommentStartDelimiter the <em>start</em> block comment delimiter
 * @param blockCommentEndDelimiter the <em>end</em> block comment delimiter
 * @throws ScriptException if an error occurred while executing the SQL script
 * @see #DEFAULT_STATEMENT_SEPARATOR
 * @see #FALLBACK_STATEMENT_SEPARATOR
 * @see #EOF_STATEMENT_SEPARATOR
 * @see org.springframework.r2dbc.connection.ConnectionFactoryUtils#getConnection
 * @see org.springframework.r2dbc.connection.ConnectionFactoryUtils#releaseConnection
 */
public static Mono<Void> executeSqlScript(Connection connection, EncodedResource resource, DataBufferFactory dataBufferFactory, boolean continueOnError, boolean ignoreFailedDrops, String[] commentPrefixes, @Nullable String separator, String blockCommentStartDelimiter, String blockCommentEndDelimiter) throws ScriptException {
    if (logger.isDebugEnabled()) {
        logger.debug("Executing SQL script from " + resource);
    }
    long startTime = System.currentTimeMillis();
    Mono<String> inputScript = readScript(resource, dataBufferFactory, separator).onErrorMap(IOException.class, ex -> new CannotReadScriptException(resource, ex));
    AtomicInteger statementNumber = new AtomicInteger();
    Flux<Void> executeScript = inputScript.flatMapIterable(script -> {
        String separatorToUse = separator;
        if (separatorToUse == null) {
            separatorToUse = DEFAULT_STATEMENT_SEPARATOR;
        }
        if (!EOF_STATEMENT_SEPARATOR.equals(separatorToUse) && !containsStatementSeparator(resource, script, separatorToUse, commentPrefixes, blockCommentStartDelimiter, blockCommentEndDelimiter)) {
            separatorToUse = FALLBACK_STATEMENT_SEPARATOR;
        }
        return splitSqlScript(resource, script, separatorToUse, commentPrefixes, blockCommentStartDelimiter, blockCommentEndDelimiter);
    }).concatMap(statement -> {
        statementNumber.incrementAndGet();
        return runStatement(statement, connection, resource, continueOnError, ignoreFailedDrops, statementNumber);
    });
    if (logger.isDebugEnabled()) {
        executeScript = executeScript.doOnComplete(() -> {
            long elapsedTime = System.currentTimeMillis() - startTime;
            logger.debug("Executed SQL script from " + resource + " in " + elapsedTime + " ms.");
        });
    }
    return executeScript.onErrorMap(ex -> !(ex instanceof ScriptException), ex -> new UncategorizedScriptException("Failed to execute database script from resource [" + resource + "]", ex)).then();
}

Example 9 with Connection

use of io.r2dbc.spi.Connection in project spring-framework by spring-projects.

the class R2dbcTransactionManagerUnitTests method testRollbackFails.

@Test
@SuppressWarnings("unchecked")
void testRollbackFails() {
    when(connectionMock.rollbackTransaction()).thenReturn(Mono.defer(() -> Mono.error(new R2dbcBadGrammarException("Commit should fail"))), Mono.empty());
    TransactionalOperator operator = TransactionalOperator.create(tm);
    operator.execute(reactiveTransaction -> {
        reactiveTransaction.setRollbackOnly();
        return ConnectionFactoryUtils.getConnection(connectionFactoryMock).doOnNext(connection -> connection.createStatement("foo")).then();
    }).as(StepVerifier::create).verifyError(IllegalTransactionStateException.class);
    verify(connectionMock).isAutoCommit();
    verify(connectionMock).beginTransaction();
    verify(connectionMock).createStatement("foo");
    verify(connectionMock, never()).commitTransaction();
    verify(connectionMock).rollbackTransaction();
    verify(connectionMock).close();
    verifyNoMoreInteractions(connectionMock);
}

Example 10 with Connection

use of io.r2dbc.spi.Connection in project spring-framework by spring-projects.

the class TransactionAwareConnectionFactoryProxyUnitTests method shouldEmitBoundConnection.

@Test
void shouldEmitBoundConnection() {
    when(connectionMock1.beginTransaction()).thenReturn(Mono.empty());
    when(connectionMock1.commitTransaction()).thenReturn(Mono.empty());
    when(connectionMock1.close()).thenReturn(Mono.empty());
    TransactionalOperator rxtx = TransactionalOperator.create(tm);
    AtomicReference<Connection> transactionalConnection = new AtomicReference<>();
    TransactionAwareConnectionFactoryProxy proxyCf = new TransactionAwareConnectionFactoryProxy(connectionFactoryMock);
    ConnectionFactoryUtils.getConnection(connectionFactoryMock).doOnNext(transactionalConnection::set).flatMap(connection -> proxyCf.create().doOnNext(wrappedConnection -> assertThat(((Wrapped<?>) wrappedConnection).unwrap()).isSameAs(connection))).as(rxtx::transactional).flatMapMany(Connection::close).as(StepVerifier::create).verifyComplete();
    verifyNoInteractions(connectionMock2);
    verifyNoInteractions(connectionMock3);
    verify(connectionFactoryMock, times(1)).create();
}