all_builds.py: support for sharding builds

Allow sharding of the builds with the --shard= and --shards= options.

Change-Id: I9d5552ad2edc0b1210e96f0e94ce7dfd645c45f8

diff --git a/all_builds.py b/all_builds.py
index 765086528ec97081072f89f932c61e0386cbbf41..b2bab730003975fa07bd5813a5bdd81b4f1c4bc6 100755 (executable)
--- a/all_builds.py
+++ b/all_builds.py
@@ -1,8 +1,12 @@
 #!/usr/bin/python
 
+import getopt
 import subprocess
 import sys
 
+LONG_OPTIONS = ["shard=", "shards="]
+BASE_COMMAND = "./configure --enable-internal-stats --enable-experimental"
+
 def RunCommand(command):
   run = subprocess.Popen(command, shell=True)
   output = run.communicate()
@@ -26,12 +30,25 @@ def list_of_experiments():
         experiments.append(experiment)
   return experiments
 
-def main():
-  base_command = "./configure --enable-internal-stats"
-  test_build(base_command)
-  for experiment_name in list_of_experiments():
-    test_build("%s --enable-experimental --enable-%s" % (base_command,
-      experiment_name))
+def main(argv):
+  # Parse arguments
+  options = {"--shard": 0, "--shards": 1}
+  o, _ = getopt.getopt(argv[1:], None, LONG_OPTIONS)
+  options.update(o)
+
+  # Shard experiment list
+  shard = int(options["--shard"])
+  shards = int(options["--shards"])
+  experiments = list_of_experiments()
+  configs = [BASE_COMMAND]
+  configs += ["%s --enable-%s" % (BASE_COMMAND, e) for e in experiments]
+  my_configs = zip(configs, range(len(configs)))
+  my_configs = filter(lambda x: x[1] % shards == shard, my_configs)
+  my_configs = [e[0] for e in my_configs]
+
+  # Run configs for this shard
+  for config in my_configs:
+    test_build(config)
 
 def test_build(configure_command):
   print "\033[34m\033[47mTesting %s\033[0m" % (configure_command)
@@ -40,4 +57,4 @@ def test_build(configure_command):
   RunCommand("make")
 
 if __name__ == "__main__":
-  main()
+  main(sys.argv)