com.ecyshor.cassmig.model.MigrationFile Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cassandra-migration Show documentation

Java schema migration library for Cassandra

There is a newer version: 1.1.6

package com.ecyshor.cassmig.model;

import com.google.common.collect.Lists;
import org.apache.commons.codec.digest.DigestUtils;
import org.joda.time.DateTime;

import java.util.ArrayList;
import java.util.List;

public class MigrationFile extends BaseMigration {

	protected List commands;
	protected String usedKeyspace;
	private boolean hasKeyspaceSet = false;

	public MigrationFile(String schema, int order, String description, List commands, String usedKeyspace, boolean useKeyspace) {
		this.schema = schema;
		this.order = order;
		this.description = description;
		this.commands = commands;
		this.usedKeyspace = usedKeyspace;
		this.hasKeyspaceSet = useKeyspace;
		this.md5Sum = getMd5ForMigrationStatements();
	}

	public MigrationFile(String schema, int order, String description, List commands, String usedKeyspace) {
		this(schema, order, description, commands, usedKeyspace, true);
	}

	public MigrationFile(int order, String description, List migrations, String keyspace, boolean usedKeyspace) {
		this("primary", order, description, migrations, keyspace, usedKeyspace);
	}

	public MigrationFile(int order, String description, List statements, String keyspace) {
		this(order, description, statements, keyspace, true);
	}

	public MigrationFile(int order, String description, String md5sum, ArrayList commands) {
		this.schema = "primary";
		this.order = order;
		this.description = description;
		this.md5Sum = md5sum;
		this.commands = commands;
	}

	public MigrationFile(int order, String desc, ArrayList commands) {
		this.schema = "primary";
		this.order = order;
		this.description = desc;
		this.commands = commands;
	}

	public List getCommands() {
		List finalCommands = Lists.newArrayList(commands);
		if (hasKeyspaceSet) {
			finalCommands.add(0, "USE " + usedKeyspace.trim() + ";");
		}
		return finalCommands;
	}

	public String getUsedKeyspace() {
		return usedKeyspace;
	}

	private String getMd5ForMigrationStatements() {
		StringBuilder builder = new StringBuilder();
		for (String configurationLine : getCommands()) {
			builder.append(configurationLine);
		}
		return DigestUtils.md5Hex(builder.toString());

	}

	public void setSchema(String schema) {
		this.schema = schema;
	}

	public AppliedMigration toAppliedMigration() {
		return new AppliedMigration(this.getSchema(), this.getOrder(), this.getMd5Sum(), DateTime.now());
	}

}