package tachyon.master;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.google.common.io.Closer;
import tachyon.Constants;
/**
* The recompute command class. Used to execute the recomputation.
*/
public class RecomputeCommand implements Runnable {
private static final Logger LOG = LoggerFactory.getLogger(Constants.LOGGER_TYPE);
private final String mCommand;
private final String mFilePath;
/**
* Create a new RecomputeCommand.
*
* @param cmd The command to execute
* @param filePath The path of the output file, which records the output of the recompute process.
*/
public RecomputeCommand(String cmd, String filePath) {
mCommand = cmd;
mFilePath = filePath;
}
@Override
public void run() {
try {
LOG.info("Exec " + mCommand + " output to " + mFilePath);
Process p = java.lang.Runtime.getRuntime().exec(mCommand);
String line;
Closer closer = Closer.create();
try {
BufferedReader bri =
closer.register(new BufferedReader(new InputStreamReader(p.getInputStream())));
BufferedReader bre =
closer.register(new BufferedReader(new InputStreamReader(p.getErrorStream())));
File file = new File(mFilePath);
FileWriter fw = new FileWriter(file.getAbsoluteFile());
BufferedWriter bw = closer.register(new BufferedWriter(fw));
while ((line = bri.readLine()) != null) {
bw.write(line + "\n");
}
while ((line = bre.readLine()) != null) {
bw.write(line + "\n");
}
bw.flush();
} finally {
closer.close();
}
p.waitFor();
LOG.info("Exec " + mCommand + " output to " + mFilePath + " done.");
} catch (IOException e) {
LOG.error(e.getMessage());
} catch (InterruptedException e) {
LOG.error(e.getMessage());
}
}
}