org.flywaydb.database.mongodb.MongoDBDatabase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flyway-database-mongodb Show documentation
The newest version!
/*-
 * ========================LICENSE_START=================================
 * flyway-database-mongodb
 * ========================================================================
 * 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.mongodb;

import java.sql.SQLException;
import static org.flywaydb.core.internal.database.base.DatabaseConstants.DATABASE_HOSTING_MONGODB_ATLAS;

import org.flywaydb.core.api.CoreMigrationType;
import org.flywaydb.core.api.configuration.Configuration;
import org.flywaydb.core.internal.database.base.Database;
import org.flywaydb.core.internal.database.base.Table;
import org.flywaydb.core.internal.jdbc.JdbcConnectionFactory;
import org.flywaydb.core.internal.jdbc.StatementInterceptor;
import org.flywaydb.core.internal.util.AbbreviationUtils;

import java.sql.Timestamp;
import java.time.Instant;
import org.flywaydb.core.internal.util.Pair;

public class MongoDBDatabase extends Database {

    public MongoDBDatabase(Configuration configuration, JdbcConnectionFactory jdbcConnectionFactory, StatementInterceptor statementInterceptor) {
        super(configuration, jdbcConnectionFactory, statementInterceptor);
    }

    @Override
    protected MongoDBConnection doGetConnection(java.sql.Connection connection) {
        return new MongoDBConnection(this, connection);
    }

    @Override
    protected String doGetCatalog() throws SQLException {
        return this.getMainConnection().getDatabaseName();
    }

    @Override
    public void ensureSupported(Configuration configuration) {

    }

    @Override
    public boolean supportsDdlTransactions() {
        return false;
    }

    @Override
    public String getBooleanTrue() {
        return "true";
    }

    @Override
    public String getBooleanFalse() {
        return "false";
    }

    @Override
    public boolean catalogIsSchema() {
        return true;
    }

    @Override
    public String getRawCreateScript(Table table, boolean baseline) {
        return "db.getSiblingDB('" + table.getSchema().getName() + "').createCollection(" + quote(table.getName()) + ");\n" +
                (baseline ? getBaselineStatement(table) + ";\n" : "") +
                "db." + table.getName() + ".createIndex({ success: -1 })";
    }

    @Override
    public String getInsertStatement(Table table) {

        return "db.getSiblingDB('" + table.getSchema().getName() + "')." + table.getName() + ".insertOne({"
                + quote("installed_rank") + ": ?, "
                + quote("version") + ": ?, "
                + quote("description") + ": ?, "
                + quote("type") + ": ?, "
                + quote("script") + ": ?, "
                + quote("checksum") + ": ?, "
                + quote("installed_by") + ": ?, "
                + quote("installed_on") + ": '" + Timestamp.from(Instant.now()) + "', "
                + quote("execution_time") + ": ?, "
                + quote("success") + ": ?"
                + "})";
    }

    @Override
    public String getUpdateStatement(Table table) {
        return "db.getSiblingDB('" + table.getSchema().getName() + "')." + table.getName() + ".updateOne({"
                + quote("installed_rank") + ": ?"
                + "}, {"
                + "$set: {"
                + quote("description") + ": ?,"
                + quote("type") + ": ?,"
                + quote("checksum") + ": ?"
                + "}"
                + "})";
    }

    @Override
    public String getSelectStatement(Table table) {
        return "db.getSiblingDB('" + table.getSchema().getName() + "')." + table.getName() + ".find({"
                + quote("installed_rank") + ": { $gt: ? }"
                + "}).sort({"
                + quote("installed_rank") + ": 1"
                + "})";
    }

    @Override
    public Pair getDeleteStatement(Table table, boolean version, String filter) {
        String deleteStatement =  "db.getSiblingDB('" + table.getSchema().getName() + "')." + table.getName() + ".deleteMany({ 'success': " + getBooleanFalse() + ", " +
                (version ?
                        "'version': ? " :
                        "'desciption': ? ") +
                "})";

        return Pair.of(deleteStatement, filter);
    }

    @Override
    protected String getBaselineStatement(Table table) {
        return String.format(getInsertStatement(table).replace("?", "%s"),
                             1,
                             quoteIfNotNull(configuration.getBaselineVersion().toString()),
                             quoteIfNotNull(AbbreviationUtils.abbreviateDescription(configuration.getBaselineDescription())),
                             quoteIfNotNull(CoreMigrationType.BASELINE.toString()),
                             quoteIfNotNull(AbbreviationUtils.abbreviateScript(configuration.getBaselineDescription())),
                             null,
                             quoteIfNotNull(getInstalledBy()),
                             0,
                             getBooleanTrue()
                            );
    }

    private String quoteIfNotNull(String value) {
        return value == null ? null : quote(value);
    }

    @Override
    public String getDatabaseHosting() {
        if (configuration.getUrl().contains("mongodb+srv")) {
            return DATABASE_HOSTING_MONGODB_ATLAS;
        }

        return super.getDatabaseHosting();
    }
}