DBDatabaseClusterSettingsBuilder.java

/*
 * Copyright 2019 Gregory Graham.
 *
 * Commercial licenses are available, please contact info@gregs.co.nz for details.
 * 
 * This work is licensed under the Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License. 
 * To view a copy of this license, visit http://creativecommons.org/licenses/by-nc-sa/4.0/ 
 * or send a letter to Creative Commons, PO Box 1866, Mountain View, CA 94042, USA.
 * 
 * You are free to:
 *     Share - copy and redistribute the material in any medium or format
 *     Adapt - remix, transform, and build upon the material
 * 
 *     The licensor cannot revoke these freedoms as long as you follow the license terms.               
 *     Under the following terms:
 *                 
 *         Attribution - 
 *             You must give appropriate credit, provide a link to the license, and indicate if changes were made. 
 *             You may do so in any reasonable manner, but not in any way that suggests the licensor endorses you or your use.
 *         NonCommercial - 
 *             You may not use the material for commercial purposes.
 *         ShareAlike - 
 *             If you remix, transform, or build upon the material, 
 *             you must distribute your contributions under the same license as the original.
 *         No additional restrictions - 
 *             You may not apply legal terms or technological measures that legally restrict others from doing anything the 
 *             license permits.
 * 
 * Check the Creative Commons website for any details, legalese, and updates.
 */
package nz.co.gregs.dbvolution.databases.settingsbuilders;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import nz.co.gregs.dbvolution.databases.DBDatabaseCluster;
import nz.co.gregs.dbvolution.databases.DatabaseConnectionSettings;
import nz.co.gregs.dbvolution.databases.definitions.ClusterDatabaseDefinition;
import nz.co.gregs.dbvolution.databases.definitions.DBDefinition;

/**
 *
 * @author gregorygraham
 */
public class DBDatabaseClusterSettingsBuilder extends AbstractSettingsBuilder<DBDatabaseClusterSettingsBuilder, DBDatabaseCluster>
		implements ClusterCapableSettingsBuilder<DBDatabaseClusterSettingsBuilder, DBDatabaseCluster> 
{

	private final static HashMap<String, String> DEFAULT_EXTRAS_MAP = new HashMap<>();
	private static final long serialVersionUID = 1L;
	private boolean useAutoRebuild = false;
	private boolean useAutoReconnect = false;
	private boolean useAutoStart = false;
	private boolean useAutoConnect = false;
	private DBDefinition defn;

	@Override
	public Map<String, String> getDefaultConfigurationExtras() {
		return DEFAULT_EXTRAS_MAP;
	}

	@Override
	public DatabaseConnectionSettings generateSettingsInternal(String jdbcURL, DatabaseConnectionSettings settings) {
		String noPrefix = jdbcURL.replaceAll("^" + getJDBCURLPreamble(), "");
		if (jdbcURL.matches(";")) {
			String extrasString = jdbcURL.split("\\?", 2)[1];
			settings.setExtras(DatabaseConnectionSettings.decodeExtras(extrasString, "", "=", "&", ""));
		}
		settings.setPort(noPrefix
				.split("/", 2)[0]
				.replaceAll("^[^:]*:+", ""));
		settings.setHost(noPrefix
				.split("/", 2)[0]
				.split(":")[0]);
		settings.setInstance(settings.getExtras().get("instance"));
		settings.setSchema("");
		return settings;
	}

	@Override
	protected String getJDBCURLPreamble(DatabaseConnectionSettings settings) {
		return getJDBCURLPreamble();
	}

	protected String getJDBCURLPreamble() {
		return "jdbc:dbvolution-cluster://";
	}

	@Override
	public Class<DBDatabaseCluster> generatesURLForDatabase() {
		return DBDatabaseCluster.class;
	}

	@Override
	public DatabaseConnectionSettings setDefaultsInternal(DatabaseConnectionSettings settings) {
		return settings;
	}

	@Override
	public Integer getDefaultPort() {
		return 1; //cluster doesn't use a port
	}

	@Override
	public String encodeHost(DatabaseConnectionSettings settings) {
		return settings.getHost() + ":"
				+ settings.getPort() + "/"
				+ settings.getDatabaseName()
				+ encodeExtras(settings, "?", "=", "&", "");
	}

	@Override
	public DBDatabaseCluster getDBDatabase() throws Exception {
		return new DBDatabaseCluster(this);
	}

	public DBDatabaseClusterSettingsBuilder setConfiguration(DBDatabaseCluster.Configuration  config) {
		this.useAutoRebuild = config.isUseAutoRebuild();
		this.useAutoConnect = config.isUseAutoConnect();
		this.useAutoReconnect = config.isUseAutoReconnect();
		this.useAutoStart = config.isUseAutoStart();
		return this;
	}

	public DBDatabaseClusterSettingsBuilder setAutoRebuild(boolean useAutoRebuild) {
		this.useAutoRebuild = useAutoRebuild;
		return this;
	}

	public DBDatabaseClusterSettingsBuilder setAutoReconnect(boolean useAutoReconnect) {
		this.useAutoReconnect = useAutoReconnect;
		return this;
	}

	public DBDatabaseClusterSettingsBuilder setAutoStart(boolean useAutoStart) {
		this.useAutoStart = useAutoStart;
		return this;
	}

	public DBDatabaseClusterSettingsBuilder setAutoConnect(boolean useAutoConnect) {
		this.useAutoConnect = useAutoConnect;
		return this;
	}

	public boolean getAutoRebuild() {
		return this.useAutoRebuild;
	}

	public boolean getAutoReconnect() {
		return this.useAutoReconnect;
	}

	public DBDatabaseCluster.Configuration getConfiguration() {
		return new DBDatabaseCluster.Configuration(this.useAutoRebuild, useAutoReconnect, useAutoStart, useAutoConnect);
	}

	@Override
	public DBDefinition getDefinition() {
		return this.defn;
	}

	@Override
	public DBDefinition getDefaultDefinition() {
		return new ClusterDatabaseDefinition();
	}

	@Override
	public DBDatabaseClusterSettingsBuilder setDefinition(DBDefinition defn) {
		this.defn = defn;
		return this;
	}

	@Override
	public DBDatabaseClusterSettingsBuilder setClusterHosts(List<DatabaseConnectionSettings> hosts) {
		getStoredSettings().setClusterHosts(hosts);
		return this;
	}

	@Override
	public List<DatabaseConnectionSettings> getClusterHosts() {
		return getStoredSettings().getClusterHosts();
	}
}