| // Copyright 2021 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. |
| |
| #ifndef COMPONENTS_FEEDBACK_CONTENT_CONTENT_TRACING_MANAGER_H_ |
| #define COMPONENTS_FEEDBACK_CONTENT_CONTENT_TRACING_MANAGER_H_ |
| |
| #include "components/feedback/tracing_manager.h" |
| |
| #include <map> |
| #include <memory> |
| #include <string> |
| |
| #include "base/memory/weak_ptr.h" |
| #include "base/sequence_checker.h" |
| |
| // This class is used to manage performance metrics that can be attached to |
| // feedback reports. This class is a Singleton that is owned by the preference |
| // system. It should only be created when it is enabled, and should only be |
| // accessed elsewhere via Get(). |
| // |
| // When a performance trace is desired, TracingManager::Get()->RequestTrace() |
| // should be invoked. The TracingManager will then start preparing a zipped |
| // version of the performance data. That data can then be requested via |
| // GetTraceData(). When the data is no longer needed, it should be discarded |
| // via DiscardTraceData(). |
| class ContentTracingManager : public TracingManager { |
| public: |
| ~ContentTracingManager() override; |
| |
| // Create a ContentTracingManager. Can only be called when none exists. |
| static std::unique_ptr<ContentTracingManager> Create(); |
| |
| // Get the current ContentTracingManager. Returns NULL if one doesn't exist. |
| static ContentTracingManager* Get(); |
| |
| // Request a trace ending at the current time. If a trace is already being |
| // collected, the id for that trace is returned. |
| int RequestTrace() override; |
| |
| // Get the trace data for |id|. On success, true is returned, and the data is |
| // returned via |callback|. Returns false on failure. |
| bool GetTraceData(int id, TraceDataCallback callback) override; |
| |
| // Discard the data for trace |id|. |
| void DiscardTraceData(int id) override; |
| |
| private: |
| ContentTracingManager(); |
| |
| void StartTracing(); |
| void OnTraceDataCollected(std::unique_ptr<std::string> data); |
| |
| // ID of the trace that is being collected. |
| int current_trace_id_ = 0; |
| |
| // Mapping of trace ID to trace data. |
| std::map<int, scoped_refptr<base::RefCountedString>> trace_data_; |
| |
| // Callback for the current trace request. |
| TraceDataCallback trace_callback_; |
| |
| base::WeakPtrFactory<ContentTracingManager> weak_ptr_factory_{this}; |
| }; |
| |
| #endif // COMPONENTS_FEEDBACK_CONTENT_CONTENT_TRACING_MANAGER_H_ |