content/browser/profiler_controller_impl.cc

   1 // Copyright (c) 2012 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
   5 #include "content/browser/profiler_controller_impl.h"
   6
   7 #include "base/bind.h"
   8 #include "base/tracked_objects.h"
   9 #include "content/common/child_process_messages.h"
  10 #include "content/public/browser/browser_child_process_host_iterator.h"
  11 #include "content/public/browser/browser_thread.h"
  12 #include "content/public/browser/child_process_data.h"
  13 #include "content/public/browser/profiler_subscriber.h"
  14 #include "content/public/browser/render_process_host.h"
  15
  16 namespace content {
  17
  18 ProfilerController* ProfilerController::GetInstance() {
  19   return ProfilerControllerImpl::GetInstance();
  20 }
  21
  22 ProfilerControllerImpl* ProfilerControllerImpl::GetInstance() {
  23   return Singleton<ProfilerControllerImpl>::get();
  24 }
  25
  26 ProfilerControllerImpl::ProfilerControllerImpl() : subscriber_(NULL) {
  27 }
  28
  29 ProfilerControllerImpl::~ProfilerControllerImpl() {
  30 }
  31
  32 void ProfilerControllerImpl::OnPendingProcesses(int sequence_number,
  33                                                 int pending_processes,
  34                                                 bool end) {
  35   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  36   if (subscriber_)
  37     subscriber_->OnPendingProcesses(sequence_number, pending_processes, end);
  38 }
  39
  40 void ProfilerControllerImpl::OnProfilerDataCollected(
  41     int sequence_number,
  42     const tracked_objects::ProcessDataSnapshot& profiler_data,
  43     ProcessType process_type) {
  44   if (!BrowserThread::CurrentlyOn(BrowserThread::UI)) {
  45     BrowserThread::PostTask(
  46         BrowserThread::UI, FROM_HERE,
  47         base::Bind(&ProfilerControllerImpl::OnProfilerDataCollected,
  48                    base::Unretained(this),
  49                    sequence_number,
  50                    profiler_data,
  51                    process_type));
  52     return;
  53   }
  54
  55   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  56   if (subscriber_) {
  57     subscriber_->OnProfilerDataCollected(sequence_number, profiler_data,
  58                                          process_type);
  59   }
  60 }
  61
  62 void ProfilerControllerImpl::Register(ProfilerSubscriber* subscriber) {
  63   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  64   DCHECK(!subscriber_);
  65   subscriber_ = subscriber;
  66 }
  67
  68 void ProfilerControllerImpl::Unregister(const ProfilerSubscriber* subscriber) {
  69   DCHECK_EQ(subscriber_, subscriber);
  70   subscriber_ = NULL;
  71 }
  72
  73 void ProfilerControllerImpl::GetProfilerDataFromChildProcesses(
  74     int sequence_number) {
  75   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  76
  77   int pending_processes = 0;
  78   for (BrowserChildProcessHostIterator iter; !iter.Done(); ++iter) {
  79     ++pending_processes;
  80     if (!iter.Send(new ChildProcessMsg_GetChildProfilerData(sequence_number)))
  81       --pending_processes;
  82   }
  83
  84   BrowserThread::PostTask(
  85       BrowserThread::UI,
  86       FROM_HERE,
  87       base::Bind(
  88           &ProfilerControllerImpl::OnPendingProcesses,
  89           base::Unretained(this),
  90           sequence_number,
  91           pending_processes,
  92           true));
  93 }
  94
  95 void ProfilerControllerImpl::GetProfilerData(int sequence_number) {
  96   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  97
  98   int pending_processes = 0;
  99   for (RenderProcessHost::iterator it(RenderProcessHost::AllHostsIterator());
 100        !it.IsAtEnd(); it.Advance()) {
 101     ++pending_processes;
 102     if (!it.GetCurrentValue()->Send(
 103             new ChildProcessMsg_GetChildProfilerData(sequence_number))) {
 104       --pending_processes;
 105     }
 106   }
 107   OnPendingProcesses(sequence_number, pending_processes, false);
 108
 109   BrowserThread::PostTask(
 110       BrowserThread::IO,
 111       FROM_HERE,
 112       base::Bind(&ProfilerControllerImpl::GetProfilerDataFromChildProcesses,
 113                  base::Unretained(this),
 114                  sequence_number));
 115 }
 116
 117 }  // namespace content