Remote: Fix a bug that outputs of actions tagged with no-remote are u…

src/main/java/com/google/devtools/build/lib/remote/ByteStreamBuildEventArtifactUploader.java

@@ -30,6 +30,7 @@
 import com.google.devtools.build.lib.events.Event;
 import com.google.devtools.build.lib.events.ExtendedEventHandler;
 import com.google.devtools.build.lib.remote.common.RemoteActionExecutionContext;
+import com.google.devtools.build.lib.remote.common.RemoteActionExecutionContext.Step;
 import com.google.devtools.build.lib.remote.util.DigestUtil;
 import com.google.devtools.build.lib.remote.util.TracingMetadataUtils;
 import com.google.devtools.build.lib.vfs.Path;
@@ -256,7 +257,8 @@ private Single<PathConverter> upload(Set<Path> files) {
 
     RequestMetadata metadata =
         TracingMetadataUtils.buildMetadata(buildRequestId, commandId, "bes-upload", null);
-    RemoteActionExecutionContext context = RemoteActionExecutionContext.createForBES(metadata);
+    RemoteActionExecutionContext context = RemoteActionExecutionContext.create(metadata);
+    context.setStep(Step.UPLOAD_BES_FILES);
 
     return Single.using(
         remoteCache::retain,

src/main/java/com/google/devtools/build/lib/remote/RemoteAction.java

@@ -0,0 +1,121 @@
+package com.google.devtools.build.lib.remote;
+
+import build.bazel.remote.execution.v2.Action;
+import build.bazel.remote.execution.v2.Command;
+import build.bazel.remote.execution.v2.Digest;
+import com.google.devtools.build.lib.actions.ActionInput;
+import com.google.devtools.build.lib.actions.ForbiddenActionInputException;
+import com.google.devtools.build.lib.actions.Spawn;
+import com.google.devtools.build.lib.exec.SpawnRunner.SpawnExecutionContext;
+import com.google.devtools.build.lib.remote.common.NetworkTime;
+import com.google.devtools.build.lib.remote.common.RemoteActionExecutionContext;
+import com.google.devtools.build.lib.remote.common.RemoteCacheClient.ActionKey;
+import com.google.devtools.build.lib.remote.common.RemotePathResolver;
+import com.google.devtools.build.lib.remote.merkletree.MerkleTree;
+import com.google.devtools.build.lib.vfs.PathFragment;
+import java.io.IOException;
+import java.util.SortedMap;
+
+/** A value class representing an action which can be executed remotely. */
+public class RemoteAction {
+
+  private final Spawn spawn;
+  private final SpawnExecutionContext spawnExecutionContext;
+  private final RemoteActionExecutionContext remoteActionExecutionContext;
+  private final RemotePathResolver remotePathResolver;
+  private final MerkleTree merkleTree;
+  private final Digest commandHash;
+  private final Command command;
+  private final Action action;
+  private final ActionKey actionKey;
+
+  RemoteAction(
+      Spawn spawn,
+      SpawnExecutionContext spawnExecutionContext,
+      RemoteActionExecutionContext remoteActionExecutionContext,
+      RemotePathResolver remotePathResolver,
+      MerkleTree merkleTree,
+      Digest commandHash,
+      Command command,
+      Action action,
+      ActionKey actionKey) {
+    this.spawn = spawn;
+    this.spawnExecutionContext = spawnExecutionContext;
+    this.remoteActionExecutionContext = remoteActionExecutionContext;
+    this.remotePathResolver = remotePathResolver;
+    this.merkleTree = merkleTree;
+    this.commandHash = commandHash;
+    this.command = command;
+    this.action = action;
+    this.actionKey = actionKey;
+  }
+
+  public RemoteActionExecutionContext getRemoteActionExecutionContext() {
+    return remoteActionExecutionContext;
+  }
+
+  public SpawnExecutionContext getSpawnExecutionContext() {
+    return spawnExecutionContext;
+  }
+
+  /** Returns the {@link Spawn} that owns this action. */
+  public Spawn getSpawn() {
+    return spawn;
+  }
+
+  /**
+   * Returns the sum of file sizes plus protobuf sizes used to represent the inputs of this action.
+   */
+  public long getInputBytes() {
+    return merkleTree.getInputBytes();
+  }
+
+  /** Returns the number of input files of this action. */
+  public long getInputFiles() {
+    return merkleTree.getInputFiles();
+  }
+
+  /** Returns the id this is action. */
+  public String getActionId() {
+    return actionKey.getDigest().getHash();
+  }
+
+  /** Returns the {@link ActionKey} of this action. */
+  public ActionKey getActionKey() {
+    return actionKey;
+  }
+
+  /** Returns underlying {@link Action} of this remote action. */
+  public Action getAction() {
+    return action;
+  }
+
+  public Digest getCommandHash() {
+    return commandHash;
+  }
+
+  public Command getCommand() {
+    return command;
+  }
+
+  public MerkleTree getMerkleTree() {
+    return merkleTree;
+  }
+
+  /**
+   * Returns a {@link SortedMap} which maps from input paths for remote action to {@link
+   * ActionInput}.
+   */
+  public SortedMap<PathFragment, ActionInput> getInputMap()
+      throws IOException, ForbiddenActionInputException {
+    return remotePathResolver.getInputMapping(spawnExecutionContext);
+  }
+
+  /**
+   * Returns the {@link NetworkTime} instance used to measure the network time during the action
+   * execution.
+   */
+  public NetworkTime getNetworkTime() {
+    return remoteActionExecutionContext.getNetworkTime();
+  }
+}