ghOSt: Add print_last command line argument to RocksDB

When this command line argument is true, RocksDB only prints the
end-to-end results, rather than the results for each stage.

PiperOrigin-RevId: 489336301

Author: jackhumphries

experiments/rocksdb/latency.cc

@@ -216,14 +216,16 @@ void Print(const std::vector<Request>& requests, absl::Duration runtime,
     PrintPrettyPreface(options);
   }
 
-  HANDLE_STAGE("Ingress Queue Time", requests, runtime, request_generated,
-               request_received, options);
-  HANDLE_STAGE("Repeatable Handle Time", requests, runtime, request_received,
-               request_assigned, options);
-  HANDLE_STAGE("Worker Queue Time", requests, runtime, request_assigned,
-               request_start, options);
-  HANDLE_STAGE("Worker Handle Time", requests, runtime, request_start,
-               request_finished, options);
+  if (!options.print_last) {
+    HANDLE_STAGE("Ingress Queue Time", requests, runtime, request_generated,
+                 request_received, options);
+    HANDLE_STAGE("Repeatable Handle Time", requests, runtime, request_received,
+                 request_assigned, options);
+    HANDLE_STAGE("Worker Queue Time", requests, runtime, request_assigned,
+                 request_start, options);
+    HANDLE_STAGE("Worker Handle Time", requests, runtime, request_start,
+                 request_finished, options);
+  }
   // Total time in system
   HANDLE_STAGE("Total", requests, runtime, request_generated, request_finished,
                options);

experiments/rocksdb/latency.h

@@ -18,6 +18,9 @@ struct PrintOptions {
   // If true, prints the results in human-readable form. Otherwise, prints the
   // results in CSV form.
   bool pretty;
+  // If true, only prints the end-to-end results, rather than the results for
+  // each stage.
+  bool print_last;
   // If true, prints the entire distribution.
   bool distribution;
   // If true, prints the latencies in units of nanoseconds. If false, prints the

experiments/rocksdb/main.cc

@@ -14,6 +14,9 @@
 
 ABSL_FLAG(std::string, print_format, "pretty",
           "Results print format (\"pretty\" or \"csv\", default: \"pretty\")");
+ABSL_FLAG(bool, print_last, false,
+          "If true, only prints the end-to-end results, rather than the "
+          "results for each stage (default: false)");
 ABSL_FLAG(bool, print_distribution, false,
           "Prints every request's results (default: false)");
 ABSL_FLAG(bool, print_ns, false,
@@ -98,6 +101,7 @@ ghost_test::Options GetOptions() {
   CHECK(print_format == "pretty" || print_format == "csv");
   options.print_options.pretty = (print_format == "pretty");
 
+  options.print_options.print_last = absl::GetFlag(FLAGS_print_last);
   options.print_options.distribution = absl::GetFlag(FLAGS_print_distribution);
   options.print_options.ns = absl::GetFlag(FLAGS_print_ns);
   options.print_options.os = &std::cout;