Execution API: support fetching a single artifact to memory

2 files changed, 13 insertions, 0 deletions

diff --git a/src/buildtool/execution_api/remote/bazel/bazel_api.cpp b/src/buildtool/execution_api/remote/bazel/bazel_api.cpp
index c4f170a3..e2da0188 100644
--- a/src/buildtool/execution_api/remote/bazel/bazel_api.cpp
+++ b/src/buildtool/execution_api/remote/bazel/bazel_api.cpp
@@ -251,6 +251,15 @@ auto BazelApi::CreateAction(
     return api->Upload(container, /*skip_find_missing=*/true);
 }
 
+[[nodiscard]] auto BazelApi::RetrieveToMemory(
+    Artifact::ObjectInfo const& artifact_info) -> std::optional<std::string> {
+    auto blobs = network_->ReadBlobs({artifact_info.digest}).Next();
+    if (blobs.size() == 1) {
+        return blobs.at(0).data;
+    }
+    return std::nullopt;
+}
+
 [[nodiscard]] auto BazelApi::Upload(BlobContainer const& blobs,
                                     bool skip_find_missing) noexcept -> bool {
     return network_->UploadBlobs(blobs, skip_find_missing);
diff --git a/src/buildtool/execution_api/remote/bazel/bazel_api.hpp b/src/buildtool/execution_api/remote/bazel/bazel_api.hpp
index 6edfae5f..310fb347 100644
--- a/src/buildtool/execution_api/remote/bazel/bazel_api.hpp
+++ b/src/buildtool/execution_api/remote/bazel/bazel_api.hpp
@@ -81,6 +81,10 @@ class BazelApi final : public IExecutionApi {
     [[nodiscard]] auto IsAvailable(std::vector<ArtifactDigest> const& digests)
         const noexcept -> std::vector<ArtifactDigest> final;
 
+    [[nodiscard]] auto RetrieveToMemory(
+        Artifact::ObjectInfo const& artifact_info)
+        -> std::optional<std::string> final;
+
   private:
     std::shared_ptr<BazelNetwork> network_;