Keep track of hashes triggered instead of builds.

scripts/slave/gatekeeper_ng.py

 #!/usr/bin/env 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.
 
 """Closes tree if configured masters have failed tree-closing steps.
 
 Given a list of masters, gatekeeper_ng will get a list of the latest builds from
 the specified masters. It then checks if any tree-closing steps have failed, and
 if so closes the tree and emails appropriate parties. Configuration for which
@@ skipping 12 matching lines @@
 import os
 import random
 import re
 import sys
 import time
 import urllib
 import urllib2
 
 from common import chromium_utils
 from slave import gatekeeper_ng_config
+from slave import gatekeeper_ng_db
 
 SCRIPTS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)),
                            '..', '..')
 
 # Buildbot status enum.
 SUCCESS, WARNINGS, FAILURE, SKIPPED, EXCEPTION, RETRY = range(6)
 
 
 def get_pwd(password_file):
   if os.path.isfile(password_file):
@@ skipping 19 matching lines @@
 def get_root_json(master_url):
   """Pull down root JSON which contains builder and build info."""
   logging.info('opening %s' % (master_url + '/json'))
   with closing(urllib2.urlopen(master_url + '/json')) as f:
     return json.load(f)
 
 
 def find_new_builds(master_url, root_json, build_db):
   """Given a dict of previously-seen builds, find new builds on each builder.
 
