final String oldClusterId = getStringData(curator.get(), ZkPath.CONFIG_ENSEMBLES.getPath());
if (oldClusterId != null) {
String profileId = "fabric-ensemble-" + oldClusterId;
String pid = "io.fabric8.zookeeper.server-" + oldClusterId;
Profile ensProfile = profileRegistry.get().getRequiredProfile(versionId, profileId);
Map<String, String> p = ensProfile.getConfiguration(pid);
if (p == null) {
throw new EnsembleModificationFailed("Failed to find old cluster configuration for ID " + oldClusterId, EnsembleModificationFailed.Reason.ILLEGAL_STATE);
}
for (Object n : p.keySet()) {
String node = (String) n;
if (node.startsWith("server.")) {
Map<String, String> zkconfig = ensProfile.getConfiguration("io.fabric8.zookeeper.server-" + oldClusterId);
String data = getSubstitutedData(curator.get(), zkconfig.get(node));
addUsedPorts(usedPorts, data);
}
}
Profile defaultProfile = profileRegistry.get().getRequiredProfile(versionId, "default");
Map<String, String> zkConfig = defaultProfile.getConfiguration(Constants.ZOOKEEPER_CLIENT_PID);
if (zkConfig == null) {
throw new FabricException("Failed to find old zookeeper configuration in default profile");
}
String zkUrl = getSubstitutedData(curator.get(), zkConfig.get("zookeeper.url"));
for (String data : zkUrl.split(",")) {
addUsedPorts(usedPorts, data);
}
}
String newClusterId;
if (oldClusterId == null) {
newClusterId = "0000";
} else {
newClusterId = new DecimalFormat("0000").format(Integer.parseInt(oldClusterId) + 1);
}
// Ensemble properties
Properties ensembleProperties = new Properties();
String ensemblePropertiesName = "io.fabric8.zookeeper.server-" + newClusterId + ".properties";
ensembleProperties.put("tickTime", String.valueOf(options.getZooKeeperServerTickTime()));
ensembleProperties.put("initLimit", String.valueOf(options.getZooKeeperServerInitLimit()));
ensembleProperties.put("syncLimit", String.valueOf(options.getZooKeeperServerSyncLimit()));
ensembleProperties.put("dataDir", options.getZooKeeperServerDataDir() + File.separator + newClusterId);
// create new ensemble
String ensembleProfileId = "fabric-ensemble-" + newClusterId;
IllegalStateAssertion.assertFalse(profileRegistry.get().hasProfile(versionId, ensembleProfileId), "Profile already exists: " + versionId + "/" + ensembleProfileId);
ProfileBuilder ensembleProfileBuilder = ProfileBuilder.Factory.create(versionId, ensembleProfileId);
ensembleProfileBuilder.addAttribute(Profile.ABSTRACT, "true").addAttribute(Profile.HIDDEN, "true");
int index = 1;
String connectionUrl = "";
String realConnectionUrl = "";
String containerList = "";
List<Profile> memberProfiles = new ArrayList<>();
for (String container : containers) {
String ip = getSubstitutedPath(curator.get(), ZkPath.CONTAINER_IP.getPath(container));
String minimumPort = String.valueOf(Ports.MIN_PORT_NUMBER);
String maximumPort = String.valueOf(Ports.MAX_PORT_NUMBER);
String bindAddress = "0.0.0.0";
if (exists(curator.get(), ZkPath.CONTAINER_PORT_MIN.getPath(container)) != null) {
minimumPort = getSubstitutedPath(curator.get(), ZkPath.CONTAINER_PORT_MIN.getPath(container));
}
if (exists(curator.get(), ZkPath.CONTAINER_PORT_MAX.getPath(container)) != null) {
maximumPort = getSubstitutedPath(curator.get(), ZkPath.CONTAINER_PORT_MAX.getPath(container));
}
if (exists(curator.get(), ZkPath.CONTAINER_BINDADDRESS.getPath(container)) != null) {
bindAddress = getSubstitutedPath(curator.get(), ZkPath.CONTAINER_BINDADDRESS.getPath(container));
}
// Ensemble member properties
Properties memberProperties = new Properties();
String memberPropertiesName = "io.fabric8.zookeeper.server-" + newClusterId + ".properties";
String port1 = publicPort(container, Integer.toString(findPort(usedPorts, ip, mapPortToRange(Ports.DEFAULT_ZOOKEEPER_SERVER_PORT, minimumPort, maximumPort))));
if (containers.size() > 1) {
String port2 = publicPort(container, Integer.toString(findPort(usedPorts, ip, mapPortToRange(Ports.DEFAULT_ZOOKEEPER_PEER_PORT, minimumPort, maximumPort))));
String port3 = publicPort(container, Integer.toString(findPort(usedPorts, ip, mapPortToRange(Ports.DEFAULT_ZOOKEEPER_ELECTION_PORT, minimumPort, maximumPort))));
ensembleProperties.put("server." + Integer.toString(index), "${zk:" + container + "/ip}:" + port2 + ":" + port3);
memberProperties.put("server.id", Integer.toString(index));
}
memberProperties.put("clientPort", port1);
memberProperties.put("clientPortAddress", bindAddress);
// Create ensemble member profile
String memberProfileId = "fabric-ensemble-" + newClusterId + "-" + index;
IllegalStateAssertion.assertFalse(profileRegistry.get().hasProfile(versionId, memberProfileId), "Profile already exists: " + versionId + "/" + memberProfileId);
ProfileBuilder memberProfileBuilder = ProfileBuilder.Factory.create(versionId, memberProfileId);
memberProfileBuilder.addAttribute(Profile.HIDDEN, "true").addAttribute(Profile.PARENTS, ensembleProfileId);
memberProfileBuilder.addFileConfiguration(memberPropertiesName, DataStoreUtils.toBytes(memberProperties));
memberProfiles.add(memberProfileBuilder.getProfile());
if (connectionUrl.length() > 0) {
connectionUrl += ",";
realConnectionUrl += ",";
}
connectionUrl += "${zk:" + container + "/ip}:" + port1;
realConnectionUrl += ip + ":" + port1;
if (containerList.length() > 0) {
containerList += ",";
}
containerList += container;
index++;
}
LockHandle writeLock = profileRegistry.get().aquireWriteLock();
try {
// Create the ensemble profile
ensembleProfileBuilder.addFileConfiguration(ensemblePropertiesName, DataStoreUtils.toBytes(ensembleProperties));
Profile ensembleProfile = ensembleProfileBuilder.getProfile();
LOGGER.info("Creating parent ensemble profile: {}", ensembleProfile);
profileRegistry.get().createProfile(ensembleProfile);
// Create the member profiles
for (Profile memberProfile : memberProfiles) {
LOGGER.info("Creating member ensemble profile: {}", memberProfile);
profileRegistry.get().createProfile(memberProfile);
}
} finally {
writeLock.unlock();
}
index = 1;
for (String container : containers) {
// add this container to the ensemble
List<String> profiles = new LinkedList<String>(dataStore.get().getContainerProfiles(container));
profiles.add("fabric-ensemble-" + newClusterId + "-" + Integer.toString(index));
LOGGER.info("Assigning member ensemble profile with id: {} to {}.", ensembleProfileId + "-" + index, container);
dataStore.get().setContainerProfiles(container, profiles);
index++;
}
Profile defaultProfile = profileRegistry.get().getRequiredProfile(versionId, "default");
Map<String, String> zkConfig = defaultProfile.getConfiguration(Constants.ZOOKEEPER_CLIENT_PID);
if (oldClusterId != null) {
Properties properties = DataStoreUtils.toProperties(zkConfig);
properties.put("zookeeper.url", getSubstitutedData(curator.get(), realConnectionUrl));
properties.put("zookeeper.password", options.getZookeeperPassword());
CuratorFramework dst = CuratorFrameworkFactory.builder().connectString(realConnectionUrl).retryPolicy(new RetryOneTime(500))