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