/*
* Copyright 2009 LinkedIn, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package voldemort.utils.app;
import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import joptsimple.OptionSet;
import voldemort.utils.ClusterGenerator;
import voldemort.utils.CmdUtils;
import voldemort.utils.HostNamePair;
public class VoldemortClusterGeneratorApp extends VoldemortApp {
public static void main(String[] args) throws Exception {
new VoldemortClusterGeneratorApp().run(args);
}
@Override
protected String getScriptName() {
return "voldemort-clustergenerator.sh";
}
@Override
public void run(String[] args) throws Exception {
parser.accepts("help", "Prints this help");
parser.accepts("logging",
"Options are \"debug\", \"info\" (default), \"warn\", \"error\", or \"off\"")
.withRequiredArg();
parser.accepts("hostnames", "File containing host names").withRequiredArg();
parser.accepts("partitions", "Number of partitions per cluster node")
.withRequiredArg()
.ofType(Integer.class);
parser.accepts("clustername", "Cluster name; defaults to mycluster").withRequiredArg();
parser.accepts("useinternal", "Use internal host name; defaults to true")
.withRequiredArg()
.ofType(Boolean.class);
OptionSet options = parse(args);
List<File> hostNamesFiles = getRequiredInputFiles(options, "hostnames");
int partitions = getRequiredInt(options, "partitions");
String clusterName = CmdUtils.valueOf(options, "clustername", "mycluster");
boolean useInternal = CmdUtils.valueOf(options, "useinternal", true);
HashMap<Integer, List<String>> zoneToHostNames = new HashMap<Integer, List<String>>();
int zoneId = 0;
// Every file specified is considered as one zone
for(File hostNameFile: hostNamesFiles) {
List<HostNamePair> hostNamePairs = getHostNamesPairsFromFile(hostNameFile);
List<String> hostNames = new ArrayList<String>();
for(HostNamePair hostNamePair: hostNamePairs) {
if(useInternal)
hostNames.add(hostNamePair.getInternalHostName());
else
hostNames.add(hostNamePair.getExternalHostName());
}
zoneToHostNames.put(zoneId++, hostNames);
}
String clusterXml = new ClusterGenerator().createClusterDescriptor(clusterName,
zoneToHostNames,
partitions);
System.out.print(clusterXml);
}
}