content/browser/devtools/protocol/tracing_handler.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_BROWSER_DEVTOOLS_PROTOCOL_TRACING_HANDLER_H_
   6 #define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_TRACING_HANDLER_H_
   7
   8 #include <set>
   9 #include <string>
  10
  11 #include "base/memory/weak_ptr.h"
  12 #include "base/trace_event/trace_event.h"
  13 #include "content/browser/devtools/protocol/devtools_protocol_dispatcher.h"
  14 #include "content/public/browser/tracing_controller.h"
  15
  16 namespace base {
  17 class RefCountedString;
  18 class Timer;
  19 }
  20
  21 namespace content {
  22 namespace devtools {
  23
  24 class DevToolsIOContext;
  25
  26 namespace tracing {
  27
  28 class TracingHandler {
  29  public:
  30   typedef DevToolsProtocolClient::Response Response;
  31
  32   enum Target { Browser, Renderer };
  33   TracingHandler(Target target, DevToolsIOContext* io_context);
  34   virtual ~TracingHandler();
  35
  36   void SetClient(scoped_ptr<Client> client);
  37   void Detached();
  38
  39   void OnTraceDataCollected(const std::string& trace_fragment);
  40   void OnTraceComplete();
  41   void OnTraceToStreamComplete(const std::string& stream_handle);
  42
  43   // Protocol methods.
  44   Response Start(DevToolsCommandId command_id,
  45                  const std::string* categories,
  46                  const std::string* options,
  47                  const double* buffer_usage_reporting_interval,
  48                  const std::string* transfer_mode);
  49   Response End(DevToolsCommandId command_id);
  50   Response GetCategories(DevToolsCommandId command);
  51   Response RequestMemoryDump(DevToolsCommandId command_id);
  52
  53   bool did_initiate_recording() { return did_initiate_recording_; }
  54
  55  private:
  56   void OnRecordingEnabled(DevToolsCommandId command_id);
  57   void OnBufferUsage(float percent_full, size_t approximate_event_count);
  58   void OnCategoriesReceived(DevToolsCommandId command_id,
  59                             const std::set<std::string>& category_set);
  60   void OnMemoryDumpFinished(DevToolsCommandId command_id,
  61                             uint64 dump_guid,
  62                             bool success);
  63
  64   void SetupTimer(double usage_reporting_interval);
  65   void DisableRecording(
  66       const scoped_refptr<TracingController::TraceDataSink>& trace_data_sink);
  67   bool IsRecording() const;
  68   static bool IsStartupTracingActive();
  69
  70   scoped_ptr<base::Timer> buffer_usage_poll_timer_;
  71   Target target_;
  72
  73   scoped_ptr<Client> client_;
  74   DevToolsIOContext* io_context_;
  75   bool did_initiate_recording_;
  76   bool return_as_stream_;
  77   base::WeakPtrFactory<TracingHandler> weak_factory_;
  78
  79   DISALLOW_COPY_AND_ASSIGN(TracingHandler);
  80 };
  81
  82 }  // namespace tracing
  83 }  // namespace devtools
  84 }  // namespace content
  85
  86 #endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_TRACING_HANDLER_H_