import java.io.IOException;
import java.lang.ProcessBuilder.Redirect;
+import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.SimpleFileVisitor;
+import java.nio.file.attribute.BasicFileAttributes;
+import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
});
}
+ /**
+ * Send a separate enable-event command.
+ *
+ * @param sessionName
+ * Name of the session in which to enable events. Use null for
+ * current session.
+ * @param domain
+ * The tracing domain
+ * @param enabledEvents
+ * The list of events to enable. Should not be null or empty
+ * @return If the command executed successfully (return code = 0).
+ */
+ public static boolean enableEvents(String sessionName, Domain domain, String... enabledEvents) {
+ if (enabledEvents == null || enabledEvents.length == 0) {
+ throw new IllegalArgumentException();
+ }
+ List<String> command = new ArrayList<String>();
+ command.add("lttng");
+ command.add("enable-event");
+ command.add(domain.flag());
+ command.add(Arrays.stream(enabledEvents).collect(Collectors.joining(",")));
+ if (sessionName != null) {
+ command.add("-s");
+ command.add(sessionName);
+ }
+ return executeCommand(command.toArray(new String[0]));
+ }
+
+ /**
+ * Send a disable-event command. Used to disable events that were previously
+ * enabled.
+ *
+ * @param sessionName
+ * Name of the session in which to disable events. Use null for
+ * current session.
+ * @param domain
+ * The tracing domain
+ * @param disabledEvents
+ * The list of disabled events. Should not be null or empty
+ * @return If the command executed successfully (return code = 0).
+ */
+ public static boolean disableEvents(String sessionName, Domain domain, String... disabledEvents) {
+ if (disabledEvents == null || disabledEvents.length == 0) {
+ throw new IllegalArgumentException();
+ }
+ List<String> command = new ArrayList<String>();
+ command.add("lttng");
+ command.add("disable-event");
+ command.add(domain.flag());
+ command.add(Arrays.stream(disabledEvents).collect(Collectors.joining(",")));
+ if (sessionName != null) {
+ command.add("-s");
+ command.add(sessionName);
+ }
+ return executeCommand(command.toArray(new String[0]));
+ }
+
/**
* Stop the current tracing session
*
+ * @param sessionName
+ * The name of the session to stop. Use null for the current
+ * session.
* @return If the command executed successfully (return code = 0).
*/
- public static boolean stopSession() {
- return executeCommand(new String[] { "lttng", "stop" });
+ public static boolean stopSession(String sessionName) {
+ List<String> command = new ArrayList<String>();
+ command.add("lttng");
+ command.add("stop");
+ if (sessionName != null) {
+ command.add(sessionName);
+ }
+ return executeCommand(command.toArray(new String[0]));
}
- public static List<String> viewSession() {
- return getOutputFromCommand(new String[] { "lttng", "view" });
+ /**
+ * Issue a "lttng view" command on the provided session, and returns its
+ * output. This effectively returns the current content of the trace in text
+ * form.
+ *
+ * @param sessionName
+ * The name of the session to print. Use null for the current
+ * session.
+ * @return The output of Babeltrace on the session's current trace
+ */
+ public static List<String> viewSession(String sessionName) {
+ List<String> command = new ArrayList<String>();
+ command.add("lttng");
+ command.add("view");
+ if (sessionName != null) {
+ command.add(sessionName);
+ }
+ return getOutputFromCommand(command.toArray(new String[0]));
}
/**
* Destroy the current tracing session
*
+ * @param sessionName
+ * The name of the session to destroy. Use null for the current
+ * session.
* @return If the command executed successfully (return code = 0).
*/
- public static boolean destroySession() {
- return executeCommand(new String[] { "lttng", "destroy" });
+ public static boolean destroySession(String sessionName) {
+ List<String> command = new ArrayList<String>();
+ command.add("lttng");
+ command.add("destroy");
+ if (sessionName != null) {
+ command.add(sessionName);
+ }
+ return executeCommand(command.toArray(new String[0]));
+ }
+
+ /**
+ * Try destroying the given tracing session, fail silently if there is no
+ * session.
+ *
+ * @param sessionName
+ * The name of the session to destroy. Use null for the current
+ * session.
+ */
+ public static void tryDestroySession(String sessionName) {
+ getOutputFromCommand(false, new String[] { "lttng", "destroy" });
+ }
+
+ /**
+ * Outside of the scope of lttng-tools, but this utility method can be used
+ * to delete all traces currently under ~/lttng-traces/. This can be used by
+ * tests to cleanup a trace they have created.
+ *
+ * @return True if the command completes successfully, false if there was an
+ * error.
+ */
+ public static boolean deleteAllTracee() {
+ String tracesDir = new String(System.getProperty("user.home") + "/lttng-traces/");
+ return deleteDirectory(Paths.get(tracesDir));
+ }
+
+ private static boolean deleteDirectory(Path directory) {
+ try {
+ Files.walkFileTree(directory, new SimpleFileVisitor<Path>() {
+ @Override
+ public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
+ Files.delete(file);
+ return FileVisitResult.CONTINUE;
+ }
+
+ @Override
+ public FileVisitResult postVisitDirectory(Path dir, IOException exc) throws IOException {
+ Files.delete(dir);
+ return FileVisitResult.CONTINUE;
+ }
+ });
+ } catch (IOException e) {
+ /* At least we tried... */
+ return false;
+ }
+ return true;
}
// ------------------------------------------------------------------------
}
private static List<String> getOutputFromCommand(String[] command) {
+ return getOutputFromCommand(true, command);
+ }
+
+ private static List<String> getOutputFromCommand(boolean print, String[] command) {
try {
Path tempFile = Files.createTempFile("test-output", null);
List<String> lines = Files.readAllLines(tempFile);
Files.delete(tempFile);
- /* Also print the output to the console */
- lines.stream().forEach(s -> System.out.println(s));
+ if (print) {
+ /* Also print the output to the console */
+ lines.stream().forEach(s -> System.out.println(s));
+ }
return lines;
projects / lttng-ust-java-tests.git / blobdiff