blob: 3b11c32bf117724cd4e31eac0e7f6c033a89fbfb [file] [log] [blame]
[email protected]83ab4a282012-07-12 18:19:451// 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"
[email protected]42a3ff572014-06-13 02:44:469#include "base/process/process_handle.h"
[email protected]83ab4a282012-07-12 18:19:4510#include "content/browser/histogram_subscriber.h"
11#include "content/common/child_process_messages.h"
12#include "content/public/browser/browser_child_process_host_iterator.h"
13#include "content/public/browser/browser_thread.h"
14#include "content/public/browser/child_process_data.h"
15#include "content/public/browser/render_process_host.h"
[email protected]f3b357692013-03-22 05:16:1316#include "content/public/common/process_type.h"
[email protected]83ab4a282012-07-12 18:19:4517
18namespace content {
19
20HistogramController* HistogramController::GetInstance() {
21 return Singleton<HistogramController>::get();
22}
23
24HistogramController::HistogramController() : subscriber_(NULL) {
25}
26
27HistogramController::~HistogramController() {
28}
29
30void HistogramController::OnPendingProcesses(int sequence_number,
31 int pending_processes,
32 bool end) {
33 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
34 if (subscriber_)
35 subscriber_->OnPendingProcesses(sequence_number, pending_processes, end);
36}
37
38void HistogramController::OnHistogramDataCollected(
39 int sequence_number,
40 const std::vector<std::string>& pickled_histograms) {
41 if (!BrowserThread::CurrentlyOn(BrowserThread::UI)) {
42 BrowserThread::PostTask(
43 BrowserThread::UI, FROM_HERE,
44 base::Bind(&HistogramController::OnHistogramDataCollected,
45 base::Unretained(this),
46 sequence_number,
47 pickled_histograms));
48 return;
49 }
50
51 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
52 if (subscriber_) {
53 subscriber_->OnHistogramDataCollected(sequence_number,
54 pickled_histograms);
55 }
56}
57
58void HistogramController::Register(HistogramSubscriber* subscriber) {
59 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
60 DCHECK(!subscriber_);
61 subscriber_ = subscriber;
62}
63
64void HistogramController::Unregister(
65 const HistogramSubscriber* subscriber) {
66 DCHECK_EQ(subscriber_, subscriber);
67 subscriber_ = NULL;
68}
69
70void HistogramController::GetHistogramDataFromChildProcesses(
71 int sequence_number) {
72 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
73
74 int pending_processes = 0;
75 for (BrowserChildProcessHostIterator iter; !iter.Done(); ++iter) {
[email protected]42a3ff572014-06-13 02:44:4676 const ChildProcessData& data = iter.GetData();
77 int type = data.process_type;
[email protected]9fe3e6f12013-04-23 00:36:4978 if (type != PROCESS_TYPE_PLUGIN &&
79 type != PROCESS_TYPE_GPU &&
80 type != PROCESS_TYPE_PPAPI_PLUGIN &&
81 type != PROCESS_TYPE_PPAPI_BROKER) {
[email protected]dfdb6a952012-08-31 21:08:0582 continue;
[email protected]9fe3e6f12013-04-23 00:36:4983 }
84
[email protected]42a3ff572014-06-13 02:44:4685 // In some cases, there may be no child process of the given type (for
86 // example, the GPU process may not exist and there may instead just be a
87 // GPU thread in the browser process). If that's the case, then the process
88 // handle will be base::kNullProcessHandle and we shouldn't ask it for data.
89 if (data.handle == base::kNullProcessHandle)
90 continue;
91
[email protected]83ab4a282012-07-12 18:19:4592 ++pending_processes;
93 if (!iter.Send(new ChildProcessMsg_GetChildHistogramData(sequence_number)))
94 --pending_processes;
95 }
[email protected]83ab4a282012-07-12 18:19:4596
97 BrowserThread::PostTask(
98 BrowserThread::UI,
99 FROM_HERE,
100 base::Bind(
101 &HistogramController::OnPendingProcesses,
102 base::Unretained(this),
103 sequence_number,
104 pending_processes,
105 true));
106}
107
108void HistogramController::GetHistogramData(int sequence_number) {
109 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
110
111 int pending_processes = 0;
112 for (RenderProcessHost::iterator it(RenderProcessHost::AllHostsIterator());
113 !it.IsAtEnd(); it.Advance()) {
114 ++pending_processes;
115 if (!it.GetCurrentValue()->Send(
116 new ChildProcessMsg_GetChildHistogramData(sequence_number))) {
117 --pending_processes;
118 }
119 }
120 OnPendingProcesses(sequence_number, pending_processes, false);
121
122 BrowserThread::PostTask(
123 BrowserThread::IO,
124 FROM_HERE,
125 base::Bind(&HistogramController::GetHistogramDataFromChildProcesses,
126 base::Unretained(this),
127 sequence_number));
128}
129
130} // namespace content