device_utils.Install: Calculate host md5s at the same time as device md5s

build/android/pylib/device/device_utils.py

 # Copyright 2014 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.
 
 """Provides a variety of device interactions based on adb.
 
 Eventually, this will be based on adb_wrapper.
 """
 # pylint: disable=unused-argument
 
 import collections
 import contextlib
 import itertools
 import logging
 import multiprocessing
 import os
 import posixpath
 import re
 import shutil
 import sys
 import tempfile
+import threading
 import time
 import zipfile
 
 import pylib.android_commands
 from pylib import cmd_helper
 from pylib import constants
 from pylib import device_signal
 from pylib.constants import keyevent
 from pylib.device import adb_wrapper
 from pylib.device import decorators
@@ skipping 1053 matching lines @@
   def _ComputeDeviceChecksumsForApks(self, package_name):
     ret = self._cache['package_apk_checksums'].get(package_name)
     if ret is None:
       device_paths = self._GetApplicationPathsInternal(package_name)
       file_to_checksums = md5sum.CalculateDeviceMd5Sums(device_paths, self)
       ret = set(file_to_checksums.values())
       self._cache['package_apk_checksums'][package_name] = ret
     return ret
 
   def _ComputeStaleApks(self, package_name, host_apk_paths):
-    host_checksums = md5sum.CalculateHostMd5Sums(host_apk_paths)
+    host_checksums_holder = [None]
+    def compute_host_checksums():
+      host_checksums_holder[0] = md5sum.CalculateHostMd5Sums(host_apk_paths)
+
+    host_thread = threading.Thread(target=compute_host_checksums)
+    host_thread.start()
     device_checksums = self._ComputeDeviceChecksumsForApks(package_name)
+    host_thread.join()
+    host_checksums = host_checksums_holder[0]
     stale_apks = [k for (k, v) in host_checksums.iteritems()
                   if v not in device_checksums]
     return stale_apks, set(host_checksums.values())
 
   def _PushFilesImpl(self, host_device_tuples, files):
     size = sum(host_utils.GetRecursiveDiskUsage(h) for h, _ in files)
     file_count = len(files)
     dir_size = sum(host_utils.GetRecursiveDiskUsage(h)
                    for h, _ in host_device_tuples)
     dir_file_count = 0
@@ skipping 773 matching lines @@
     return [cls(adb) for adb in adb_wrapper.AdbWrapper.Devices()
             if not blacklisted(adb)]
 
   @decorators.WithTimeoutAndRetriesFromInstance()
   def RestartAdbd(self, timeout=None, retries=None):
     logging.info('Restarting adbd on device.')
     with device_temp_file.DeviceTempFile(self.adb, suffix='.sh') as script:
       self.WriteFile(script.name, _RESTART_ADBD_SCRIPT)
       self.RunShellCommand(['source', script.name], as_root=True)
       self.adb.WaitForDevice()