-  Note that we use the 'cachedBuilds here since it should be faster, and this
+  Note that we use the 'cachedBuilds' here since it should be faster, and this
   script is meant to be run frequently enough that it shouldn't skip any builds.
 
   'Frequently enough' means 1 minute in the case of Buildbot or cron, so the
   only way for gatekeeper_ng to be overwhelmed is if > cachedBuilds builds
   complete within 1 minute. As cachedBuilds is scaled per number of slaves per
   builder, the only way for this to really happen is if a build consistently
   takes < 1 minute to complete.
   """
   new_builds = {}
-  build_db[master_url] = build_db.get(master_url, {})
+  build_db.masters[master_url] = build_db.masters.get(master_url, {})
+
+  last_finished_build = {}
+  for builder, builds in build_db.masters[master_url].iteritems():
+    finished = [int(y[0]) for y in filter(
+        lambda x: x[1].finished, builds.iteritems())]
+    if finished:
+      last_finished_build[builder] = max(finished)
+
   for buildername, builder in root_json['builders'].iteritems():
+    # cachedBuilds are the builds in the cache, while currentBuilds are the
+    # currently running builds. Thus cachedBuilds can be unfinished or finished,
+    # while currentBuilds are always unfinished.
     candidate_builds = set(builder['cachedBuilds'] + builder['currentBuilds'])
-    if buildername in build_db[master_url]:
-      new_builds[buildername] = [x for x in candidate_builds
-                                 if x > build_db[master_url][buildername]]
+    if buildername in last_finished_build:
+      new_builds[buildername] = [
+          buildnum for buildnum in candidate_builds
+          if buildnum > last_finished_build[buildername]]
     else:
-      new_builds[buildername] = candidate_builds
-
-    # This is a heuristic, as currentBuilds may become completed by the time we
-    # scan them. The buildDB is fixed up later to account for this.
-    completed = set(builder['cachedBuilds']) - set(builder['currentBuilds'])
-    if completed:
-      build_db[master_url][buildername] = max(completed)
+      if buildername in build_db.masters[master_url]:
+        # Scan finished builds as well as unfinished.
+        new_builds[buildername] = candidate_builds
+      else:
+        # New builder or master, ignore past builds.
+        new_builds[buildername] = builder['currentBuilds']
 
   return new_builds
 
 
 def find_new_builds_per_master(masters, build_db):
   """Given a list of masters, find new builds and collect them under a dict."""
   builds = {}
   master_jsons = {}
   for master in masters:
     root_json = get_root_json(master)
     master_jsons[master] = root_json
     builds[master] = find_new_builds(master, root_json, build_db)
   return builds, master_jsons
 
 
-def get_build_json(url_pair):
-  url, master = url_pair
+def get_build_json(url_tuple):
+  """Doenloads the json of a specific build."""
+  url, master, builder, buildnum = url_tuple
   logging.debug('opening %s...' % url)
   with closing(urllib2.urlopen(url)) as f:
-    return json.load(f), master
+    return json.load(f), master, builder, buildnum
 
 
-def get_build_jsons(master_builds, build_db, processes):
+def get_build_jsons(master_builds, processes):
   """Get all new builds on specified masters.
 
   This takes a dict in the form of [master][builder][build], formats that URL
   and appends that to url_list. Then, it forks out and queries each build_url
   for build information.
   """
   url_list = []
   for master, builder_dict in master_builds.iteritems():
     for builder, new_builds in builder_dict.iteritems():
-      for build in new_builds:
+      for buildnum in new_builds:
         safe_builder = urllib.quote(builder)
-        url = master + '/json/builders/%s/builds/%s' % (safe_builder, build)
-        url_list.append((url, master))
-  # The async/get is so that ctrl-c can interrupt the scans.
-  # See http://stackoverflow.com/questions/1408356/
-  # keyboard-interrupts-with-pythons-multiprocessing-pool
-  with chromium_utils.MultiPool(processes) as pool:
-    builds = filter(bool, pool.map_async(get_build_json, url_list).get(9999999))
+        url = master + '/json/builders/%s/builds/%s' % (safe_builder,
+                                                        buildnum)
+        url_list.append((url, master, builder, buildnum))
 
-  for build_json, master in builds:
-    if build_json.get('results', None) is not None:
-      build_db[master][build_json['builderName']] = max(
-          build_json['number'],
-          build_db[master][build_json['builderName']])
+  # Prevent map from hanging, see http://bugs.python.org/issue12157.
+  if url_list:
+    # The async/get is so that ctrl-c can interrupt the scans.
+    # See http://stackoverflow.com/questions/1408356/
+    # keyboard-interrupts-with-pythons-multiprocessing-pool
+    with chromium_utils.MultiPool(processes) as pool:
+      builds = filter(bool, pool.map_async(get_build_json, url_list).get(
+          9999999))
+  else:
+    builds = []
+
   return builds
 
 
-def check_builds(master_builds, master_jsons, build_db, gatekeeper_config):
+def propagate_build_json_to_db(build_db, builds):
+  """Propagates build status changes from build_json to build_db."""
+  for build_json, master, builder, buildnum in builds:
+    build = build_db.masters[master].setdefault(builder, {}).get(buildnum)
+    if not build:
+      build = gatekeeper_ng_db.gen_build()
+
+    if build_json.get('results', None) is not None:
+      build = build._replace(finished=True)  # pylint: disable=W0212
+
+    build_db.masters[master][builder][buildnum] = build
+
+
+def check_builds(master_builds, master_jsons, gatekeeper_config):
   """Given a gatekeeper configuration, see which builds have failed."""
   failed_builds = []
-  for build_json, master_url in master_builds:
+  for build_json, master_url, builder, buildnum in master_builds:
     gatekeeper_sections = gatekeeper_config.get(master_url, [])
     for gatekeeper_section in gatekeeper_sections:
+      section_hash = gatekeeper_ng_config.gatekeeper_section_hash(
+          gatekeeper_section)
+
       if build_json['builderName'] in gatekeeper_section:
         gatekeeper = gatekeeper_section[build_json['builderName']]
       elif '*' in gatekeeper_section:
         gatekeeper = gatekeeper_section['*']
       else:
         gatekeeper = {}
+
       steps = build_json['steps']
       forgiving = set(gatekeeper.get('forgiving_steps', []))
       forgiving_optional = set(gatekeeper.get('forgiving_optional', []))
       closing_steps = set(gatekeeper.get('closing_steps', [])) | forgiving
       closing_optional = set(
           gatekeeper.get('closing_optional', [])) | forgiving_optional
       tree_notify = set(gatekeeper.get('tree_notify', []))
       sheriff_classes = set(gatekeeper.get('sheriff_classes', []))
       subject_template = gatekeeper.get('subject_template',
                                         gatekeeper_ng_config.DEFAULTS[
@@ skipping 23 matching lines @@
         unsatisfied_steps &= finished_steps
 
       # If the entire build failed.
       if (not unsatisfied_steps and 'results' in build_json and
           build_json['results'] != SUCCESS and respect_build_status):
         unsatisfied_steps.add('[overall build status]')
 
       buildbot_url = master_jsons[master_url]['project']['buildbotURL']
       project_name = master_jsons[master_url]['project']['title']
 
-      logging.debug('%sbuilders/%s/builds/%d ----', buildbot_url,
-                    build_json['builderName'], build_json['number'])
-      logging.debug('  build steps: %s', ', '.join(s['name'] for s in steps))
-      logging.debug('  closing steps: %s', ', '.join(closing_steps))
-      logging.debug('  closing optional steps: %s', ', '.join(closing_optional))
-      logging.debug('  finished steps: %s', ', '.join(finished_steps))
-      logging.debug('  successful: %s', ', '.join(successful_steps))
-      logging.debug('  build complete: %s', bool(
-          build_json.get('results', None) is not None))
-      logging.debug('  unsatisfied steps: %s', ', '.join(unsatisfied_steps))
-      logging.debug('  set to close tree: %s', close_tree)
-      logging.debug('  build failed: %s', bool(unsatisfied_steps))
-      logging.debug('----')
+
+      if unsatisfied_steps:
+        failed_builds.append(({'base_url': buildbot_url,
+                               'build': build_json,
+                               'close_tree': close_tree,
+                               'forgiving_steps': (
+                                   forgiving | forgiving_optional),
+                               'project_name': project_name,
+                               'sheriff_classes': sheriff_classes,
+                               'subject_template': subject_template,
+                               'tree_notify': tree_notify,
+                               'unsatisfied': unsatisfied_steps,
+                              },
+                              master_url,
+                              builder,
+                              buildnum,
+                              section_hash))
+
+  return failed_builds
 
 
-      if unsatisfied_steps:
-        build_db[master_url][build_json['builderName']] = max(
-            build_json['number'],
-            build_db[master_url][build_json['builderName']])
+def debounce_failures(failed_builds, build_db):
+  """Using trigger information in build_db, make sure we don't double-fire."""
+  true_failed_builds = []
+  for build, master_url, builder, buildnum, section_hash in failed_builds:
+    logging.debug('%sbuilders/%s/builds/%d ----', build['base_url'],
+                  builder, buildnum)
 
