OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/power/process_power_collector.h" | 5 #include "chrome/browser/power/process_power_collector.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
| 9 #include "base/memory/ptr_util.h" |
9 #include "base/process/process_handle.h" | 10 #include "base/process/process_handle.h" |
10 #include "base/process/process_metrics.h" | 11 #include "base/process/process_metrics.h" |
| 12 #include "base/stl_util.h" |
11 #include "build/build_config.h" | 13 #include "build/build_config.h" |
12 #include "chrome/browser/browser_process.h" | 14 #include "chrome/browser/browser_process.h" |
13 #include "chrome/browser/profiles/profile.h" | 15 #include "chrome/browser/profiles/profile.h" |
14 #include "chrome/browser/profiles/profile_manager.h" | 16 #include "chrome/browser/profiles/profile_manager.h" |
15 #include "chrome/browser/ui/tab_contents/tab_contents_iterator.h" | 17 #include "chrome/browser/ui/tab_contents/tab_contents_iterator.h" |
16 #include "components/power/origin_power_map.h" | 18 #include "components/power/origin_power_map.h" |
17 #include "components/power/origin_power_map_factory.h" | 19 #include "components/power/origin_power_map_factory.h" |
18 #include "content/public/browser/browser_context.h" | 20 #include "content/public/browser/browser_context.h" |
19 #include "content/public/browser/render_process_host.h" | 21 #include "content/public/browser/render_process_host.h" |
20 #include "content/public/browser/web_contents.h" | 22 #include "content/public/browser/web_contents.h" |
(...skipping 14 matching lines...) Expand all Loading... |
35 std::unique_ptr<base::ProcessMetrics> metrics, | 37 std::unique_ptr<base::ProcessMetrics> metrics, |
36 const GURL& origin, | 38 const GURL& origin, |
37 Profile* profile) | 39 Profile* profile) |
38 : metrics_(std::move(metrics)), | 40 : metrics_(std::move(metrics)), |
39 profile_(profile), | 41 profile_(profile), |
40 last_origin_(origin), | 42 last_origin_(origin), |
41 last_cpu_percent_(0), | 43 last_cpu_percent_(0), |
42 seen_this_cycle_(true) {} | 44 seen_this_cycle_(true) {} |
43 | 45 |
44 ProcessPowerCollector::PerProcessData::PerProcessData() | 46 ProcessPowerCollector::PerProcessData::PerProcessData() |
45 : profile_(NULL), | 47 : profile_(nullptr), last_cpu_percent_(0.0), seen_this_cycle_(false) {} |
46 last_cpu_percent_(0.0), | |
47 seen_this_cycle_(false) { | |
48 } | |
49 | 48 |
50 ProcessPowerCollector::PerProcessData::~PerProcessData() { | 49 ProcessPowerCollector::PerProcessData::~PerProcessData() { |
51 } | 50 } |
52 | 51 |
53 ProcessPowerCollector::ProcessPowerCollector() | 52 ProcessPowerCollector::ProcessPowerCollector() |
54 : scale_factor_(1.0) { | 53 : scale_factor_(1.0) { |
55 #if defined(OS_CHROMEOS) | 54 #if defined(OS_CHROMEOS) |
56 chromeos::DBusThreadManager::Get()->GetPowerManagerClient()->AddObserver( | 55 chromeos::DBusThreadManager::Get()->GetPowerManagerClient()->AddObserver( |
57 this); | 56 this); |
58 #endif | 57 #endif |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
91 DCHECK(!timer_.IsRunning()); | 90 DCHECK(!timer_.IsRunning()); |
92 timer_.Start(FROM_HERE, | 91 timer_.Start(FROM_HERE, |
93 base::TimeDelta::FromSeconds(kSecondsPerSample), | 92 base::TimeDelta::FromSeconds(kSecondsPerSample), |
94 this, | 93 this, |
95 &ProcessPowerCollector::HandleUpdateTimeout); | 94 &ProcessPowerCollector::HandleUpdateTimeout); |
96 } | 95 } |
97 | 96 |
98 double ProcessPowerCollector::UpdatePowerConsumption() { | 97 double ProcessPowerCollector::UpdatePowerConsumption() { |
99 double total_cpu_percent = SynchronizeProcesses(); | 98 double total_cpu_percent = SynchronizeProcesses(); |
100 | 99 |
101 for (ProcessMetricsMap::iterator it = metrics_map_.begin(); | 100 // Invalidate the process for the next cycle. |
102 it != metrics_map_.end(); | 101 for (auto& metrics : metrics_map_) |
103 ++it) { | 102 metrics.second->set_seen_this_cycle(false); |
104 // Invalidate the process for the next cycle. | |
105 it->second->set_seen_this_cycle(false); | |
106 } | |
107 | 103 |
108 RecordCpuUsageByOrigin(total_cpu_percent); | 104 RecordCpuUsageByOrigin(total_cpu_percent); |
109 return total_cpu_percent; | 105 return total_cpu_percent; |
110 } | 106 } |
111 | 107 |
112 void ProcessPowerCollector::HandleUpdateTimeout() { | 108 void ProcessPowerCollector::HandleUpdateTimeout() { |
113 UpdatePowerConsumption(); | 109 UpdatePowerConsumption(); |
114 } | 110 } |
115 | 111 |
116 double ProcessPowerCollector::SynchronizeProcesses() { | 112 double ProcessPowerCollector::SynchronizeProcesses() { |
117 // Update all tabs. | 113 // Update all tabs. |
118 for (TabContentsIterator it; !it.done(); it.Next()) { | 114 for (TabContentsIterator it; !it.done(); it.Next()) { |
119 content::RenderProcessHost* render_process = it->GetRenderProcessHost(); | 115 content::RenderProcessHost* render_process = it->GetRenderProcessHost(); |
120 // Skip incognito web contents. | 116 // Skip incognito web contents. |
121 if (render_process->GetBrowserContext()->IsOffTheRecord()) | 117 if (render_process->GetBrowserContext()->IsOffTheRecord()) |
122 continue; | 118 continue; |
123 UpdateProcessInMap(render_process, it->GetLastCommittedURL().GetOrigin()); | 119 UpdateProcessInMap(render_process, it->GetLastCommittedURL().GetOrigin()); |
124 } | 120 } |
125 | 121 |
126 // Iterate over all profiles to find all app windows to attribute all apps. | 122 // Iterate over all profiles to find all app windows to attribute all apps. |
127 ProfileManager* pm = g_browser_process->profile_manager(); | 123 ProfileManager* pm = g_browser_process->profile_manager(); |
128 std::vector<Profile*> open_profiles = pm->GetLoadedProfiles(); | 124 std::vector<Profile*> open_profiles = pm->GetLoadedProfiles(); |
129 for (std::vector<Profile*>::const_iterator it = open_profiles.begin(); | 125 for (Profile* profile : open_profiles) { |
130 it != open_profiles.end(); | |
131 ++it) { | |
132 const extensions::AppWindowRegistry::AppWindowList& app_windows = | 126 const extensions::AppWindowRegistry::AppWindowList& app_windows = |
133 extensions::AppWindowRegistry::Get(*it)->app_windows(); | 127 extensions::AppWindowRegistry::Get(profile)->app_windows(); |
134 for (extensions::AppWindowRegistry::AppWindowList::const_iterator itr = | 128 for (auto* window : app_windows) { |
135 app_windows.begin(); | 129 content::WebContents* web_contents = window->web_contents(); |
136 itr != app_windows.end(); | |
137 ++itr) { | |
138 content::WebContents* web_contents = (*itr)->web_contents(); | |
139 | |
140 UpdateProcessInMap(web_contents->GetRenderProcessHost(), | 130 UpdateProcessInMap(web_contents->GetRenderProcessHost(), |
141 web_contents->GetLastCommittedURL().GetOrigin()); | 131 web_contents->GetLastCommittedURL().GetOrigin()); |
142 } | 132 } |
143 } | 133 } |
144 | 134 |
145 // Remove invalid processes and sum up the cpu cycle. | 135 // Remove invalid processes and sum up the cpu cycle. |
146 double total_cpu_percent = 0.0; | 136 double total_cpu_percent = 0.0; |
147 ProcessMetricsMap::iterator it = metrics_map_.begin(); | 137 ProcessMetricsMap::iterator it = metrics_map_.begin(); |
148 while (it != metrics_map_.end()) { | 138 while (it != metrics_map_.end()) { |
149 if (!it->second->seen_this_cycle()) { | 139 if (!it->second->seen_this_cycle()) { |
150 metrics_map_.erase(it++); | 140 metrics_map_.erase(it++); |
151 continue; | 141 continue; |
152 } | 142 } |
153 | 143 |
154 total_cpu_percent += it->second->last_cpu_percent(); | 144 total_cpu_percent += it->second->last_cpu_percent(); |
155 ++it; | 145 ++it; |
156 } | 146 } |
157 | 147 |
158 return total_cpu_percent; | 148 return total_cpu_percent; |
159 } | 149 } |
160 | 150 |
161 void ProcessPowerCollector::RecordCpuUsageByOrigin(double total_cpu_percent) { | 151 void ProcessPowerCollector::RecordCpuUsageByOrigin(double total_cpu_percent) { |
162 DCHECK_GE(total_cpu_percent, 0); | 152 DCHECK_GE(total_cpu_percent, 0); |
163 if (total_cpu_percent == 0) | 153 if (total_cpu_percent == 0) |
164 return; | 154 return; |
165 | 155 |
166 for (ProcessMetricsMap::iterator it = metrics_map_.begin(); | 156 for (auto& metrics : metrics_map_) { |
167 it != metrics_map_.end(); | 157 double last_process_power_usage = metrics.second->last_cpu_percent(); |
168 ++it) { | |
169 double last_process_power_usage = it->second->last_cpu_percent(); | |
170 last_process_power_usage *= scale_factor_ / total_cpu_percent; | 158 last_process_power_usage *= scale_factor_ / total_cpu_percent; |
171 | 159 |
172 GURL origin = it->second->last_origin(); | 160 const GURL& origin = metrics.second->last_origin(); |
173 power::OriginPowerMap* origin_power_map = | 161 power::OriginPowerMap* origin_power_map = |
174 power::OriginPowerMapFactory::GetForBrowserContext( | 162 power::OriginPowerMapFactory::GetForBrowserContext( |
175 it->second->profile()); | 163 metrics.second->profile()); |
176 // |origin_power_map| can be NULL, if the profile is a guest profile in | 164 // |origin_power_map| can be NULL, if the profile is a guest profile in |
177 // Chrome OS. | 165 // Chrome OS. |
178 if (!origin_power_map) | 166 if (origin_power_map) |
179 continue; | 167 origin_power_map->AddPowerForOrigin(origin, last_process_power_usage); |
180 origin_power_map->AddPowerForOrigin(origin, last_process_power_usage); | |
181 } | 168 } |
182 | 169 |
183 // Iterate over all profiles to let them know we've finished updating. | 170 // Iterate over all profiles to let them know we've finished updating. |
184 ProfileManager* pm = g_browser_process->profile_manager(); | 171 ProfileManager* pm = g_browser_process->profile_manager(); |
185 std::vector<Profile*> open_profiles = pm->GetLoadedProfiles(); | 172 std::vector<Profile*> open_profiles = pm->GetLoadedProfiles(); |
186 for (std::vector<Profile*>::const_iterator it = open_profiles.begin(); | 173 for (Profile* profile : open_profiles) { |
187 it != open_profiles.end(); | |
188 ++it) { | |
189 power::OriginPowerMap* origin_power_map = | 174 power::OriginPowerMap* origin_power_map = |
190 power::OriginPowerMapFactory::GetForBrowserContext(*it); | 175 power::OriginPowerMapFactory::GetForBrowserContext(profile); |
191 if (!origin_power_map) | 176 if (origin_power_map) |
192 continue; | 177 origin_power_map->OnAllOriginsUpdated(); |
193 origin_power_map->OnAllOriginsUpdated(); | |
194 } | 178 } |
195 } | 179 } |
196 | 180 |
197 void ProcessPowerCollector::UpdateProcessInMap( | 181 void ProcessPowerCollector::UpdateProcessInMap( |
198 const content::RenderProcessHost* rph, | 182 const content::RenderProcessHost* rph, |
199 const GURL& origin) { | 183 const GURL& origin) { |
200 base::ProcessHandle handle = rph->GetHandle(); | 184 base::ProcessHandle handle = rph->GetHandle(); |
201 if (metrics_map_.find(handle) == metrics_map_.end()) { | 185 if (!ContainsKey(metrics_map_, handle)) { |
202 metrics_map_[handle] = linked_ptr<PerProcessData>(new PerProcessData( | 186 metrics_map_[handle] = base::MakeUnique<PerProcessData>( |
203 #if defined(OS_MACOSX) | 187 #if defined(OS_MACOSX) |
204 std::unique_ptr<base::ProcessMetrics>( | 188 base::ProcessMetrics::CreateProcessMetrics(handle, nullptr), |
205 base::ProcessMetrics::CreateProcessMetrics(handle, NULL)), | |
206 #else | 189 #else |
207 std::unique_ptr<base::ProcessMetrics>( | 190 base::ProcessMetrics::CreateProcessMetrics(handle), |
208 base::ProcessMetrics::CreateProcessMetrics(handle)), | |
209 #endif | 191 #endif |
210 origin, Profile::FromBrowserContext(rph->GetBrowserContext()))); | 192 origin, Profile::FromBrowserContext(rph->GetBrowserContext())); |
211 } | 193 } |
212 | 194 |
213 linked_ptr<PerProcessData>& process_data = metrics_map_[handle]; | 195 PerProcessData* process_data = metrics_map_[handle].get(); |
214 process_data->set_last_cpu_percent(std::max(0.0, | 196 process_data->set_last_cpu_percent(std::max(0.0, |
215 cpu_usage_callback_.is_null() ? process_data->metrics()->GetCPUUsage() | 197 cpu_usage_callback_.is_null() ? process_data->metrics()->GetCPUUsage() |
216 : cpu_usage_callback_.Run(handle))); | 198 : cpu_usage_callback_.Run(handle))); |
217 process_data->set_seen_this_cycle(true); | 199 process_data->set_seen_this_cycle(true); |
218 } | 200 } |
OLD | NEW |