liquibase.statement.ExecutablePreparedStatementBase Maven / Gradle / Ivy
package liquibase.statement;
import liquibase.Scope;
import liquibase.change.ColumnConfig;
import liquibase.changelog.ChangeSet;
import liquibase.database.Database;
import liquibase.database.PreparedStatementFactory;
import liquibase.database.core.PostgresDatabase;
import liquibase.database.core.SQLiteDatabase;
import liquibase.datatype.DataTypeFactory;
import liquibase.datatype.LiquibaseDataType;
import liquibase.exception.DatabaseException;
import liquibase.exception.LiquibaseException;
import liquibase.exception.UnexpectedLiquibaseException;
import liquibase.listener.SqlListener;
import liquibase.logging.Logger;
import liquibase.resource.InputStreamList;
import liquibase.resource.ResourceAccessor;
import liquibase.util.JdbcUtil;
import liquibase.util.StreamUtil;
import liquibase.util.FilenameUtil;
import java.io.*;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.sql.*;
import java.util.*;
import static java.util.ResourceBundle.getBundle;
import liquibase.change.core.LoadDataChange;
public abstract class ExecutablePreparedStatementBase implements ExecutablePreparedStatement {
private static ResourceBundle coreBundle = getBundle("liquibase/i18n/liquibase-core");
protected Database database;
private String catalogName;
private String schemaName;
private String tableName;
private List columns;
private ChangeSet changeSet;
private Set closeables;
private ResourceAccessor resourceAccessor;
protected ExecutablePreparedStatementBase(Database database, String catalogName, String schemaName, String
tableName, List columns, ChangeSet changeSet, ResourceAccessor resourceAccessor) {
this.database = database;
this.changeSet = changeSet;
this.catalogName = catalogName;
this.schemaName = schemaName;
this.tableName = tableName;
this.columns = columns;
this.changeSet = changeSet;
this.closeables = new HashSet<>();
this.resourceAccessor = resourceAccessor;
}
private static InputStream createStream(InputStream in) {
return (in instanceof BufferedInputStream) ? in : new BufferedInputStream(in);
}
@Override
public void execute(PreparedStatementFactory factory) throws DatabaseException {
// build the sql statement
List cols = new ArrayList<>(getColumns().size());
String sql = generateSql(cols);
for (SqlListener listener : Scope.getCurrentScope().getListeners(SqlListener.class)) {
listener.writeSqlWillRun(sql);
}
Scope.getCurrentScope().getLog(getClass()).fine("Number of columns = " + cols.size());
// create prepared statement
PreparedStatement stmt = factory.create(sql);
try {
attachParams(cols, stmt);
// trigger execution
executePreparedStatement(stmt);
} catch (SQLException e) {
throw new DatabaseException(e);
} finally {
for (Closeable closeable : closeables) {
try {
closeable.close();
} catch (IOException ignore) {
}
}
JdbcUtil.closeStatement(stmt);
}
}
protected void executePreparedStatement(PreparedStatement stmt) throws SQLException {
stmt.execute();
}
/**
* Sets the list of bind variables for the execution of a DML statement
* @param cols a list of columns with their designated values
* @param stmt the PreparedStatement to which the values are to be attached
* @throws SQLException if JDBC objects to a setting (non-existent bind number, wrong column type etc.)
* @throws DatabaseException if an I/O error occurs during the read of LOB values
*/
protected void attachParams(List cols, PreparedStatement stmt)
throws SQLException, DatabaseException {
int i = 1; // index starts from 1
for (ColumnConfig col : cols) {
Scope.getCurrentScope().getLog(getClass()).fine("Applying column parameter = " + i + " for column " + col.getName());
applyColumnParameter(stmt, i, col);
i++;
}
}
protected abstract String generateSql(List cols);
/**
* Sets a single bind variable for a statement to its designated value
* @param stmt the PreparedStatement whose parameter is to be set
* @param i the parameter index (first bind variable is 1)
* @param col a ColumnConfig with information about the column, its type, and the desired value
* @throws SQLException if JDBC objects to a setting (non-existent bind number, wrong column type etc.)
* @throws DatabaseException if an I/O error occurs during the read of LOB values
*/
private void applyColumnParameter(PreparedStatement stmt, int i, ColumnConfig col) throws SQLException,
DatabaseException {
final Logger LOG = Scope.getCurrentScope().getLog(getClass());
if (col.getValue() != null) {
LOG.fine("value is string/UUID/blob = " + col.getValue());
if (col.getType() != null && col.getType().equalsIgnoreCase(LoadDataChange.LOAD_DATA_TYPE.UUID.name())) {
stmt.setObject(i, UUID.fromString(col.getValue()));
} else if (col.getType() != null && col.getType().equalsIgnoreCase(LoadDataChange.LOAD_DATA_TYPE.OTHER.name())) {
stmt.setObject(i, col.getValue(), Types.OTHER);
} else if (LoadDataChange.LOAD_DATA_TYPE.BLOB.name().equalsIgnoreCase(col.getType())) {
stmt.setBlob(i, new ByteArrayInputStream(Base64.getDecoder().decode(col.getValue())));
} else if (LoadDataChange.LOAD_DATA_TYPE.CLOB.name().equalsIgnoreCase(col.getType())) {
try {
if (database instanceof PostgresDatabase || database instanceof SQLiteDatabase) {
// JDBC driver does not have the .createClob() call implemented yet
stmt.setString(i, col.getValue());
} else {
Clob clobValue = stmt.getConnection().createClob();
clobValue.setString(1, col.getValue());
stmt.setClob(i, clobValue);
}
} catch (SQLFeatureNotSupportedException e) {
stmt.setString(i, col.getValue());
}
} else {
stmt.setString(i, col.getValue());
}
} else if (col.getValueBoolean() != null) {
LOG.fine("value is boolean = " + col.getValueBoolean());
stmt.setBoolean(i, col.getValueBoolean());
} else if (col.getValueNumeric() != null) {
LOG.fine("value is numeric = " + col.getValueNumeric());
Number number = col.getValueNumeric();
if (number instanceof ColumnConfig.ValueNumeric) {
ColumnConfig.ValueNumeric valueNumeric = (ColumnConfig.ValueNumeric) number;
number = valueNumeric.getDelegate();
}
if (number instanceof Long) {
stmt.setLong(i, number.longValue());
} else if (number instanceof Integer) {
stmt.setInt(i, number.intValue());
} else if (number instanceof Double) {
stmt.setDouble(i, number.doubleValue());
} else if (number instanceof Float) {
stmt.setFloat(i, number.floatValue());
} else if (number instanceof BigDecimal) {
stmt.setBigDecimal(i, (BigDecimal) number);
} else if (number instanceof BigInteger) {
stmt.setInt(i, number.intValue());
} else {
throw new UnexpectedLiquibaseException(
String.format(
coreBundle.getString("jdbc.bind.parameter.unknown.numeric.value.type"),
col.getName(),
col.getValueNumeric().toString(),
col.getValueNumeric().getClass().getName()
)
);
}
} else if (col.getValueDate() != null) {
LOG.fine("value is date = " + col.getValueDate());
if (col.getValueDate() instanceof Timestamp) {
stmt.setTimestamp(i, (Timestamp) col.getValueDate());
} else if (col.getValueDate() instanceof Time) {
stmt.setTime(i, (Time) col.getValueDate());
} else {
stmt.setDate(i, new java.sql.Date(col.getValueDate().getTime()));
}
} else if (col.getValueBlobFile() != null) {
LOG.fine("value is blob = " + col.getValueBlobFile());
try {
LOBContent lob = toBinaryStream(col.getValueBlobFile());
if (lob.length <= Integer.MAX_VALUE) {
stmt.setBinaryStream(i, lob.content, (int) lob.length);
} else {
stmt.setBinaryStream(i, lob.content, lob.length);
}
} catch (IOException | LiquibaseException e) {
throw new DatabaseException(e.getMessage(), e); // wrap
}
} else if (col.getValueClobFile() != null) {
try {
LOG.fine("value is clob = " + col.getValueClobFile());
LOBContent lob = toCharacterStream(col.getValueClobFile(), col.getEncoding());
if (lob.length <= Integer.MAX_VALUE) {
stmt.setCharacterStream(i, lob.content, (int) lob.length);
} else {
stmt.setCharacterStream(i, lob.content, lob.length);
}
} catch (IOException | LiquibaseException e) {
throw new DatabaseException(e.getMessage(), e); // wrap
}
} else {
// NULL values might intentionally be set into a change, we must also add them to the prepared statement
LOG.fine("value is explicit null");
if (col.getType() == null) {
stmt.setNull(i, java.sql.Types.NULL);
return;
}
if (col.getType().toLowerCase().contains("datetime")) {
stmt.setNull(i, java.sql.Types.TIMESTAMP);
} else {
//
// Get the array of aliases and use them to find the
// correct java.sql.Types constant for the call to setNull
//
boolean isSet = false;
LiquibaseDataType dataType = DataTypeFactory.getInstance().fromDescription(col.getType(), database);
String[] aliases = dataType.getAliases();
for (String alias : aliases) {
if (! alias.contains("java.sql.Types")) {
continue;
}
String name = alias.replaceAll("java.sql.Types.","");
try {
JDBCType jdbcType = Enum.valueOf(JDBCType.class, name);
stmt.setNull(i, jdbcType.getVendorTypeNumber());
isSet = true;
}
catch (Exception e) {
//
// fall back to using java.sql.Types.NULL by catching any exceptions
//
}
break;
}
if (! isSet) {
LOG.info(String.format("Using java.sql.Types.NULL to set null value for type %s", dataType.getName()));
stmt.setNull(i, java.sql.Types.NULL);
}
}
}
}
private LOBContent toBinaryStream(String valueLobFile) throws LiquibaseException, IOException {
InputStream in = getResourceAsStream(valueLobFile);
if (in == null) {
throw new DatabaseException("BLOB resource not found: " + valueLobFile);
}
try {
if (in instanceof FileInputStream) {
InputStream bufferedInput = createStream(in);
return new LOBContent<>(bufferedInput, ((FileInputStream) in).getChannel().size());
}
in = createStream(in);
final int IN_MEMORY_THRESHOLD = 100_000;
if (in.markSupported()) {
in.mark(IN_MEMORY_THRESHOLD);
}
long length = getContentLength(in);
if (in.markSupported() && (length <= IN_MEMORY_THRESHOLD)) {
in.reset();
} else {
try {
in.close();
} catch (IOException ignored) {
}
in = getResourceAsStream(valueLobFile);
in = createStream(in);
}
return new LOBContent<>(in, length);
} finally {
if (in != null) {
closeables.add(in);
}
}
}
private LOBContent toCharacterStream(String valueLobFile, String encoding)
throws IOException, LiquibaseException {
InputStream in = getResourceAsStream(valueLobFile);
if (in == null) {
throw new DatabaseException("CLOB resource not found: " + valueLobFile);
}
final int IN_MEMORY_THRESHOLD = 100_000;
Reader reader = null;
try {
reader = StreamUtil.readStreamWithReader(in, encoding);
if (reader.markSupported()) {
reader.mark(IN_MEMORY_THRESHOLD);
}
long length = getContentLength(reader);
if (reader.markSupported() && (length <= IN_MEMORY_THRESHOLD)) {
reader.reset();
} else {
try {
reader.close();
} catch (IOException ignored) {
}
in = getResourceAsStream(valueLobFile);
reader = StreamUtil.readStreamWithReader(in, encoding);
}
return new LOBContent<>(reader, length);
} finally {
if (reader != null) {
closeables.add(reader);
}
if (in != null) {
closeables.add(in);
}
}
}
@java.lang.SuppressWarnings("squid:S2095")
private InputStream getResourceAsStream(String valueLobFile) throws IOException, LiquibaseException {
String fileName = getFileName(valueLobFile);
InputStreamList streams = this.resourceAccessor.openStreams(null, fileName);
if ((streams == null) || streams.isEmpty()) {
return null;
}
if (streams.size() > 1) {
for (InputStream stream : streams) {
stream.close();
}
throw new IOException(streams.size() + " matched " + valueLobFile);
}
return streams.iterator().next();
}
private String getFileName(String fileName) {
String relativeBaseFileName = changeSet.getChangeLog().getPhysicalFilePath();
return FilenameUtil.concat(FilenameUtil.getDirectory(relativeBaseFileName), fileName);
}
@Override
public boolean skipOnUnsupported() {
return false;
}
public String getCatalogName() {
return catalogName;
}
public String getSchemaName() {
return schemaName;
}
public String getTableName() {
return tableName;
}
public List getColumns() {
return columns;
}
public ChangeSet getChangeSet() {
return changeSet;
}
public ResourceAccessor getResourceAccessor() {
return resourceAccessor;
}
protected long getContentLength(InputStream in) throws IOException {
long length = 0;
byte[] buf = new byte[4096];
int bytesRead = in.read(buf);
while (bytesRead > 0) {
length += bytesRead;
bytesRead = in.read(buf);
}
return length;
}
protected long getContentLength(Reader reader) throws IOException {
long length = 0;
char[] buf = new char[2048];
int charsRead = reader.read(buf);
while (charsRead > 0) {
length += charsRead;
charsRead = reader.read(buf);
}
return length;
}
private class LOBContent {
private final T content;
private final long length;
LOBContent(T content, long length) {
this.content = content;
this.length = length;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy