/*
* Copyright 2010-2017 Boxfuse GmbH
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.flywaydb.core.internal.util.jdbc;
import org.flywaydb.core.api.FlywayException;
import org.flywaydb.core.internal.dbsupport.FlywaySqlException;
import org.flywaydb.core.internal.util.logging.Log;
import org.flywaydb.core.internal.util.logging.LogFactory;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Savepoint;
import java.util.concurrent.Callable;
/**
* Spring-like template for executing transactions.
*/
public class TransactionTemplate {
private static final Log LOG = LogFactory.getLog(TransactionTemplate.class);
/**
* The connection for the transaction.
*/
private final Connection connection;
/**
* Whether to roll back the transaction when an exception is thrown.
*/
private final boolean rollbackOnException;
/**
* Creates a new transaction template for this connection.
*
* @param connection The connection for the transaction.
*/
public TransactionTemplate(Connection connection) {
this(connection, true);
}
/**
* Creates a new transaction template for this connection.
*
* @param connection The connection for the transaction.
* @param rollbackOnException Whether to roll back the transaction when an exception is thrown.
*/
public TransactionTemplate(Connection connection, boolean rollbackOnException) {
this.connection = connection;
this.rollbackOnException = rollbackOnException;
}
/**
* Executes this callback within a transaction.
*
* @param transactionCallback The callback to execute.
* @return The result of the transaction code.
*/
public <T> T execute(Callable<T> transactionCallback) {
boolean oldAutocommit = true;
try {
oldAutocommit = connection.getAutoCommit();
connection.setAutoCommit(false);
T result = transactionCallback.call();
connection.commit();
return result;
} catch (SQLException e) {
throw new FlywaySqlException("Unable to commit transaction", e);
} catch (Exception e) {
Savepoint savepoint = null;
RuntimeException rethrow;
if (e instanceof RollbackWithSavepointException) {
savepoint = ((RollbackWithSavepointException) e).getSavepoint();
rethrow = (RuntimeException) e.getCause();
} else if (e instanceof RuntimeException) {
rethrow = (RuntimeException) e;
} else {
rethrow = new FlywayException(e);
}
if (rollbackOnException) {
try {
LOG.debug("Rolling back transaction...");
if (savepoint == null) {
connection.rollback();
} else {
connection.rollback(savepoint);
}
LOG.debug("Transaction rolled back");
} catch (SQLException se) {
LOG.error("Unable to rollback transaction", se);
}
} else {
try {
connection.commit();
} catch (SQLException se) {
LOG.error("Unable to commit transaction", se);
}
}
throw rethrow;
} finally {
try {
connection.setAutoCommit(oldAutocommit);
} catch (SQLException e) {
LOG.error("Unable to restore autocommit to original value for connection", e);
}
}
}
}