Delay download after sharding

Test: presubmit
Bug: 319480034
Change-Id: I6b62d5e25eb4b5bc7b87ac7f8cba162011567e9c

1 files changed, 13 insertions, 0 deletions

diff --git a/src/com/android/tradefed/sandbox/SandboxOptions.java b/src/com/android/tradefed/sandbox/SandboxOptions.java
index d021db3d4..97d74c6c0 100644
--- a/src/com/android/tradefed/sandbox/SandboxOptions.java
+++ b/src/com/android/tradefed/sandbox/SandboxOptions.java
@@ -53,6 +53,7 @@ public final class SandboxOptions {
             "sandbox-use-partial-download-cache";
     private static final String SANDBOX_SPLIT_DISCOVERY = "sandbox-split-discovery";
     private static final String SANDBOX_PARALLEL_DOWNLOAD = "sandbox-parallel-download";
+    private static final String DELAY_DOWNLOAD_AFTER_SHARDING = "delay-download-after-sharding";
 
     @Option(
         name = TF_LOCATION,
@@ -173,6 +174,13 @@ public final class SandboxOptions {
             description = "Enable parallel download during sandbox setup.")
     private boolean mUseSandboxParallelDownload = true;
 
+    @Option(
+            name = DELAY_DOWNLOAD_AFTER_SHARDING,
+            description =
+                    "Feature to delegate most of the heavy download after sharding to reduce"
+                            + " downloaded size.")
+    private boolean mDelayDownloadAfterSharding = false;
+
     /**
      * Returns the provided directories containing the Trade Federation version to use for
      * sandboxing the run.
@@ -280,4 +288,9 @@ public final class SandboxOptions {
     public boolean shouldUseParallelDownload() {
         return mUseSandboxParallelDownload;
     }
+
+    /** Returns whether or not to delay download after the sharding. */
+    public boolean delayDownloadAfterSharding() {
+        return mDelayDownloadAfterSharding;
+    }
 }