public static void main(String[] args) throws Exception {
if (args.length < 1 || args.length > 5) {
System.out.println("Invalid number of arguments.\n" + usage);
return;
}
String confDir = Server.DEFAULT_CONF_DIR;
String outputDir = DEFAULT_OUTPUT_DIR;
if (args.length >= 3 && args[1].equals("--conf")) {
confDir = args[2];
args = ParallelUtility.removeArg(args, 1);
args = ParallelUtility.removeArg(args, 1);
}
if (args.length >= 3 && args[1].equals("--output")) {
outputDir = args[2];
args = ParallelUtility.removeArg(args, 1);
args = ParallelUtility.removeArg(args, 1);
}
Catalog c = Database.getCatalog();
SocketInfo[] workers = ParallelUtility.loadWorkers(confDir);
c.loadSchema(args[0]);
TableStats.computeStatistics();
File catalogFile = new File(args[0]);
for (SocketInfo worker : workers) {
File folder = new File(outputDir + "/" + worker.getHost() + "_" + worker.getPort());
folder.mkdirs();
ParallelUtility.copyFileFolder(
catalogFile, new File(folder.getAbsolutePath() + "/catalog.schema"), true);
}
TransactionId fateTid = new TransactionId();
Iterator<Integer> tableIds = c.tableIdIterator();
while (tableIds.hasNext()) {
int tableid = tableIds.next();
int numTuples = getTotalTuples(tableid);
HeapFile h = (HeapFile) c.getDatabaseFile(tableid);
int eachSplitSize = numTuples / workers.length;
int[] splitSizes = new int[workers.length];
Arrays.fill(splitSizes, eachSplitSize);
for (int i = 0; i < numTuples % workers.length; i++) {
splitSizes[i] += 1;
}
DbFileIterator dfi = h.iterator(fateTid);
dfi.open();
for (int i = 0; i < workers.length; i++) {
ArrayList<Tuple> buffer = new ArrayList<Tuple>();
for (int j = 0; j < splitSizes[i]; j++) {
dfi.hasNext();
buffer.add(dfi.next());
}
Iterator<TDItem> items = h.getTupleDesc().iterator();
ArrayList<Type> types = new ArrayList<Type>();
while (items.hasNext()) types.add(items.next().fieldType);
writeHeapFile(
buffer,
new File(
outputDir
+ "/"
+ workers[i].getHost()
+ "_"
+ workers[i].getPort()
+ "/"
+ c.getTableName(tableid)
+ ".dat"),
BufferPool.getPageSize(),
types.toArray(new Type[] {}));
}
}
}