-        failed_builds.append({'base_url': buildbot_url,
-                              'build': build_json,
-                              'close_tree': close_tree,
-                              'forgiving_steps': forgiving | forgiving_optional,
-                              'project_name': project_name,
-                              'sheriff_classes': sheriff_classes,
-                              'subject_template': subject_template,
-                              'tree_notify': tree_notify,
-                              'unsatisfied': unsatisfied_steps,
-                             })
+    build_db_builder = build_db.masters[master_url][builder]
+    triggered = build_db_builder[buildnum].triggered
+    if section_hash in triggered:
+      logging.debug('  section has already been triggered for this build, '
+                    'skipping...')
+    else:
+      # Propagates since the dictionary is the same as in build_db.
+      triggered.append(section_hash)
+      true_failed_builds.append(build)
 
-  return failed_builds
+      logging.debug('  section hash: %s', section_hash)
+      logging.debug('  build steps: %s', ', '.join(
+          s['name'] for s in build['build']['steps']))
+      logging.debug('  build complete: %s', bool(
+          build['build'].get('results', None) is not None))
+      logging.debug('  unsatisfied steps: %s', ', '.join(build['unsatisfied']))
+      logging.debug('  set to close tree: %s', build['close_tree'])
+      logging.debug('  build failed: %s', bool(build['unsatisfied']))
+
+    logging.debug('----')
+
+  return true_failed_builds
 
 
 def parse_sheriff_file(url):
   """Given a sheriff url, download and parse the appropirate sheriff list."""
   with closing(urllib2.urlopen(url)) as f:
     line = f.readline()
   usernames_matcher_ = re.compile(r'document.write\(\'([\w, ]+)\'\)')
   usernames_match = usernames_matcher_.match(line)
   sheriffs = set()
   if usernames_match:
