Spring中带有绑定线程连接的事务

Question

我想将一个连接绑定到一个线程，并将该连接用于任何JdbcTemplate调用，最终提交更改或执行回滚。

我从Groovy脚本声明所有的句子，所以我不能控制有多少SQL查询将被调用，这就是为什么我必须使用这个方法，而不是一个TransactionalTemplate或类似的东西。这个脚本将调用一个使用该连接和JdbcTemplate的helper类，让我们调用这个类SqlHelper。

现在我的非按需工作解决方案是从groovy脚本调用SqlHelper初始化事务:

initTransaction(ecommerce)

调用

public void initTransaction(DataSource dataSource) {
    DefaultTransactionDefinition transactionDefinition = new DefaultTransactionDefinition();
    transactionDefinition.setReadOnly(false);
    transactionDefinition.setIsolationLevel(TransactionDefinition.ISOLATION_SERIALIZABLE);
    // dataSourceTransactionManager.setDataSource(dataSource);
    // dataSourceTransactionManager.setRollbackOnCommitFailure(true);
    // dataSourceTransactionManager.setTransactionSynchronization(TransactionSynchronization.STATUS_COMMITTED);
    // dataSourceTransactionManager.setDataSource(dataSource);
    try {
        Connection connection = DataSourceUtils.getConnection(dataSource);
        connection.setAutoCommit(false);
        DataSourceUtils.prepareConnectionForTransaction(connection, transactionDefinition);
    } catch (CannotGetJdbcConnectionException e) {
        throw new RuntimeException(e);
    } catch (SQLException e) {
        throw new RuntimeException(e);
    }
}

之后脚本将调用一些SQL操作，如

sqlUpdate(ecommerce, insertSentence, insertParams)

调用

public Integer update(DataSource dataSource, String sql, Map<String, Object> paramMap) {
    return new NamedParameterJdbcTemplate(dataSource).update(sql, paramMap);
}

最后，我想使用

完成提交更改的事务

commitTransaction(dataSource)

调用

public void commitTransaction(DataSource dataSource) {
     Connection connection = DataSourceUtils.getConnection(dataSource);
     try {
     connection.commit();
     } catch (Exception e) {
     rollbackTransaction(dataSource);
     }
    // DataSourceUtils.resetConnectionAfterTransaction(connection, TransactionDefinition.ISOLATION_DEFAULT);
    //        SimpleTransactionStatus transactionStatus = new SimpleTransactionStatus(false);
    //        try {
    //            dataSourceTransactionManager.commit(transactionStatus);
    //            jta.commit(transactionStatus);
    //        } catch (TransactionException e) {
    //             dataSourceTransactionManager.rollback(transactionStatus);
    //            throw new RuntimeException(e);
    //        }
}
private void rollbackTransaction(DataSource dataSource) {
    Connection connection = DataSourceUtils.getConnection(dataSource);
    try {
        connection.rollback();
    } catch (SQLException e) {
        throw new RuntimeException(e);
    }
    DataSourceUtils.resetConnectionAfterTransaction(connection, TransactionDefinition.ISOLATION_DEFAULT);
}

我留下了一些测试的注释块，以向您展示我尝试了哪些方法。我不是很清楚Spring事务是如何工作的，所以我只是尝试不同的东西，并试图了解所有这些东西是如何工作的……如果你需要，我可以提供更多的信息;)

提前谢谢你。

正如M. Denium所建议的，这就是我现在所拥有的:

我声明了变量，使用TransactionStatus作为ThreadSafe，最终解决为:

private DataSourceTransactionManager dataSourceTransactionManager = null;
private DefaultTransactionDefinition transactionDefinition = null;
private static final ThreadLocal<TransactionStatus> transactionStatus = new ThreadLocal<TransactionStatus>() {
    @Override
    protected TransactionStatus initialValue() {
        return null;
    }
};

然后从Groovy脚本中使用相同的调用，使用helper方法:

public void initTransaction(DataSource dataSource) {
    dataSourceTransactionManager = new DataSourceTransactionManager();
    transactionDefinition = new DefaultTransactionDefinition();
    transactionDefinition.setReadOnly(false);
    transactionDefinition.setIsolationLevel(TransactionDefinition.ISOLATION_SERIALIZABLE);
    dataSourceTransactionManager.setRollbackOnCommitFailure(true);
    dataSourceTransactionManager.setTransactionSynchronization(TransactionSynchronization.STATUS_UNKNOWN);
    dataSourceTransactionManager.setDataSource(dataSource);
    transactionStatus.set(dataSourceTransactionManager.getTransaction(null));
}
public void commitTransaction() {
    try {
        LOG.info("Finishing transaction...");
        dataSourceTransactionManager.commit(transactionStatus.get());
        dataSourceTransactionManager.getDataSource().getConnection().close();
        LOG.info("Done.");
    } catch (Throwable e) {
        dataSourceTransactionManager.rollback(transactionStatus.get());
        throw new RuntimeException("An exception during transaction. Rolling back.");
    }
}

Answer 1

您正在尝试重新实现已经由Spring的事务抽象实现的东西。只需使用适当的PlatformTransactionManager(您可能可以从ApplicationContext中获取)，保持对TransactionStatus的引用而不是DataSource，并使用它来提交/回滚。

public TransactionStatus initTransaction() {
    return transactionManager.getTransaction(null);
}
public void commit(TransactionStatus status) {
    transactionManager.commit(status);
}

除了传递TransactionStatus之外，您还可以将其存储在ThreadLocal中并在commit方法中检索它。这将减轻痛苦。

另一个提示，你不应该创建JdbcTemplate s和NamedParameterJdbcTemplate s，这些是沉重的对象创建。在构造过程中，它们会咨询连接，以确定异常转换需要哪个数据库和版本。所以性能方面，这不是一个明智的做法。创建一个实例并重用，模板是线程安全的，所以你只需要一个实例。

然而，我强烈建议你应该使用Groovy，而不是试图绕过它。Groovy有Sql类可以帮助您。您已经可以访问DataSource，所以做这样的事情将是所有需要的。

def sql = new Sql(dataSource);
sql.withTransaction {
    sql.execute "INSERT INTO city (name, state, founded_year) VALUES ('Minneapolis', 'Minnesota', 1867)"
    sql.execute "INSERT INTO city (name, state, founded_year) VALUES ('Orlando', 'Florida', 1875)"
    sql.execute "INSERT INTO city (name, state, founded_year) VALUES ('Gulfport', 'Mississippi', 1887)"
}

这是普通的Groovy，不需要开发额外的类或编写大量的文档来使其工作。只是Groovy…