Introduce TraceLog::AsyncEnabledStateObserver

base/trace_event/trace_log.cc

 // 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 "base/trace_event/trace_log.h"
 
 #include <algorithm>
 #include <cmath>
 #include <memory>
 #include <utility>
@@ skipping 315 matching lines @@
 
   trace_log_->lock_.AssertAcquired();
   if (trace_log_->CheckGeneration(generation_)) {
     // Return the chunk to the buffer only if the generation matches.
     trace_log_->logged_events_->ReturnChunk(chunk_index_, std::move(chunk_));
   }
   // Otherwise this method may be called from the destructor, or TraceLog will
   // find the generation mismatch and delete this buffer soon.
 }
 
+struct TraceLog::RegisteredAsyncObserver {
+  RegisteredAsyncObserver(WeakPtr<AsyncEnabledStateObserver> observer)
+      : observer(observer), task_runner(ThreadTaskRunnerHandle::Get()) {}
+  ~RegisteredAsyncObserver() {}
+
+  WeakPtr<AsyncEnabledStateObserver> observer;
+  scoped_refptr<SequencedTaskRunner> task_runner;
+};
+
 TraceLogStatus::TraceLogStatus() : event_capacity(0), event_count(0) {}
 
 TraceLogStatus::~TraceLogStatus() {}
 
 // static
 TraceLog* TraceLog::GetInstance() {
   return Singleton<TraceLog, LeakySingletonTraits<TraceLog>>::get();
 }
 
 TraceLog::TraceLog()
@@ skipping 279 matching lines @@
           Bind(&TraceSamplingThread::DefaultSamplingCallback));
       sampling_thread_->RegisterSampleBucket(
           &g_trace_state[2], "bucket2",
           Bind(&TraceSamplingThread::DefaultSamplingCallback));
       if (!PlatformThread::Create(0, sampling_thread_.get(),
                                   &sampling_thread_handle_)) {
         DCHECK(false) << "failed to create thread";
       }
     }
 
+    for (auto& it : async_observers_)

[Primiano Tucci (use gerrit), 2016/05/10 14:20:26]

I think this could even be +const, i.e. const auto& (not a big deal if that's
not the case)

[Xiaocheng, 2016/05/11 05:19:20]

Done.

+      it.second.task_runner->PostTask(

[Sami, 2016/05/10 14:42:16]

nit: Please add braces around this loop.

[Xiaocheng, 2016/05/11 05:19:20]

Done.

+          FROM_HERE, Bind(&AsyncEnabledStateObserver::OnTraceLogEnabled,
+                          it.second.observer));
+
     dispatching_to_observer_list_ = true;
     observer_list = enabled_state_observer_list_;
   }
   // Notify observers outside the lock in case they trigger trace events.
   for (size_t i = 0; i < observer_list.size(); ++i)
     observer_list[i]->OnTraceLogEnabled();
 
   {
     AutoLock lock(lock_);
     dispatching_to_observer_list_ = false;
@@ skipping 64 matching lines @@
 
   trace_config_.Clear();
   subtle::NoBarrier_Store(&watch_category_, 0);
   watch_event_name_ = "";
   UpdateCategoryGroupEnabledFlags();
   AddMetadataEventsWhileLocked();
 
   // Remove metadata events so they will not get added to a subsequent trace.
   metadata_events_.clear();
 
+  for (auto& it : async_observers_)
+    it.second.task_runner->PostTask(

[Sami, 2016/05/10 14:42:16]

nit: Please add braces around this loop.

[Xiaocheng, 2016/05/11 05:19:20]

Done.

+        FROM_HERE, Bind(&AsyncEnabledStateObserver::OnTraceLogDisabled,
+                        it.second.observer));
+
   dispatching_to_observer_list_ = true;
   std::vector<EnabledStateObserver*> observer_list =
       enabled_state_observer_list_;
 
   {
     // Dispatch to observers outside the lock in case the observer triggers a
     // trace event.
     AutoUnlock unlock(lock_);
     for (size_t i = 0; i < observer_list.size(); ++i)
       observer_list[i]->OnTraceLogDisabled();
@@ skipping 974 matching lines @@
     }
   }
 }
 #endif  // defined(OS_WIN)
 
 void ConvertableToTraceFormat::EstimateTraceMemoryOverhead(
     TraceEventMemoryOverhead* overhead) {
   overhead->Add("ConvertableToTraceFormat(Unknown)", sizeof(*this));
 }
 
+void TraceLog::AddAsyncEnabledStateObserver(
+    WeakPtr<AsyncEnabledStateObserver> listener) {
+  AutoLock lock(lock_);
+  async_observers_.insert(
+      std::make_pair(listener.get(), RegisteredAsyncObserver(listener)));
+}
+
+void TraceLog::RemoveAsyncEnabledStateObserver(
+    AsyncEnabledStateObserver* listener) {
+  AutoLock lock(lock_);
+  async_observers_.erase(listener);
+}
+
+bool TraceLog::HasAsyncEnabledStateObserver(
+    AsyncEnabledStateObserver* listener) const {
+  AutoLock lock(lock_);
+  return async_observers_.find(listener) != async_observers_.end();

[Primiano Tucci (use gerrit), 2016/05/10 14:20:26]

I think that .count(listener) != 0 is, in some STL implementations, a bit more
efficient, as you don't need to populate the iterator

[Xiaocheng, 2016/05/11 05:19:20]

Let's use ContainsKey() which looks less hacky (and makes it someone else's
responsibility to care about efficiency).

+}
+
 }  // namespace trace_event
 }  // namespace base
 
 namespace trace_event_internal {
 
 ScopedTraceBinaryEfficient::ScopedTraceBinaryEfficient(
     const char* category_group,
     const char* name) {
   // The single atom works because for now the category_group can only be "gpu".
   DCHECK_EQ(strcmp(category_group, "gpu"), 0);
@@ skipping 21 matching lines @@
 }
 
 ScopedTraceBinaryEfficient::~ScopedTraceBinaryEfficient() {
   if (*category_group_enabled_) {
     TRACE_EVENT_API_UPDATE_TRACE_EVENT_DURATION(category_group_enabled_, name_,
                                                 event_handle_);
   }
 }
 
 }  // namespace trace_event_internal