| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 # Copyright 2016 The Chromium Authors. All rights reserved. | 2 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """Inspection of the prefetch predictor database. | 6 """Inspection of the prefetch predictor database. |
| 7 | 7 |
| 8 On Android, the database can be extracted using: | 8 On Android, the database can be extracted using: |
| 9 adb pull \ | 9 adb pull \ |
| 10 '/data/user/0/$package_name/app_chrome/Default/Network Action Predictor' | 10 '/data/user/0/$package_name/app_chrome/Default/Network Action Predictor' |
| (...skipping 11 matching lines...) Expand all Loading... |
| 22 | 22 |
| 23 | 23 |
| 24 class Entry(object): | 24 class Entry(object): |
| 25 """Represents an entry in the predictor database.""" | 25 """Represents an entry in the predictor database.""" |
| 26 HEADER = ( | 26 HEADER = ( |
| 27 'score,main_page_url,resource_type,number_of_hits,number_of_misses,' | 27 'score,main_page_url,resource_type,number_of_hits,number_of_misses,' |
| 28 'consecutive_misses,average_position,confidence,resource_url') | 28 'consecutive_misses,average_position,confidence,resource_url') |
| 29 | 29 |
| 30 def __init__( | 30 def __init__( |
| 31 self, main_page_url, resource_url, resource_type, number_of_hits, | 31 self, main_page_url, resource_url, resource_type, number_of_hits, |
| 32 number_of_misses, consecutive_misses, average_position): | 32 number_of_misses, consecutive_misses, average_position, priority): |
| 33 self.main_page_url = main_page_url | 33 self.main_page_url = main_page_url |
| 34 self.resource_url = resource_url | 34 self.resource_url = resource_url |
| 35 self.resource_type = resource_type | 35 self.resource_type = resource_type |
| 36 self.number_of_hits = int(number_of_hits) | 36 self.number_of_hits = int(number_of_hits) |
| 37 self.number_of_misses = int(number_of_misses) | 37 self.number_of_misses = int(number_of_misses) |
| 38 self.consecutive_misses = int(consecutive_misses) | 38 self.consecutive_misses = int(consecutive_misses) |
| 39 self.average_position = int(average_position) | 39 self.average_position = int(average_position) |
| 40 self.priority = int(priority) |
| 40 self.confidence = float(number_of_hits) / ( | 41 self.confidence = float(number_of_hits) / ( |
| 41 number_of_hits + number_of_misses) | 42 number_of_hits + number_of_misses) |
| 42 self.score = self._Score() | 43 self.score = self._Score() |
| 43 | 44 |
| 44 def _Score(self): | 45 def _Score(self): |
| 45 """Mirrors ResourcePrefetchPredictorTables::ResourceRow::UpdateScore.""" | 46 """Mirrors ResourcePrefetchPredictorTables::ResourceRow::UpdateScore.""" |
| 46 multiplier = 1 | 47 multiplier = 1 |
| 47 if self.resource_type in (ResourceType.STYLESHEET, ResourceType.SCRIPT): | 48 if self.resource_type in (ResourceType.STYLESHEET, ResourceType.SCRIPT): |
| 48 multiplier = 2 | 49 multiplier = 2 |
| 49 return multiplier * 100 - self.average_position | 50 return multiplier * 100 - self.average_position |
| 50 | 51 |
| 51 @classmethod | 52 @classmethod |
| 52 def FromRow(cls, row): | 53 def FromRow(cls, row): |
| 53 """Builds an entry from a database row.""" | 54 """Builds an entry from a database row.""" |
| 54 return Entry(*row) | 55 return Entry(*row) |
| 55 | 56 |
| 56 def __str__(self): | 57 def __str__(self): |
| 57 return '%f,%s,%d,%d,%d,%d,%d,%f\t%s' % ( | 58 return '%f,%s,%d,%d,%d,%d,%d,%f,%d\t%s' % ( |
| 58 self.score, self.main_page_url, self.resource_type, | 59 self.score, self.main_page_url, self.resource_type, |
| 59 self.number_of_hits, self.number_of_misses, self.consecutive_misses, | 60 self.number_of_hits, self.number_of_misses, self.consecutive_misses, |
| 60 self.average_position, self.confidence, self.resource_url) | 61 self.average_position, self.confidence, self.priority, |
| 62 self.resource_url) |
| 61 | 63 |
| 62 | 64 |
| 63 def FilterAndSort(entries, domain): | 65 def FilterAndSort(entries, domain): |
| 64 """Filters and sorts the entries to be prefetched for a given domain. | 66 """Filters and sorts the entries to be prefetched for a given domain. |
| 65 | 67 |
| 66 Uses the default thresholds defined in resource_prefetch_common.cc. | 68 Uses the default thresholds defined in resource_prefetch_common.cc. |
| 67 """ | 69 """ |
| 68 result = filter( | 70 result = filter( |
| 69 lambda x: ((domain is None or x.main_page_url == domain) | 71 lambda x: ((domain is None or x.main_page_url == domain) |
| 70 and x.confidence > .7 | 72 and x.confidence > .7 |
| 71 and x.number_of_hits >= 2), entries) | 73 and x.number_of_hits >= 2), entries) |
| 72 return sorted(result, key=operator.attrgetter('score'), reverse=True) | 74 return sorted(result, key=operator.attrgetter('score'), reverse=True) |
| 73 | 75 |
| 74 | 76 |
| 75 def DatabaseStats(filename, domain): | 77 def DatabaseStats(filename, domain): |
| 76 connection = sqlite3.connect(filename) | 78 connection = sqlite3.connect(filename) |
| 77 c = connection.cursor() | 79 c = connection.cursor() |
| 78 query = ('SELECT main_page_url, resource_url, resource_type, number_of_hits, ' | 80 query = ('SELECT main_page_url, resource_url, resource_type, number_of_hits, ' |
| 79 'number_of_misses, consecutive_misses, average_position ' | 81 'number_of_misses, consecutive_misses, average_position, priority ' |
| 80 'FROM resource_prefetch_predictor_host') | 82 'FROM resource_prefetch_predictor_host') |
| 81 entries = [Entry.FromRow(row) for row in c.execute(query)] | 83 entries = [Entry.FromRow(row) for row in c.execute(query)] |
| 82 prefetched = FilterAndSort(entries, domain) | 84 prefetched = FilterAndSort(entries, domain) |
| 83 print Entry.HEADER | 85 print Entry.HEADER |
| 84 for x in prefetched: | 86 for x in prefetched: |
| 85 print x | 87 print x |
| 86 | 88 |
| 87 | 89 |
| 88 def main(): | 90 def main(): |
| 89 parser = argparse.ArgumentParser() | 91 parser = argparse.ArgumentParser() |
| 90 parser.add_argument('-f', dest='database_filename', required=True, | 92 parser.add_argument('-f', dest='database_filename', required=True, |
| 91 help='Path to the database') | 93 help='Path to the database') |
| 92 parser.add_argument('-d', dest='domain', default=None, help='Domain') | 94 parser.add_argument('-d', dest='domain', default=None, help='Domain') |
| 93 args = parser.parse_args() | 95 args = parser.parse_args() |
| 94 DatabaseStats(args.database_filename, args.domain) | 96 DatabaseStats(args.database_filename, args.domain) |
| 95 | 97 |
| 96 | 98 |
| 97 if __name__ == '__main__': | 99 if __name__ == '__main__': |
| 98 main() | 100 main() |
| OLD | NEW |