Enable --git-cache-dir option in chromite/cbuildbot.

scripts/slave/recipe_modules/chromite/config.py

 # 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 os
-import types
+import re
 
 from recipe_engine.config import config_item_context, ConfigGroup
 from recipe_engine.config import Dict, Single, Set
 
 
+# Regular expression to match branch versions.
+#
+# Examples:
+# - release-R54-8743.B
+# - stabilize-8743.B
+# - factory-gale-8743.19.B
+# - stabilize-8743.25.B
+_VERSION_RE = re.compile(r'^.*-(\d+)\.(\d+\.)?B$')
+
+
 def BaseConfig(CBB_CONFIG=None, CBB_BRANCH=None, CBB_BUILD_NUMBER=None,
                CBB_DEBUG=False, CBB_CLOBBER=False, CBB_BUILDBUCKET_ID=None,
                CBB_MASTER_BUILD_ID=None, **_kwargs):
   return ConfigGroup(
     # Base mapping of repository key to repository name.
     repositories = Dict(value_type=Set(basestring)),
 
     # Checkout Chromite at this branch. "origin/" will be prepended.
     chromite_branch = Single(basestring, empty_val=CBB_BRANCH or 'master'),
 
@@ skipping 46 matching lines @@
       config_repo = Single(basestring),
 
       # This disables Chromite bootstrapping by omitting the explicit "--branch"
       # argument.
       disable_bootstrap = Single(bool),
 
       # Whether this Chromite version supports warm cache.
       # https://chromium-review.googlesource.com/#/c/348011
       supports_repo_cache = Single(bool),
 
+      # If set, supply the "--git-cache-dir" option with this value.
+      git_cache_dir = Single(basestring),
+
       # If supplied, forward to cbuildbot as '--buildbucket-id'
       buildbucket_id = Single(basestring, empty_val=CBB_BUILDBUCKET_ID)
     ),
 
     # A list of branches whose Chromite version is "old". Old Chromite
     # buildbot commands reside in the "buildbot" subdirectory of the Chromite
     # repository instead of the "bin".
     old_chromite_branches = Set(basestring),
 
     # A list of branches whose builders should not use a shared buildroot.
     non_shared_root_branches = Set(basestring),
 
     # A list of branches whose builders checkout Chrome from SVN instead of Git.
     chrome_svn_branches = Set(basestring),
 
+    # If "chromite_branch" includes a branch version, this will be set to the
+    # version value. Otherwise, this will be None.
+    #
+    # Set in "base".
+    branch_version = Single(int),
+
     # Directory where a warm repo cache is stored. If set, and if the current
     # build supports a warm cache, this will be used to bootstrap the Chromite
     # checkout.
-    repo_cache_dir = Single(basestring)
+    repo_cache_dir = Single(basestring),
+
+    # The branch version where the "--git-cache" flag was introduced.
+    # Set to a ToT build after R54 branch, "release-R54-8743.B".
+    git_cache_min_branch_version = Single(int, empty_val=8829),
   )
 
 config_ctx = config_item_context(BaseConfig)
 
 
 @config_ctx()
 def base(c):
   c.repositories['tryjob'] = []
   c.repositories['chromium'] = []
   c.repositories['cros_manifest'] = []
@@ skipping 11 matching lines @@
   c.chrome_svn_branches.update((
      # TODO(dnj): Remove this once internal expectations are updated.
     'factory-4455.B',
      # TODO(dnj): Remove this once internal expectations are updated.
     'factory-zako-5220.B',
 
     'factory-rambi-5517.B',
     'factory-nyan-5772.B',
   ))
 
+  # Resolve branch version, if available.
+  assert c.chromite_branch, "A Chromite branch must be configured."
+  version = _VERSION_RE.match(c.chromite_branch)
+  if version:
+    c.branch_version = int(version.group(1))
+
   # If running on a testing slave, enable "--debug" so Chromite doesn't cause
   # actual production effects.
   if 'TESTING_MASTER_HOST' in os.environ:  # pragma: no cover
     c.cbb.debug = True
 
 
 @config_ctx(includes=['base'])
+def cros(c):
+  """Base configuration for CrOS builders to inherit from."""
+  # Enable Git cache on all ToT builds and release branches that support it.
+  if not (c.branch_version and
+          c.branch_version < c.git_cache_min_branch_version):
+    c.cbb.git_cache_dir = '/b/cros_git_cache'
+
+
+@config_ctx(includes=['cros'])
 def external(c):
   c.repositories['tryjob'].extend([
       'https://chromium.googlesource.com/chromiumos/tryjobs',
       'https://chrome-internal.googlesource.com/chromeos/tryjobs',
       ])
   c.repositories['chromium'].append(
       'https://chromium.googlesource.com/chromium/src')
   c.repositories['cros_manifest'].append(
       'https://chromium.googlesource.com/chromiumos/manifest-versions')
 
@@ skipping 24 matching lines @@
 def chromiumos_coverage(c):
   c.use_chrome_version = True
   c.read_cros_manifest = True
   c.cbb.chrome_rev = 'stable'
   c.cbb.config_repo = 'https://example.com/repo.git'
 
 # TODO(dnj): Remove this config once variant support is removed.
 @config_ctx()
 def coverage_variant(c):
   c.cbb.chrome_rev = 'canary'