@@ skipping 144 matching lines @@
   # Deduplicate emails.
   keyfunc = lambda x: x[1]
   for k, g in itertools.groupby(sorted(filtered_emails_to_send, key=keyfunc),
                                 keyfunc):
     watchers = list(reduce(operator.or_, [set(e[0]) for e in g], set()))
     build_data = json.loads(k)
     build_data['recipients'] = watchers
     submit_email(email_app_url, build_data, secret)
 
 
-
-def get_build_db(filename):
-  """Open the build_db file.
-
-  filename: the filename of the build db.
-  """
-  build_db = None
-  if os.path.isfile(filename):
-    print 'loading build_db from', filename
-    with open(filename) as f:
-      build_db = json.load(f)
-
-  return build_db or {}
-
-
-def save_build_db(build_db, filename):
-  """Save the build_db file.
-
-  build_db: dictionary to jsonize and store as build_db.
-  filename: the filename of the build db.
-  """
-  print 'saving build_db to', filename
-  with open(filename, 'wb') as f:
-    json.dump(build_db, f)
-
-
 def get_options():
   prog_desc = 'Closes the tree if annotated builds fail.'
   usage = '%prog [options] <one or more master urls>'
   parser = optparse.OptionParser(usage=(usage + '\n\n' + prog_desc))
   parser.add_option('--build-db', default='build_db.json',
                     help='records the last-seen build for each builder')
   parser.add_option('--clear-build-db', action='store_true',
                     help='reset build_db to be empty')
   parser.add_option('--sync-build-db', action='store_true',
                     help='don\'t process any builds, but update build_db '
@@ skipping 86 matching lines @@
   if options.verify:
     return 0
 
   if options.flatten_json:
     if not options.no_hashes:
       gatekeeper_config = gatekeeper_ng_config.inject_hashes(gatekeeper_config)
     gatekeeper_ng_config.flatten_to_json(gatekeeper_config, sys.stdout)
     print
     return 0
 
+  if options.set_status:

[ghost stip (do not use), 2014/02/22 10:03:07]

moved from line 549

+    options.password = get_pwd(options.password_file)
+
   masters = set(args)
   if not masters <= set(gatekeeper_config):
     print 'The following masters are not present in the gatekeeper config:'
     for m in masters - set(gatekeeper_config):
       print '  ' + m
     return 1
 
   if options.clear_build_db:
     build_db = {}
-    save_build_db(build_db, options.build_db)
+    gatekeeper_ng_db.save_build_db(build_db, gatekeeper_config,
+                                   options.build_db)
   else:
-    build_db = get_build_db(options.build_db)
+    build_db = gatekeeper_ng_db.get_build_db(options.build_db)
 
   new_builds, master_jsons = find_new_builds_per_master(masters, build_db)
+  build_jsons = get_build_jsons(new_builds, options.parallelism)
+  propagate_build_json_to_db(build_db, build_jsons)
+
   if options.sync_build_db:
-    save_build_db(build_db, options.build_db)
+    gatekeeper_ng_db.save_build_db(build_db, gatekeeper_config,
+                                   options.build_db)
     return 0
-  build_jsons = get_build_jsons(new_builds, build_db, options.parallelism)
-  failed_builds = check_builds(build_jsons, master_jsons, build_db,
-                               gatekeeper_config)
-  if options.set_status:
-    options.password = get_pwd(options.password_file)
+
+  failed_builds = check_builds(build_jsons, master_jsons, gatekeeper_config)
+  failed_builds = debounce_failures(failed_builds, build_db)
 
   close_tree_if_failure(failed_builds, options.status_user, options.password,
                         options.status_url, options.set_status,
                         options.sheriff_url, options.default_from_email,
                         options.email_app_url, options.email_app_secret,
                         options.email_domain, options.filter_domain,
                         options.disable_domain_filter)
 
   if not options.skip_build_db_update:
-    save_build_db(build_db, options.build_db)
+    gatekeeper_ng_db.save_build_db(build_db, gatekeeper_config,
+                                   options.build_db)
 
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())