80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
# Copyright 2013 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.
|
|
import sys
|
|
|
|
from metrics import Metric
|
|
|
|
class MemoryMetric(Metric):
|
|
"""MemoryMetric gathers memory statistics from the browser object."""
|
|
|
|
def __init__(self, browser):
|
|
super(MemoryMetric, self).__init__()
|
|
self._browser = browser
|
|
self._memory_stats = None
|
|
self._start_commit_charge = None
|
|
|
|
def Start(self, page=None, tab=None):
|
|
"""Record the initial value of 'SystemCommitCharge'."""
|
|
self._start_commit_charge = self._browser.memory_stats['SystemCommitCharge']
|
|
|
|
def Stop(self, page=None, tab=None):
|
|
"""Fetch the browser memory stats."""
|
|
assert self._start_commit_charge, 'Must call Start() first'
|
|
self._memory_stats = self._browser.memory_stats
|
|
|
|
def AddResults(self, tab, results):
|
|
"""Add summary results to the results object."""
|
|
assert self._memory_stats, 'Must call Stop() first'
|
|
if not self._memory_stats['Browser']:
|
|
return
|
|
|
|
metric = 'resident_set_size'
|
|
if sys.platform == 'win32':
|
|
metric = 'working_set'
|
|
|
|
def AddSummariesForProcessTypes(process_types_memory, process_type_trace):
|
|
"""Add all summaries to the results for a given set of process types.
|
|
|
|
Args:
|
|
process_types_memory: A list of process types, e.g. Browser, 'Renderer'
|
|
process_type_trace: The name of this set of process types in the output
|
|
"""
|
|
def AddSummary(value_name_memory, value_name_trace):
|
|
"""Add a summary to the results for a given statistic.
|
|
|
|
Args:
|
|
value_name_memory: Name of some statistic, e.g. VM, WorkingSetSize
|
|
value_name_trace: Name of this statistic to be used in the output
|
|
"""
|
|
if len(process_types_memory) > 1 and value_name_memory.endswith('Peak'):
|
|
return
|
|
values = []
|
|
for process_type_memory in process_types_memory:
|
|
stats = self._memory_stats[process_type_memory]
|
|
if value_name_memory in stats:
|
|
values.append(stats[value_name_memory])
|
|
if values:
|
|
results.AddSummary(value_name_trace + process_type_trace,
|
|
'bytes', sum(values), data_type='unimportant')
|
|
|
|
AddSummary('VM', 'vm_final_size_')
|
|
AddSummary('WorkingSetSize', 'vm_%s_final_size_' % metric)
|
|
AddSummary('PrivateDirty', 'vm_private_dirty_final_')
|
|
AddSummary('ProportionalSetSize', 'vm_proportional_set_size_final_')
|
|
AddSummary('VMPeak', 'vm_peak_size_')
|
|
AddSummary('WorkingSetSizePeak', '%s_peak_size_' % metric)
|
|
|
|
AddSummariesForProcessTypes(['Browser'], 'browser')
|
|
AddSummariesForProcessTypes(['Renderer'], 'renderer')
|
|
AddSummariesForProcessTypes(['Gpu'], 'gpu')
|
|
AddSummariesForProcessTypes(['Browser', 'Renderer', 'Gpu'], 'total')
|
|
|
|
end_commit_charge = self._memory_stats['SystemCommitCharge']
|
|
commit_charge_difference = end_commit_charge - self._start_commit_charge
|
|
results.AddSummary('commit_charge', 'kb', commit_charge_difference,
|
|
data_type='unimportant')
|
|
results.AddSummary('processes', 'count', self._memory_stats['ProcessCount'],
|
|
data_type='unimportant')
|
|
|