net.ozwolf.mongo.migrations.internal.domain.MigrationCommand Maven / Gradle / Ivy

Go to download
package net.ozwolf.mongo.migrations.internal.domain;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.mongodb.client.MongoDatabase;
import net.ozwolf.mongo.migrations.internal.util.StrictJsonUtils;
import org.apache.commons.lang3.StringUtils;
import org.bson.Document;

import java.util.Map;
import java.util.Optional;
import java.util.stream.StreamSupport;

@JsonDeserialize
@JsonIgnoreProperties(ignoreUnknown = true)
public class MigrationCommand {
    private final String version;
    private final String description;
    private final String author;
    private final BasicDBObject command;

    @JsonCreator
    public MigrationCommand(@JsonProperty("version") String version,
                            @JsonProperty("description") String description,
                            @JsonProperty("author") String author,
                            @JsonProperty("command") Map command) {
        if (StringUtils.trimToNull(version) == null || StringUtils.trimToNull(description) == null || command == null)
            throw new IllegalStateException("A migration command requires at least a version, description and a command!");

        this.version = version;
        this.description = description;
        this.author = Optional.ofNullable(author).orElse(Migration.DEFAULT_AUTHOR);
        Map interpolated = StrictJsonUtils.interpolate(command);
        this.command = new BasicDBObject(interpolated);
    }

    public final String getVersion() {
        return version;
    }

    public final String getDescription() {
        return description;
    }

    public String getAuthor() {
        return author;
    }

    public DBObject getCommand() {
        return command;
    }

    public Document migrate(MongoDatabase database) {
        ensureMapReduceCollection(database);
        Document result = database.runCommand(command);
        if (result.get("$clusterTime") != null)
            result.append("clusterTime", result.get("$clusterTime")).remove("$clusterTime");
        return result;
    }

    private void ensureMapReduceCollection(MongoDatabase database) {
        String collection = command.getString("mapReduce", null);
        if (collection == null) return;

        boolean exists = StreamSupport.stream(database.listCollectionNames().spliterator(), false)
                .anyMatch(c -> c.equalsIgnoreCase(collection));

        if (!exists) database.createCollection(collection);
    }
}