com.mysql.cj.conf.url.ReplicationConnectionUrl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mysql-connector-java
Show all versions of mysql-connector-java
JDBC Type 4 driver for MySQL
/*
* Copyright (c) 2016, 2018, Oracle and/or its affiliates. All rights reserved.
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU General Public License, version 2.0, as published by the
* Free Software Foundation.
*
* This program is also distributed with certain software (including but not
* limited to OpenSSL) that is licensed under separate terms, as designated in a
* particular file or component or in included license documentation. The
* authors of MySQL hereby grant you an additional permission to link the
* program and your derivative works with the separately licensed software that
* they have included with MySQL.
*
* Without limiting anything contained in the foregoing, this file, which is
* part of MySQL Connector/J, is also subject to the Universal FOSS Exception,
* version 1.0, a copy of which can be found at
* http://oss.oracle.com/licenses/universal-foss-exception.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License, version 2.0,
* for more details.
*
* You should have received a copy of the GNU General Public License along with
* this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
package com.mysql.cj.conf.url;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.stream.Collectors;
import com.mysql.cj.conf.ConnectionUrl;
import com.mysql.cj.conf.ConnectionUrlParser;
import com.mysql.cj.conf.HostInfo;
import com.mysql.cj.conf.PropertyKey;
public class ReplicationConnectionUrl extends ConnectionUrl {
private static final String TYPE_MASTER = "MASTER";
private static final String TYPE_SLAVE = "SLAVE";
private List masterHosts = new ArrayList<>();
private List slaveHosts = new ArrayList<>();
/**
* Constructs an instance of {@link ReplicationConnectionUrl}, performing all the required initializations.
*
* @param connStrParser
* a {@link ConnectionUrlParser} instance containing the parsed version of the original connection string
* @param info
* the connection arguments map
*/
public ReplicationConnectionUrl(ConnectionUrlParser connStrParser, Properties info) {
super(connStrParser, info);
this.type = Type.REPLICATION_CONNECTION;
// Split masters and slaves:
LinkedList undefinedHosts = new LinkedList<>();
for (HostInfo hi : this.hosts) {
Map hostProperties = hi.getHostProperties();
if (hostProperties.containsKey(PropertyKey.TYPE.getKeyName())) {
if (TYPE_MASTER.equalsIgnoreCase(hostProperties.get(PropertyKey.TYPE.getKeyName()))) {
this.masterHosts.add(hi);
} else if (TYPE_SLAVE.equalsIgnoreCase(hostProperties.get(PropertyKey.TYPE.getKeyName()))) {
this.slaveHosts.add(hi);
} else {
undefinedHosts.add(hi);
}
} else {
undefinedHosts.add(hi);
}
}
if (!undefinedHosts.isEmpty()) {
if (this.masterHosts.isEmpty()) {
this.masterHosts.add(undefinedHosts.removeFirst());
}
this.slaveHosts.addAll(undefinedHosts);
}
// TODO: Validate the hosts list: there can't be any two hosts with same host:port.
// Although this should be required, it also is incompatible with our current tests which are creating replication connections
// using the same host configurations.
// Set visitedHosts = new HashSet<>();
// for (List hostsLists : Arrays.asList(this.masterHosts, this.slaveHosts)) {
// for (HostInfo hi : hostsLists) {
// if (visitedHosts.contains(hi.getHostPortPair())) {
// throw ExceptionFactory.createException(WrongArgumentException.class,
// Messages.getString("ConnectionString.13", new Object[] { hi.getHostPortPair(), Type.REPLICATION_CONNECTION.getProtocol() }));
// }
// visitedHosts.add(hi.getHostPortPair());
// }
// }
}
/**
* Constructs an instance of a {@link ReplicationConnectionUrl} based on a list of master hosts, a list of slave hosts and a global set of properties
* instead of connection string parsing.
* {@link ConnectionUrl} instances created by this process are not cached.
*
* @param masters
* the master hosts list to use in this connection string
* @param slaves
* the slave hosts list to use in this connection string
* @param properties
* the properties common to all hosts
*/
public ReplicationConnectionUrl(List masters, List slaves, Map properties) {
this.originalConnStr = ConnectionUrl.Type.REPLICATION_CONNECTION.getScheme() + "//**internally_generated**" + System.currentTimeMillis() + "**";
this.type = ConnectionUrl.Type.REPLICATION_CONNECTION;
this.hosts.addAll(masters);
this.hosts.addAll(slaves);
this.masterHosts.addAll(masters);
this.slaveHosts.addAll(slaves);
this.properties.putAll(properties);
injectPerTypeProperties(this.properties);
setupPropertiesTransformer(); // This is needed if new hosts come to be spawned in this connection URL.
}
/**
* Returns an existing master host info with the same host:port part or spawns a new isolated host info based on this connection URL if none was found.
*
* @param hostPortPair
* the host:port part to search for
* @return the existing host info or a new independent one
*/
public HostInfo getMasterHostOrSpawnIsolated(String hostPortPair) {
return super.getHostOrSpawnIsolated(hostPortPair, this.masterHosts);
}
/**
* Returns the list of master hosts.
*
* @return the list of master hosts.
*/
public List getMastersList() {
return Collections.unmodifiableList(this.masterHosts);
}
/**
* Returns a list of this connection URL master hosts in the form of host:port pairs.
*
* @return a list of this connection URL master hosts in the form of host:port pairs
*/
public List getMastersListAsHostPortPairs() {
return this.masterHosts.stream().map(hi -> hi.getHostPortPair()).collect(Collectors.toList());
}
/**
* Returns the list of {@link HostInfo} instances that matches the given collection of host:port pairs in the corresponding hosts list. Isolated host info
* elements are spawned for the missing elements.
*
* @param hostPortPairs
* a list of host:port pairs
* @return a list of {@link HostInfo} instances corresponding to the given host:port pairs
*/
public List getMasterHostsListFromHostPortPairs(Collection hostPortPairs) {
return hostPortPairs.stream().map(this::getMasterHostOrSpawnIsolated).collect(Collectors.toList());
}
/**
* Returns an existing slave host info with the same host:port part or spawns a new isolated host info based on this connection URL if none was found.
*
* @param hostPortPair
* the host:port part to search for
* @return the existing host info or a new independent one
*/
public HostInfo getSlaveHostOrSpawnIsolated(String hostPortPair) {
return super.getHostOrSpawnIsolated(hostPortPair, this.slaveHosts);
}
/**
* Returns the list of slave hosts.
*
* @return the list of slave hosts.
*/
public List getSlavesList() {
return Collections.unmodifiableList(this.slaveHosts);
}
/**
* Returns a list of this connection URL master hosts in the form of host:port pairs.
*
* @return a list of this connection URL master hosts in the form of host:port pairs
*/
public List getSlavesListAsHostPortPairs() {
return this.slaveHosts.stream().map(hi -> hi.getHostPortPair()).collect(Collectors.toList());
}
/**
* Returns the list of {@link HostInfo} instances that matches the given collection of host:port pairs in the corresponding hosts list. Isolated host info
* elements are spawned for the missing elements.
*
* @param hostPortPairs
* a list of host:port pairs
* @return a list of {@link HostInfo} instances corresponding to the given host:port pairs
*/
public List getSlaveHostsListFromHostPortPairs(Collection hostPortPairs) {
return hostPortPairs.stream().map(this::getSlaveHostOrSpawnIsolated).collect(Collectors.toList());
}
}