The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.4 KiB
95 lines
2.4 KiB
#ifndef TEST_CPP_MICROBENCHMARKS_CALLBACK_TEST_SERVICE_H |
|
#define TEST_CPP_MICROBENCHMARKS_CALLBACK_TEST_SERVICE_H |
|
|
|
#include "src/proto/grpc/testing/echo.grpc.pb.h" |
|
#include "test/cpp/util/string_ref_helper.h" |
|
#include <sstream> |
|
#include <memory> |
|
#include <mutex> |
|
#include <condition_variable> |
|
|
|
namespace grpc { |
|
namespace testing { |
|
|
|
const char* const kServerFinishAfterNReads = "server_finish_after_n_reads"; |
|
const char* const kServerResponseStreamsToSend = "server_responses_to_send"; |
|
|
|
class CallbackStreamingTestService |
|
: public EchoTestService::ExperimentalCallbackService { |
|
public: |
|
CallbackStreamingTestService() {} |
|
void Echo(ServerContext* context, const EchoRequest* request, |
|
EchoResponse* response, |
|
experimental::ServerCallbackRpcController* controller) override; |
|
|
|
experimental::ServerBidiReactor<EchoRequest, EchoResponse>* BidiStream() override; |
|
}; |
|
|
|
class BidiClient |
|
: public grpc::experimental::ClientBidiReactor<EchoRequest, EchoResponse> { |
|
public: |
|
BidiClient(EchoTestService::Stub* stub, EchoRequest* request, |
|
EchoResponse* response, ClientContext* context, int num_msgs_to_send) |
|
: request_{request}, |
|
response_{response}, |
|
context_{context}, |
|
msgs_to_send_{num_msgs_to_send}{ |
|
stub->experimental_async()->BidiStream(context_, this); |
|
MaybeWrite(); |
|
StartRead(response_); |
|
StartCall(); |
|
} |
|
|
|
void OnReadDone(bool ok) override { |
|
if (ok && reads_complete_ < msgs_to_send_) { |
|
reads_complete_++; |
|
StartRead(response_); |
|
} |
|
} |
|
|
|
void OnWriteDone(bool ok) override { |
|
if (!ok) { |
|
return; |
|
} |
|
writes_complete_++; |
|
MaybeWrite(); |
|
} |
|
|
|
void OnDone(const Status& s) override { |
|
GPR_ASSERT(s.ok()); |
|
std::unique_lock<std::mutex> l(mu_); |
|
done_ = true; |
|
cv_.notify_one(); |
|
} |
|
|
|
void Await() { |
|
std::unique_lock<std::mutex> l(mu_); |
|
while (!done_) { |
|
cv_.wait(l); |
|
} |
|
} |
|
|
|
private: |
|
void MaybeWrite() { |
|
if (writes_complete_ == msgs_to_send_) { |
|
StartWritesDone(); |
|
} else { |
|
StartWrite(request_); |
|
} |
|
} |
|
|
|
EchoRequest* request_; |
|
EchoResponse* response_; |
|
ClientContext* context_; |
|
int reads_complete_{0}; |
|
int writes_complete_{0}; |
|
const int msgs_to_send_; |
|
std::mutex mu_; |
|
std::condition_variable cv_; |
|
bool done_ = false; |
|
}; |
|
|
|
|
|
} // namespace testing |
|
} // namespace grpc |
|
#endif // TEST_CPP_MICROBENCHMARKS_CALLBACK_TEST_SERVICE_H
|
|
|