From 26090ce503f40502a62d7ebabcb4ee060b28d05b Mon Sep 17 00:00:00 2001 From: Klaus Aehlig Date: Thu, 27 Jul 2023 13:59:11 +0200 Subject: Execution API: support fetching a single artifact to memory --- src/buildtool/execution_api/remote/bazel/bazel_api.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/buildtool/execution_api/remote/bazel/bazel_api.cpp') 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 { + 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); -- cgit v1.2.3