All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.flywaydb.database.mysql.MySQLDatabase Maven / Gradle / Ivy
/*-
* ========================LICENSE_START=================================
* flyway-mysql
* ========================================================================
* Copyright (C) 2010 - 2024 Red Gate Software Ltd
* ========================================================================
* 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.
* =========================LICENSE_END==================================
*/
package org.flywaydb.database.mysql;
import static org.flywaydb.core.internal.database.base.DatabaseConstants.DATABASE_HOSTING_AWS_RDS;
import lombok.CustomLog;
import org.flywaydb.core.api.CoreMigrationType;
import org.flywaydb.core.api.FlywayException;
import org.flywaydb.core.api.MigrationVersion;
import org.flywaydb.core.api.configuration.Configuration;
import org.flywaydb.core.extensibility.Tier;
import org.flywaydb.core.internal.database.base.BaseDatabaseType;
import org.flywaydb.core.internal.database.base.Database;
import org.flywaydb.core.internal.database.base.Table;
import org.flywaydb.core.internal.exception.FlywaySqlException;
import org.flywaydb.core.internal.jdbc.JdbcConnectionFactory;
import org.flywaydb.core.internal.jdbc.JdbcTemplate;
import org.flywaydb.core.internal.jdbc.StatementInterceptor;
import org.flywaydb.database.mysql.mariadb.MariaDBDatabaseType;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
@CustomLog
public class MySQLDatabase extends Database {
// See https://mariadb.com/kb/en/version/
private static final Pattern MARIADB_VERSION_PATTERN = Pattern.compile("(\\d+\\.\\d+)\\.\\d+(-\\d+)*-MariaDB(-\\w+)*");
private static final Pattern MARIADB_WITH_MAXSCALE_VERSION_PATTERN = Pattern.compile("(\\d+\\.\\d+)\\.\\d+(-\\d+)* (\\d+\\.\\d+)\\.\\d+(-\\d+)*-maxscale(-\\w+)*");
private static final Pattern MYSQL_VERSION_PATTERN = Pattern.compile("(\\d+\\.\\d+)\\.\\d+\\w*");
/**
* Whether this is a Percona XtraDB Cluster in strict mode.
*/
private final boolean pxcStrict;
/**
* Whether WSREP_ON=ON
*/
private final boolean wsrepOn;
/**
* Whether this database is enforcing GTID consistency.
*/
private final boolean gtidConsistencyEnforced;
/**
* Whether the event scheduler table is queryable.
*/
final boolean eventSchedulerQueryable;
public MySQLDatabase(Configuration configuration, JdbcConnectionFactory jdbcConnectionFactory, StatementInterceptor statementInterceptor) {
super(configuration, jdbcConnectionFactory, statementInterceptor);
JdbcTemplate jdbcTemplate = new JdbcTemplate(rawMainJdbcConnection, databaseType);
pxcStrict = isMySQL() && isRunningInPerconaXtraDBClusterWithStrictMode(jdbcTemplate);
wsrepOn = isMariaDB() && isWsrepOn(jdbcTemplate);
gtidConsistencyEnforced = isMySQL() && isRunningInGTIDConsistencyMode(jdbcTemplate);
eventSchedulerQueryable = isMySQL() || isEventSchedulerQueryable(jdbcTemplate);
}
private static boolean isEventSchedulerQueryable(JdbcTemplate jdbcTemplate) {
try {
// Attempt query
jdbcTemplate.queryForString("SELECT event_name FROM information_schema.events LIMIT 1");
return true;
} catch (SQLException e) {
LOG.debug("Detected unqueryable MariaDB event scheduler, most likely due to it being OFF or DISABLED.");
return false;
}
}
static boolean isRunningInPerconaXtraDBClusterWithStrictMode(JdbcTemplate jdbcTemplate) {
try {
String pcx_strict_mode = jdbcTemplate.queryForString(
"select VARIABLE_VALUE from performance_schema.global_variables"
+ " where variable_name = 'pxc_strict_mode'");
if ("ENFORCING".equals(pcx_strict_mode) || "MASTER".equals(pcx_strict_mode)) {
LOG.debug("Detected Percona XtraDB Cluster in strict mode");
return true;
}
} catch (SQLException e) {
LOG.debug("Unable to detect whether we are running in a Percona XtraDB Cluster. Assuming not to be.");
}
return false;
}
static boolean isWsrepOn(JdbcTemplate jdbcTemplate) {
try {
boolean wsrepOn = jdbcTemplate.queryForBoolean("SELECT @@GLOBAL.WSREP_ON");
if (wsrepOn) {
LOG.debug("Detected WSREP_ON=ON");
return true;
}
} catch (SQLException e) {
LOG.debug("Unable to detect whether WSREP_ON=ON. Assuming not.");
}
return false;
}
static boolean isRunningInGTIDConsistencyMode(JdbcTemplate jdbcTemplate) {
try {
String gtidConsistency = jdbcTemplate.queryForString("SELECT @@GLOBAL.ENFORCE_GTID_CONSISTENCY");
if ("ON".equals(gtidConsistency)) {
LOG.debug("Detected GTID consistency being enforced");
return true;
}
} catch (SQLException e) {
LOG.debug("Unable to detect whether database enforces GTID consistency. Assuming not.");
}
return false;
}
boolean isMySQL() {
return databaseType instanceof MySQLDatabaseType;
}
boolean isMariaDB() {
return databaseType instanceof MariaDBDatabaseType;
}
boolean isPxcStrict() {
return pxcStrict;
}
public boolean isWsrepOn() {
return wsrepOn;
}
/*
* CREATE TABLE ... AS SELECT ... cannot be used in three scenarios:
* - Percona XtraDB Cluster in strict mode doesn't support it
* - TiDB doesn't support it (overridden elsewhere)
* - When GTID consistency is being enforced. Note that if GTID_MODE is ON, then ENFORCE_GTID_CONSISTENCY is
* necessarily ON as well.
*/
protected boolean isCreateTableAsSelectAllowed() {
return !pxcStrict && !gtidConsistencyEnforced;
}
@Override
public String getRawCreateScript(Table table, boolean baseline) {
String tablespace =
!getVersion().isAtLeast("5.5") || configuration.getTablespace() == null ? "" : " TABLESPACE \"" + configuration.getTablespace() + "\"";
String baselineMarker = "";
if (baseline) {
if (isCreateTableAsSelectAllowed()) {
baselineMarker = " AS SELECT" +
" 1 as \"installed_rank\"," +
" '" + configuration.getBaselineVersion() + "' as \"version\"," +
" '" + configuration.getBaselineDescription() + "' as \"description\"," +
" '" + CoreMigrationType.BASELINE + "' as \"type\"," +
" '" + configuration.getBaselineDescription() + "' as \"script\"," +
" NULL as \"checksum\"," +
" '" + getInstalledBy() + "' as \"installed_by\"," +
" CURRENT_TIMESTAMP as \"installed_on\"," +
" 0 as \"execution_time\"," +
" TRUE as \"success\"\n";
} else {
// Revert to regular insert, which unfortunately is not safe in concurrent scenarios
// due to MySQL implicit commits after DDL statements.
baselineMarker = ";\n" + getBaselineStatement(table);
}
}
return "CREATE TABLE " + table + " (\n" +
" `installed_rank` INT NOT NULL,\n" +
" `version` VARCHAR(50),\n" +
" `description` VARCHAR(200) NOT NULL,\n" +
" `type` VARCHAR(20) NOT NULL,\n" +
" `script` VARCHAR(1000) NOT NULL,\n" +
" `checksum` INT,\n" +
" `installed_by` VARCHAR(100) NOT NULL,\n" +
" `installed_on` TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,\n" +
" `execution_time` INT NOT NULL,\n" +
" `success` BOOL NOT NULL,\n" +
" CONSTRAINT " + getConstraintName(table.getName()) + " PRIMARY KEY (`installed_rank`)\n" +
")" + tablespace + " ENGINE=InnoDB" +
baselineMarker +
";\n" +
"CREATE INDEX `" + table.getName() + "_s_idx` ON " + table + " (`success`);";
}
protected String getConstraintName(String tableName) {
return "`" + tableName + "_pk`";
}
@Override
protected MySQLConnection doGetConnection(Connection connection) {
return new MySQLConnection(this, connection);
}
@Override
protected MigrationVersion determineVersion() {
// Ignore the version from the JDBC metadata and use the version returned by the database since proxies such as
// Azure or ProxySQL return incorrect versions
String selectVersionOutput = BaseDatabaseType.getSelectVersionOutput(rawMainJdbcConnection);
if (databaseType instanceof MariaDBDatabaseType) {
return extractMariaDBVersionFromString(selectVersionOutput);
}
return extractMySQLVersionFromString(selectVersionOutput);
}
static MigrationVersion extractMySQLVersionFromString(String selectVersionOutput) {
return extractVersionFromString(selectVersionOutput, MYSQL_VERSION_PATTERN);
}
static MigrationVersion extractMariaDBVersionFromString(String selectVersionOutput) {
return extractVersionFromString(selectVersionOutput, MARIADB_VERSION_PATTERN, MARIADB_WITH_MAXSCALE_VERSION_PATTERN);
}
/*
* Given a version string that may contain unwanted text, extract out the version part.
*/
private static MigrationVersion extractVersionFromString(String versionString, Pattern... patterns) {
for (Pattern pattern : patterns) {
Matcher matcher = pattern.matcher(versionString);
if (matcher.find()) {
return MigrationVersion.fromVersion(matcher.group(1));
}
}
throw new FlywayException("Unable to determine version from '" + versionString + "'");
}
@Override
public void ensureSupported(Configuration configuration) {
ensureDatabaseIsRecentEnough("5.1");
ensureDatabaseNotOlderThanOtherwiseRecommendUpgradeToFlywayEdition("8.0", Tier.PREMIUM, configuration);
recommendFlywayUpgradeIfNecessary("8.1");
}
@Override
protected String doGetCurrentUser() throws SQLException {
return getMainConnection().getJdbcTemplate().queryForString("SELECT SUBSTRING_INDEX(USER(),'@',1)");
}
@Override
public boolean supportsDdlTransactions() {
return false;
}
@Override
public String getBooleanTrue() {
return "1";
}
@Override
public String getBooleanFalse() {
return "0";
}
@Override
public String getOpenQuote() {
return "`";
}
@Override
public String getCloseQuote() {
return "`";
}
@Override
public boolean catalogIsSchema() {
return true;
}
@Override
public boolean useSingleConnection() {
return !pxcStrict;
}
@Override
public String getDatabaseHosting() {
if (getMainConnection().isAwsRds()) {
return DATABASE_HOSTING_AWS_RDS;
} else {
return super.getDatabaseHosting();
}
}
}
