perfetto_platform.h 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. // Copyright 2020 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. #ifndef BASE_TRACING_PERFETTO_PLATFORM_H_
  5. #define BASE_TRACING_PERFETTO_PLATFORM_H_
  6. #include "third_party/perfetto/include/perfetto/tracing/platform.h"
  7. #include "base/base_export.h"
  8. #include "base/memory/scoped_refptr.h"
  9. #include "base/threading/thread_id_name_manager.h"
  10. #include "base/threading/thread_local_storage.h"
  11. namespace base {
  12. class DeferredSequencedTaskRunner;
  13. namespace tracing {
  14. class BASE_EXPORT PerfettoPlatform : public perfetto::Platform,
  15. ThreadIdNameManager::Observer {
  16. public:
  17. // Specifies the type of task runner used by Perfetto.
  18. // TODO(skyostil): Move all scenarios to use the default task runner to
  19. // avoid problems with unexpected re-entrancy and IPC deadlocks.
  20. enum class TaskRunnerType {
  21. // Use Perfetto's own task runner which runs tasks on a dedicated (internal)
  22. // thread.
  23. kBuiltin,
  24. // Use base::ThreadPool.
  25. kThreadPool,
  26. };
  27. explicit PerfettoPlatform(TaskRunnerType = TaskRunnerType::kThreadPool);
  28. ~PerfettoPlatform() override;
  29. SequencedTaskRunner* task_runner() const;
  30. bool did_start_task_runner() const { return did_start_task_runner_; }
  31. void StartTaskRunner(scoped_refptr<SequencedTaskRunner>);
  32. // perfetto::Platform implementation:
  33. ThreadLocalObject* GetOrCreateThreadLocalObject() override;
  34. std::unique_ptr<perfetto::base::TaskRunner> CreateTaskRunner(
  35. const CreateTaskRunnerArgs&) override;
  36. std::string GetCurrentProcessName() override;
  37. // ThreadIdNameManager::Observer implementation.
  38. void OnThreadNameChanged(const char* name) override;
  39. private:
  40. const TaskRunnerType task_runner_type_;
  41. scoped_refptr<DeferredSequencedTaskRunner> deferred_task_runner_;
  42. bool did_start_task_runner_ = false;
  43. ThreadLocalStorage::Slot thread_local_object_;
  44. };
  45. } // namespace tracing
  46. } // namespace base
  47. #endif // BASE_TRACING_PERFETTO_PLATFORM_H_