Add pprof support to skydb

sky/tools/debugger/prompt/prompt.cc

 // Copyright 2014 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 <algorithm>
 
 #include "base/bind.h"
-#include "base/files/file_path.h"
+#include "base/debug/profiler.h"
 #include "base/memory/weak_ptr.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "mojo/application/application_runner_chromium.h"
 #include "mojo/public/c/system/main.h"
 #include "mojo/public/cpp/application/application_delegate.h"
 #include "mojo/public/cpp/application/application_impl.h"
 #include "net/base/net_errors.h"
 #include "net/server/http_server.h"
 #include "net/server/http_server_request_info.h"
@@ skipping 65 matching lines @@
     if (info.path == "/trace")
       ToggleTracing(connection_id);
     else if (info.path == "/reload")
       Load(connection_id, url_);
     else if (info.path == "/inspect")
       Inspect(connection_id);
     else if (info.path == "/quit")
       Quit(connection_id);
     else if (info.path == "/load")
       Load(connection_id, info.data);
+    else if (info.path == "/start_profiling")
+      StartProfiling(connection_id);
+    else if (info.path == "/stop_profiling")
+      StopProfiling(connection_id);
     else {
       Help(info.path, connection_id);
     }
   }
 
   void OnWebSocketRequest(
       int connection_id, const net::HttpServerRequestInfo& info) override {
     web_server_->Send500(connection_id, "http only");
   }
 
@@ skipping 61 matching lines @@
     tracing_->Start(pipe.producer_handle.Pass(), mojo::String("*"));
     trace_collector_.reset(new TraceCollector(pipe.consumer_handle.Pass()));
     Respond(connection_id, "Starting trace (type 'trace' to stop tracing)\n");
   }
 
   void OnTraceAvailable(int connection_id, std::string trace) {
     trace_collector_.reset();
     Respond(connection_id, trace);
   }
 
+  void StartProfiling(int connection_id) {
+    base::debug::StartProfiling("sky_viewer.pprof");
+    Respond(connection_id, "Starting profiling (type 'stop_profiling' to stop");
+  }
+
+  void StopProfiling(int connection_id) {
+    base::debug::StopProfiling();
+    Respond(connection_id, "Stopped profiling");
+  }
+
   bool is_tracing_;
   DebuggerPtr debugger_;
   tracing::TraceCoordinatorPtr tracing_;
   std::string url_;
   base::WeakPtrFactory<Prompt> weak_ptr_factory_;
   scoped_ptr<net::HttpServer> web_server_;
   uint32_t command_port_;
 
   scoped_ptr<TraceCollector> trace_collector_;
 
   DISALLOW_COPY_AND_ASSIGN(Prompt);
 };
 
 }  // namespace debugger
 }  // namespace sky
 
 MojoResult MojoMain(MojoHandle shell_handle) {
   mojo::ApplicationRunnerChromium runner(new sky::debugger::Prompt);
   runner.set_message_loop_type(base::MessageLoop::TYPE_IO);
   return runner.Run(shell_handle);
 }