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.
96 lines
2.4 KiB
96 lines
2.4 KiB
6 years ago
|
#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
|