Leak detector posts notifications to task runner

components/metrics/leak_detector/leak_detector.cc

 // Copyright 2016 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 "components/metrics/leak_detector/leak_detector.h"
 
 #include <stdint.h>
 
 #include <algorithm>
 
 #include "base/allocator/allocator_extension.h"
 #include "base/bind.h"
 #include "base/lazy_instance.h"
+#include "base/location.h"
 #include "base/logging.h"
 #include "base/numerics/safe_conversions.h"
 #include "base/threading/thread_local.h"
 #include "components/metrics/leak_detector/custom_allocator.h"
 #include "components/metrics/leak_detector/leak_detector_impl.h"
-#include "content/public/browser/browser_thread.h"
 
 #if defined(OS_CHROMEOS)
 #include <link.h>  // for dl_iterate_phdr
 #else
 #error "Getting binary mapping info is not supported on this platform."
 #endif  // defined(OS_CHROMEOS)
 
 namespace metrics {
 
 using InternalLeakReport = leak_detector::LeakDetectorImpl::LeakReport;
@@ skipping 123 matching lines @@
   g_hook_data_tls.Get().Set(reinterpret_cast<void*>(ptr_value));
 }
 
 }  // namespace
 
 // static
 LeakDetector* LeakDetector::GetInstance() {
   return g_instance.Pointer();
 }
 
-void LeakDetector::Init(const MemoryLeakReportProto::Params& params) {
+void LeakDetector::Init(const MemoryLeakReportProto::Params& params,
+                        scoped_refptr<base::TaskRunner> task_runner) {
   DCHECK(thread_checker_.CalledOnValidThread());
   DCHECK_GT(params.sampling_rate(), 0);
 
   sampling_factor_ =
       base::saturated_cast<uint64_t>(params.sampling_rate() * UINT64_MAX);
 
   analysis_interval_bytes_ = params.analysis_interval_bytes();
   max_call_stack_unwind_depth_ = params.max_stack_depth();
 
   MappingInfo mapping = {0};
 #if defined(OS_CHROMEOS)
   // Locate the Chrome binary mapping info.
   dl_iterate_phdr(IterateLoadedObjects, &mapping);
 #endif  // defined(OS_CHROMEOS)
 
+  task_runner_ = task_runner;
+
   // CustomAllocator can use the default allocator, as long as the hook
   // functions can handle recursive calls.
   leak_detector::CustomAllocator::Initialize();
 
   // The initialization should be done only once. Check for this by examining
   // whether |impl_| has already been initialized.
   CHECK(!impl_.get()) << "Cannot initialize LeakDetector more than once!";
   impl_.reset(new leak_detector::LeakDetectorImpl(
       mapping.addr, mapping.size, params.size_suspicion_threshold(),
       params.call_stack_suspicion_threshold()));
@@ skipping 124 matching lines @@
 
 inline bool LeakDetector::ShouldSample(const void* ptr) const {
   return PointerToHash(ptr) < sampling_factor_;
 }
 
 void LeakDetector::NotifyObservers(
     const std::vector<MemoryLeakReportProto>& reports) {
   if (reports.empty())
     return;
 
-  if (!content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)) {
-    content::BrowserThread::PostTask(
-        content::BrowserThread::UI, FROM_HERE,
-        base::Bind(&LeakDetector::NotifyObservers, base::Unretained(this),
-                   reports));
+  if (!task_runner_->RunsTasksOnCurrentThread()) {
+    task_runner_->PostTask(FROM_HERE,
+                           base::Bind(&LeakDetector::NotifyObservers,
+                                      base::Unretained(this), reports));
     return;
   }
 
   {
     base::AutoLock lock(observers_lock_);
     FOR_EACH_OBSERVER(Observer, observers_, OnLeaksFound(reports));
   }
 }
 
 }  // namespace metrics