private void doSave() {
ObjectOutputStream objectStream = getObjectOutputStream();
if (objectStream != null) {
try {
System.out.println("Saving " + selectedChildrenPaths.size() + " Selected Generations...");
for (int i = 0; i < selectedChildrenPaths.size(); i++) {
// Get the userObject at the supplied path
Object selectedPath =
((TreePath) selectedChildrenPaths.elementAt(i)).getLastPathComponent();
DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode) selectedPath;
objectStream.writeObject(selectedNode.getUserObject());
}
objectStream.close();
System.out.println("Save completed successfully.");
} catch (IOException e) {
System.err.println(e);
}
} else {
System.out.println("Save Selected Files has been aborted!");
}
}
// saves the current configuration (breakpoints, window sizes and positions...)
private void saveConfig() {
try {
ObjectOutputStream out = new ObjectOutputStream(new FileOutputStream(idxName + ".config"));
Breakpoints.save(out);
Properties.save(out);
out.close();
} catch (IOException exc) {
consoleFrame.echoInfo("Could not save settings: " + exc);
}
}
/** Stop talking to the server */
public void stop() {
if (socket != null) {
// Close all the streams and socket
if (out != null) {
try {
out.close();
} catch (IOException ioe) {
}
out = null;
}
if (in != null) {
try {
in.close();
} catch (IOException ioe) {
}
in = null;
}
if (socket != null) {
try {
socket.close();
} catch (IOException ioe) {
}
socket = null;
}
} else {
// Already stopped
}
// Make sure the right buttons are enabled
start_button.setEnabled(true);
stop_button.setEnabled(false);
setStatus(STATUS_STOPPED);
}
/** Start talking to the server */
public void start() {
String codehost = getCodeBase().getHost();
if (socket == null) {
try {
// Open the socket to the server
socket = new Socket(codehost, port);
// Create output stream
out = new ObjectOutputStream(socket.getOutputStream());
out.flush();
// Create input stream and start background
// thread to read data from the server
in = new ObjectInputStream(socket.getInputStream());
new Thread(this).start();
} catch (Exception e) {
// Exceptions here are unexpected, but we can't
// really do anything (so just write it to stdout
// in case someone cares and then ignore it)
System.out.println("Exception! " + e.toString());
e.printStackTrace();
setErrorStatus(STATUS_NOCONNECT);
socket = null;
}
} else {
// Already started
}
if (socket != null) {
// Make sure the right buttons are enabled
start_button.setEnabled(false);
stop_button.setEnabled(true);
setStatus(STATUS_ACTIVE);
}
}
private void doSaveAll() {
ObjectOutputStream objectStream = getObjectOutputStream();
if (objectStream != null) {
try {
System.out.println("Saving All " + generations.getLeafCount() + " Generations...");
for (Enumeration e = generations.depthFirstEnumeration(); e.hasMoreElements(); ) {
DefaultMutableTreeNode tmpNode = (DefaultMutableTreeNode) e.nextElement();
if (tmpNode.isLeaf()) objectStream.writeObject(tmpNode.getUserObject());
}
objectStream.close();
System.out.println("Save completed successfully.");
} catch (IOException e) {
System.err.println(e);
}
} else {
System.out.println("Save All Files has been aborted!");
}
}
// create a new profile: read the script and possibly execute the queries,
// save the stats (if 'import' == true, we assume the profile already exists
// and don't run the queries)
public void createWkld(String name, String scriptFile, boolean runQueries) {
System.gc();
Workload wkld = new Workload(name);
if (showCmdsItem.getState()) {
consoleFrame.echoCmd((!runQueries ? "importprof " : "newwkld ") + name + " " + scriptFile);
}
// construct the Workload object from the script;
// first, check if the file exists
try {
FileReader reader = new FileReader(scriptFile);
reader.close();
} catch (FileNotFoundException e) {
System.out.println("couldn't open " + scriptFile);
return;
} catch (IOException e) {
System.out.println("couldn't close " + scriptFile);
}
// now, check if it contains only queries
int scriptId = 0;
try {
scriptId = Libgist.openScript(scriptFile);
} catch (LibgistException e) {
System.out.println("couldn't open (C) " + scriptFile);
return;
}
char[] arg1 = new char[64 * 1024];
StringBuffer arg1Buf = new StringBuffer();
char[] arg2 = new char[64 * 1024];
StringBuffer arg2Buf = new StringBuffer();
// for (;;) {
// int cmd = Libgist.getCommand(scriptId, arg1, arg2);
// if (cmd == Libgist.EOF) break;
// if (cmd != Libgist.FETCH) {
// there should only be queries
// System.out.println("Script file contains non-SELECT command");
// return;
// }
// }
if (runQueries) {
// turn profiling on and execute queries
// Libgist.setProfilingEnabled(true);
Libgist.disableBps(true); // we don't want to stop at breakpoints
// rescan queries
try {
scriptId = Libgist.openScript(scriptFile);
} catch (LibgistException e) {
System.out.println("couldn't open (C) " + scriptFile);
return;
}
int cnt = 1;
// for (;;) {
// int cmd = Libgist.getCommand(scriptId, arg1, arg2);
// if (cmd == Libgist.EOF) break;
// arg1Buf.setLength(0);
// arg1Buf.append(arg1, 0, strlen(arg1));
// arg2Buf.setLength(0);
// arg2Buf.append(arg2, 0, strlen(arg2));
// OpThread.execCmd(LibgistCommand.FETCH, arg1Buf.toString(),
// arg2Buf.toString(), false);
// System.out.print(cnt + " ");
// System.out.println(cnt + ": execute " + arg2Buf.toString() + " "
// + arg1Buf.toString());
// cnt++;
// }
System.out.println();
Libgist.disableBps(false);
// compute optimal clustering and some more statistics
// Libgist.computeMetrics(wkld.filename);
}
// save profile
try {
// we're saving Java and C++ data in separate files (filename and filename.prof)
// the profile object only contains the filename, the queries will be
// read in from the file when the profile is opened (faster that way)
ObjectOutputStream out = new ObjectOutputStream(new FileOutputStream(wkld.filename));
out.writeObject(wkld);
out.close();
System.out.println("copy query file");
Runtime.getRuntime().exec("cp " + scriptFile + " " + wkld.filename + ".queries");
System.out.println("saving tree and profile");
Libgist.saveToFile(wkld.filename + ".idx");
if (runQueries) {
// Libgist.saveProfile(wkld.filename + ".prof");
}
} catch (Exception e) {
System.out.println("Error saving profile: " + e);
return;
}
if (runQueries) {
// turn profiling off (after the metrics were computed and
// the profile saved)
// Libgist.setProfilingEnabled(false);
}
}
