diff options
author | heretic <heretic@yandex-team.ru> | 2022-02-10 16:45:46 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:45:46 +0300 |
commit | 81eddc8c0b55990194e112b02d127b87d54164a9 (patch) | |
tree | 9142afc54d335ea52910662635b898e79e192e49 /contrib/libs/grpc/test/cpp/util/cli_call.cc | |
parent | 397cbe258b9e064f49c4ca575279f02f39fef76e (diff) | |
download | ydb-81eddc8c0b55990194e112b02d127b87d54164a9.tar.gz |
Restoring authorship annotation for <heretic@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/libs/grpc/test/cpp/util/cli_call.cc')
-rw-r--r-- | contrib/libs/grpc/test/cpp/util/cli_call.cc | 60 |
1 files changed, 30 insertions, 30 deletions
diff --git a/contrib/libs/grpc/test/cpp/util/cli_call.cc b/contrib/libs/grpc/test/cpp/util/cli_call.cc index 722ad88970..5b3631667f 100644 --- a/contrib/libs/grpc/test/cpp/util/cli_call.cc +++ b/contrib/libs/grpc/test/cpp/util/cli_call.cc @@ -25,23 +25,23 @@ #include <grpcpp/client_context.h> #include <grpcpp/support/byte_buffer.h> -#include <cmath> -#include <iostream> -#include <utility> - +#include <cmath> +#include <iostream> +#include <utility> + namespace grpc { namespace testing { namespace { void* tag(int i) { return (void*)static_cast<intptr_t>(i); } } // namespace -Status CliCall::Call(const std::shared_ptr<grpc::Channel>& channel, - const TString& method, const TString& request, - TString* response, +Status CliCall::Call(const std::shared_ptr<grpc::Channel>& channel, + const TString& method, const TString& request, + TString* response, const OutgoingMetadataContainer& metadata, IncomingMetadataContainer* server_initial_metadata, IncomingMetadataContainer* server_trailing_metadata) { - CliCall call(channel, method, metadata); + CliCall call(channel, method, metadata); call.Write(request); call.WritesDone(); if (!call.Read(response, server_initial_metadata)) { @@ -51,8 +51,8 @@ Status CliCall::Call(const std::shared_ptr<grpc::Channel>& channel, } CliCall::CliCall(const std::shared_ptr<grpc::Channel>& channel, - const TString& method, - const OutgoingMetadataContainer& metadata, CliArgs args) + const TString& method, + const OutgoingMetadataContainer& metadata, CliArgs args) : stub_(new grpc::GenericStub(channel)) { gpr_mu_init(&write_mu_); gpr_cv_init(&write_cv_); @@ -62,22 +62,22 @@ CliCall::CliCall(const std::shared_ptr<grpc::Channel>& channel, ctx_.AddMetadata(iter->first, iter->second); } } - - // Set deadline if timeout > 0 (default value -1 if no timeout specified) - if (args.timeout > 0) { - int64_t timeout_in_ns = ceil(args.timeout * 1e9); - - // Convert timeout (in nanoseconds) to a deadline - auto deadline = - gpr_time_add(gpr_now(GPR_CLOCK_MONOTONIC), - gpr_time_from_nanos(timeout_in_ns, GPR_TIMESPAN)); - ctx_.set_deadline(deadline); - } else if (args.timeout != -1) { - fprintf( - stderr, - "WARNING: Non-positive timeout value, skipping setting deadline.\n"); - } - + + // Set deadline if timeout > 0 (default value -1 if no timeout specified) + if (args.timeout > 0) { + int64_t timeout_in_ns = ceil(args.timeout * 1e9); + + // Convert timeout (in nanoseconds) to a deadline + auto deadline = + gpr_time_add(gpr_now(GPR_CLOCK_MONOTONIC), + gpr_time_from_nanos(timeout_in_ns, GPR_TIMESPAN)); + ctx_.set_deadline(deadline); + } else if (args.timeout != -1) { + fprintf( + stderr, + "WARNING: Non-positive timeout value, skipping setting deadline.\n"); + } + call_ = stub_->PrepareCall(&ctx_, method, &cq_); call_->StartCall(tag(1)); void* got_tag; @@ -91,7 +91,7 @@ CliCall::~CliCall() { gpr_mu_destroy(&write_mu_); } -void CliCall::Write(const TString& request) { +void CliCall::Write(const TString& request) { void* got_tag; bool ok; @@ -103,7 +103,7 @@ void CliCall::Write(const TString& request) { GPR_ASSERT(ok); } -bool CliCall::Read(TString* response, +bool CliCall::Read(TString* response, IncomingMetadataContainer* server_initial_metadata) { void* got_tag; bool ok; @@ -137,7 +137,7 @@ void CliCall::WritesDone() { GPR_ASSERT(ok); } -void CliCall::WriteAndWait(const TString& request) { +void CliCall::WriteAndWait(const TString& request) { grpc::Slice req_slice(request); grpc::ByteBuffer send_buffer(&req_slice, 1); @@ -161,7 +161,7 @@ void CliCall::WritesDoneAndWait() { } bool CliCall::ReadAndMaybeNotifyWrite( - TString* response, IncomingMetadataContainer* server_initial_metadata) { + TString* response, IncomingMetadataContainer* server_initial_metadata) { void* got_tag; bool ok; grpc::ByteBuffer recv_buffer; |