in src/main/java/org/apache/accumulo/testing/scalability/Run.java [48:100]
public static void main(String[] args) throws Exception {
final String sitePath = "/tmp/scale-site.conf";
final String testPath = "/tmp/scale-test.conf";
Opts opts = new Opts();
opts.parseArgs(Run.class.getName(), args);
Configuration conf = new Configuration();
FileSystem fs;
fs = FileSystem.get(conf);
fs.copyToLocalFile(new Path("/accumulo-scale/conf/site.conf"), new Path(sitePath));
fs.copyToLocalFile(new Path(String.format("/accumulo-scale/conf/%s.conf", opts.testId)),
new Path(testPath));
// load configuration file properties
Properties scaleProps = new Properties();
Properties testProps = new Properties();
try {
FileInputStream fis = new FileInputStream(sitePath);
try {
scaleProps.load(fis);
} finally {
fis.close();
}
fis = new FileInputStream(testPath);
try {
testProps.load(fis);
} finally {
fis.close();
}
} catch (Exception e) {
log.error("Error loading config file.", e);
}
ScaleTest test = (ScaleTest) Class
.forName(String.format("org.apache.accumulo.test.scalability.%s", opts.testId))
.getDeclaredConstructor().newInstance();
test.init(scaleProps, testProps, opts.numTabletServers);
if (opts.action.equalsIgnoreCase("setup")) {
test.setup();
} else if (opts.action.equalsIgnoreCase("client")) {
InetAddress addr = InetAddress.getLocalHost();
String host = addr.getHostName();
fs.createNewFile(new Path("/accumulo-scale/clients/" + host));
test.client();
fs.copyFromLocalFile(new Path("/tmp/scale.out"), new Path("/accumulo-scale/results/" + host));
} else if (opts.action.equalsIgnoreCase("teardown")) {
test.teardown();
}
}