Преглед изворни кода

[download_from_google_storage] Add envvar for num_threads

R=engeg@google.com

Change-Id: Iedfa8b1fb18a8687c134867d8db6ca9f738bb4ae
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/5100054
Reviewed-by: George Engelbrecht <engeg@google.com>
Commit-Queue: Josip Sokcevic <sokcevic@chromium.org>
Josip Sokcevic пре 1 година
родитељ
комит
cd3fa0389e
1 измењених фајлова са 7 додато и 3 уклоњено
  1. 7 3
      download_from_google_storage.py

+ 7 - 3
download_from_google_storage.py

@@ -537,7 +537,7 @@ def main(args):
                       'Must be used with -d/--directory')
                       'Must be used with -d/--directory')
     parser.add_option('-t',
     parser.add_option('-t',
                       '--num_threads',
                       '--num_threads',
-                      default=1,
+                      default=0,
                       type='int',
                       type='int',
                       help='Number of downloader threads to run.')
                       help='Number of downloader threads to run.')
     parser.add_option('-d',
     parser.add_option('-d',
@@ -661,6 +661,10 @@ def main(args):
                      '--sha1_file or --directory')
                      '--sha1_file or --directory')
 
 
     input_filename = args[0]
     input_filename = args[0]
+    num_threads = options.num_threads
+    if not num_threads:
+        num_threads = max(
+            int(os.environ.get('DOWNLOAD_FROM_GOOGLE_STORAGE_THREADS', 1)), 1)
 
 
     # Set output filename if not specified.
     # Set output filename if not specified.
     if not options.output and not options.directory:
     if not options.output and not options.directory:
@@ -684,8 +688,8 @@ def main(args):
 
 
     try:
     try:
         return download_from_google_storage(
         return download_from_google_storage(
-            input_filename, base_url, gsutil, options.num_threads,
-            options.directory, options.recursive, options.force, options.output,
+            input_filename, base_url, gsutil, num_threads, options.directory,
+            options.recursive, options.force, options.output,
             options.ignore_errors, options.sha1_file, options.verbose,
             options.ignore_errors, options.sha1_file, options.verbose,
             options.auto_platform, options.extract)
             options.auto_platform, options.extract)
     except FileNotFoundError as e:
     except FileNotFoundError as e: