NOT FOR REVIEW - Slow Reports Reference Implementation

content/browser/tracing/tracing_controller_impl_data_sinks.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 #include "content/browser/tracing/tracing_controller_impl.h"
 
 #include "base/bind.h"
 #include "base/files/file_util.h"
 #include "content/public/browser/browser_thread.h"
 #include "third_party/zlib/zlib.h"
 
 namespace content {
 
 namespace {
 
-class FileTraceDataSink : public TracingController::TraceDataSink {
+class StringTraceDataEndpoint : public TracingController::TraceDataEndpoint {
  public:
-  explicit FileTraceDataSink(const base::FilePath& trace_file_path,
-                             const base::Closure& callback)
+  typedef base::Callback<void(base::RefCountedString*)> CompletionCallback;
+
+  explicit StringTraceDataEndpoint(CompletionCallback callback)
+      : completion_callback_(callback) {}
+
+  void ReceiveTraceFinalContents(const std::string& contents) override {
+    std::string tmp = contents;
+    scoped_refptr<base::RefCountedString> str =
+        base::RefCountedString::TakeString(&tmp);
+
+    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+                            base::Bind(completion_callback_, str));
+  }
+
+ private:
+  ~StringTraceDataEndpoint() override {}
+
+  CompletionCallback completion_callback_;
+
+  DISALLOW_COPY_AND_ASSIGN(StringTraceDataEndpoint);
+};
+
+class FileTraceDataEndpoint : public TracingController::TraceDataEndpoint {
+ public:
+  explicit FileTraceDataEndpoint(const base::FilePath& trace_file_path,
+                                 const base::Closure& callback)
       : file_path_(trace_file_path),
         completion_callback_(callback),
         file_(NULL) {}
 
-  void AddTraceChunk(const std::string& chunk) override {
+  void ReceiveTraceChunk(const std::string& chunk) override {
     std::string tmp = chunk;
     scoped_refptr<base::RefCountedString> chunk_ptr =
         base::RefCountedString::TakeString(&tmp);
     BrowserThread::PostTask(
-        BrowserThread::FILE,
-        FROM_HERE,
-        base::Bind(
-            &FileTraceDataSink::AddTraceChunkOnFileThread, this, chunk_ptr));
+        BrowserThread::FILE, FROM_HERE,
+        base::Bind(&FileTraceDataEndpoint::ReceiveTraceChunkOnFileThread, this,
+                   chunk_ptr));
   }
-  void SetSystemTrace(const std::string& data) override {
-    system_trace_ = data;
-  }
-  void Close() override {
+
+  void ReceiveTraceFinalContents(const std::string& contents) override {
     BrowserThread::PostTask(
-        BrowserThread::FILE,
-        FROM_HERE,
-        base::Bind(&FileTraceDataSink::CloseOnFileThread, this));
+        BrowserThread::FILE, FROM_HERE,
+        base::Bind(&FileTraceDataEndpoint::CloseOnFileThread, this));
   }
 
  private:
-  ~FileTraceDataSink() override { DCHECK(file_ == NULL); }
+  ~FileTraceDataEndpoint() override { DCHECK(file_ == NULL); }
 
-  void AddTraceChunkOnFileThread(
+  void ReceiveTraceChunkOnFileThread(
       const scoped_refptr<base::RefCountedString> chunk) {
-    if (file_ != NULL)
-      fputc(',', file_);
-    else if (!OpenFileIfNeededOnFileThread())
+    if (!OpenFileIfNeededOnFileThread())
       return;
-    ignore_result(fwrite(chunk->data().c_str(), strlen(chunk->data().c_str()),
-        1, file_));
+    ignore_result(
+        fwrite(chunk->data().c_str(), chunk->data().size(), 1, file_));
   }
 
   bool OpenFileIfNeededOnFileThread() {
     if (file_ != NULL)
       return true;
     file_ = base::OpenFile(file_path_, "w");
     if (file_ == NULL) {
       LOG(ERROR) << "Failed to open " << file_path_.value();
       return false;
     }
-    const char preamble[] = "{\"traceEvents\": [";
-    ignore_result(fwrite(preamble, strlen(preamble), 1, file_));
     return true;
   }
 
   void CloseOnFileThread() {
     if (OpenFileIfNeededOnFileThread()) {
-      fputc(']', file_);
-      if (!system_trace_.empty()) {
-        const char systemTraceEvents[] = ",\"systemTraceEvents\": ";
-        ignore_result(fwrite(systemTraceEvents, strlen(systemTraceEvents),
-            1, file_));
-        ignore_result(fwrite(system_trace_.c_str(),
-            strlen(system_trace_.c_str()), 1, file_));
-      }
-      fputc('}', file_);
       base::CloseFile(file_);
       file_ = NULL;
     }
     BrowserThread::PostTask(
-        BrowserThread::UI,
-        FROM_HERE,
-        base::Bind(&FileTraceDataSink::FinalizeOnUIThread, this));
+        BrowserThread::UI, FROM_HERE,
+        base::Bind(&FileTraceDataEndpoint::FinalizeOnUIThread, this));
   }
 
   void FinalizeOnUIThread() { completion_callback_.Run(); }
 
   base::FilePath file_path_;
   base::Closure completion_callback_;
   FILE* file_;
-  std::string system_trace_;
 
-  DISALLOW_COPY_AND_ASSIGN(FileTraceDataSink);
+  DISALLOW_COPY_AND_ASSIGN(FileTraceDataEndpoint);
 };
 
 class StringTraceDataSink : public TracingController::TraceDataSink {
  public:
-  typedef base::Callback<void(base::RefCountedString*)> CompletionCallback;
+  explicit StringTraceDataSink(
+      scoped_refptr<TracingController::TraceDataEndpoint> endpoint)
+      : endpoint_(endpoint) {}
 
-  explicit StringTraceDataSink(CompletionCallback callback)
-      : completion_callback_(callback) {}
+  void AddTraceChunk(const std::string& chunk) override {
+    std::string trace_string;
+    if (trace_.empty())
+      trace_string = "{\"traceEvents\":[";
+    else
+      trace_string = ",";
+    trace_string += chunk;
 
-  // TracingController::TraceDataSink implementation
-  void AddTraceChunk(const std::string& chunk) override {
-    if (!trace_.empty())
-      trace_ += ",";
+    AddTraceChunkAndPassToEndpoint(trace_string);
+  }
+
+  void AddTraceChunkAndPassToEndpoint(const std::string& chunk) {
     trace_ += chunk;
+
+    endpoint_->ReceiveTraceChunk(chunk);
   }
+
   void SetSystemTrace(const std::string& data) override {
     system_trace_ = data;
   }
+
   void Close() override {
-    std::string result = "{\"traceEvents\":[" + trace_ + "]";
+    AddTraceChunkAndPassToEndpoint("]");
     if (!system_trace_.empty())
-      result += ",\"systemTraceEvents\": " + system_trace_;
-    result += "}";
+      AddTraceChunkAndPassToEndpoint(",\"systemTraceEvents\": " +
+                                     system_trace_);
+    AddTraceChunkAndPassToEndpoint("}");
 
-    scoped_refptr<base::RefCountedString> str =
-        base::RefCountedString::TakeString(&result);
-    completion_callback_.Run(str.get());
+    endpoint_->ReceiveTraceFinalContents(trace_);
   }
 
  private:
   ~StringTraceDataSink() override {}
 
+  scoped_refptr<TracingController::TraceDataEndpoint> endpoint_;
   std::string trace_;
   std::string system_trace_;
-  CompletionCallback completion_callback_;
 
   DISALLOW_COPY_AND_ASSIGN(StringTraceDataSink);
 };
 
 class CompressedStringTraceDataSink : public TracingController::TraceDataSink {
  public:
+  const int kChunkSize = 0x4000;
+
   explicit CompressedStringTraceDataSink(
       scoped_refptr<TracingController::TraceDataEndpoint> endpoint)
       : endpoint_(endpoint), already_tried_open_(false) {}
 
   void AddTraceChunk(const std::string& chunk) override {
     std::string tmp = chunk;
     scoped_refptr<base::RefCountedString> chunk_ptr =
         base::RefCountedString::TakeString(&tmp);
     BrowserThread::PostTask(
         BrowserThread::FILE, FROM_HERE,
@@ skipping 48 matching lines @@
     trace += chunk_ptr->data();
     AddTraceChunkAndCompressOnFileThread(trace, false);
   }
 
   void AddTraceChunkAndCompressOnFileThread(const std::string& chunk,
                                             bool finished) {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
     if (!OpenZStreamOnFileThread())
       return;
 
-    const int kChunkSize = 0x4000;
-
     char buffer[kChunkSize];
     int err;
     stream_->avail_in = chunk.size();
     stream_->next_in = (unsigned char*)chunk.data();
     do {
       stream_->avail_out = kChunkSize;
       stream_->next_out = (unsigned char*)buffer;
       err = deflate(stream_.get(), finished ? Z_FINISH : Z_NO_FLUSH);
       if (err != Z_OK && (err != Z_STREAM_END && finished)) {
         stream_.reset();
@@ skipping 37 matching lines @@
   std::string system_trace_;
 
   DISALLOW_COPY_AND_ASSIGN(CompressedStringTraceDataSink);
 };
 
 }  // namespace
 
 scoped_refptr<TracingController::TraceDataSink>
 TracingController::CreateStringSink(
     const base::Callback<void(base::RefCountedString*)>& callback) {
-  return new StringTraceDataSink(callback);
+  return new StringTraceDataSink(new StringTraceDataEndpoint(callback));
 }
 
 scoped_refptr<TracingController::TraceDataSink>
 TracingController::CreateCompressedStringSink(
     scoped_refptr<TracingController::TraceDataEndpoint> endpoint) {
   return new CompressedStringTraceDataSink(endpoint);
 }
 
 scoped_refptr<TracingController::TraceDataSink>
 TracingController::CreateFileSink(const base::FilePath& file_path,
                                   const base::Closure& callback) {
-  return new FileTraceDataSink(file_path, callback);
+  return new StringTraceDataSink(CreateFileEndpoint(file_path, callback));
+}
+
+scoped_refptr<TracingController::TraceDataEndpoint>
+TracingController::CreateFileEndpoint(const base::FilePath& file_path,
+                                      const base::Closure& callback) {
+  return new FileTraceDataEndpoint(file_path, callback);
 }
 
 }  // namespace content