diff options
author | Maksim Denisov <denisov.maksim@huawei.com> | 2024-09-17 10:20:55 +0200 |
---|---|---|
committer | Maksim Denisov <denisov.maksim@huawei.com> | 2024-09-18 09:45:19 +0200 |
commit | 60748d2914568e54f852281141f7c128c3ca1f2c (patch) | |
tree | 144ebe2871a46bc8b250892e3fd0ac948647763e /src/buildtool/execution_api/remote/bazel/bytestream_client.hpp | |
parent | 7d2f632b1dd1fe2ca01ef89716efe355e4d32687 (diff) | |
download | justbuild-60748d2914568e54f852281141f7c128c3ca1f2c.tar.gz |
Implement ByteStreamUtils::ReadRequest class
...and remove split serialization/deserialization implementations.
Diffstat (limited to 'src/buildtool/execution_api/remote/bazel/bytestream_client.hpp')
-rw-r--r-- | src/buildtool/execution_api/remote/bazel/bytestream_client.hpp | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/src/buildtool/execution_api/remote/bazel/bytestream_client.hpp b/src/buildtool/execution_api/remote/bazel/bytestream_client.hpp index a444e61b..02538384 100644 --- a/src/buildtool/execution_api/remote/bazel/bytestream_client.hpp +++ b/src/buildtool/execution_api/remote/bazel/bytestream_client.hpp @@ -70,11 +70,11 @@ class ByteStreamClient { IncrementalReader( gsl::not_null<google::bytestream::ByteStream::Stub*> const& stub, - Logger const* logger, - std::string const& resource_name) + ByteStreamUtils::ReadRequest&& read_request, + Logger const* logger) : logger_{logger} { google::bytestream::ReadRequest request{}; - request.set_resource_name(resource_name); + request.set_resource_name(std::move(read_request).ToString()); reader_ = stub->Read(&ctx_, request); } }; @@ -86,14 +86,14 @@ class ByteStreamClient { CreateChannelWithCredentials(server, port, auth)); } - [[nodiscard]] auto IncrementalRead( - std::string const& resource_name) const noexcept -> IncrementalReader { - return IncrementalReader{stub_.get(), &logger_, resource_name}; + [[nodiscard]] auto IncrementalRead(ByteStreamUtils::ReadRequest&& request) + const noexcept -> IncrementalReader { + return IncrementalReader{stub_.get(), std::move(request), &logger_}; } - [[nodiscard]] auto Read(std::string const& resource_name) const noexcept - -> std::optional<std::string> { - auto reader = IncrementalRead(resource_name); + [[nodiscard]] auto Read(ByteStreamUtils::ReadRequest&& request) + const noexcept -> std::optional<std::string> { + auto reader = IncrementalRead(std::move(request)); std::string output{}; auto data = reader.Next(); while (data and not data->empty()) { |