| // 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 "mojo/common/task_tracker.h" |
| |
| #include <vector> |
| |
| #include "base/location.h" |
| #include "base/logging.h" |
| #include "base/macros.h" |
| #include "base/threading/thread_local.h" |
| #include "base/tracked_objects.h" |
| |
| namespace mojo { |
| namespace common { |
| |
| namespace { |
| |
| class TrackingActivation { |
| public: |
| TrackingActivation() : birth_(nullptr) {} |
| |
| bool Start(const char* function_name, |
| const char* file_name, |
| int line_number, |
| const void* program_counter); |
| void End(); |
| bool IsAlive() const { return birth_ != nullptr; } |
| |
| private: |
| tracked_objects::TaskStopwatch* stopwatch() { |
| return reinterpret_cast<tracked_objects::TaskStopwatch*>(stopwatch_heap_); |
| } |
| |
| tracked_objects::Births* birth_; |
| // TaskStopwatch isn't copyable, but we don't want to allocate it on the heap |
| // so as not to slow things down, hence replacement new. |
| char stopwatch_heap_[sizeof(tracked_objects::TaskStopwatch)]; |
| |
| DISALLOW_COPY_AND_ASSIGN(TrackingActivation); |
| }; |
| |
| bool TrackingActivation::Start(const char* function_name, |
| const char* file_name, |
| int line_number, |
| const void* program_counter) { |
| // So far we don't track nested invocations. |
| if (IsAlive()) |
| return false; |
| birth_ = tracked_objects::ThreadData::TallyABirthIfActive( |
| tracked_objects::Location(function_name, file_name, line_number, |
| program_counter)); |
| if (!birth_) |
| return false; |
| |
| (new (stopwatch()) tracked_objects::TaskStopwatch())->Start(); |
| return true; |
| } |
| |
| void TrackingActivation::End() { |
| DCHECK(IsAlive()); |
| stopwatch()->Stop(); |
| tracked_objects::ThreadData::TallyRunInAScopedRegionIfTracking(birth_, |
| *stopwatch()); |
| stopwatch()->tracked_objects::TaskStopwatch::~TaskStopwatch(); |
| birth_ = nullptr; |
| } |
| |
| base::ThreadLocalPointer<TrackingActivation> g_activation; |
| |
| } // namespace |
| |
| // static |
| intptr_t TaskTracker::StartTracking(const char* function_name, |
| const char* file_name, |
| int line_number, |
| const void* program_counter) { |
| TrackingActivation* activation = g_activation.Get(); |
| if (!activation) { |
| // Leak this. |
| activation = new TrackingActivation(); |
| g_activation.Set(activation); |
| } |
| |
| if (!activation->Start(function_name, file_name, line_number, |
| program_counter)) |
| return 0; |
| return reinterpret_cast<intptr_t>(activation); |
| } |
| |
| // static |
| void TaskTracker::EndTracking(intptr_t id) { |
| if (0 == id) |
| return; |
| // |EndTracking()| must be called from the same thread of |StartTracking()|. |
| DCHECK_EQ(reinterpret_cast<TrackingActivation*>(id), g_activation.Get()); |
| reinterpret_cast<TrackingActivation*>(id)->End(); |
| } |
| |
| } // namespace common |
| } // namespace mojo |