|
|
|
@ -1,12 +1,30 @@ |
|
|
|
|
/*
|
|
|
|
|
* |
|
|
|
|
* Copyright 2019 gRPC authors. |
|
|
|
|
* |
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
|
* You may obtain a copy of the License at |
|
|
|
|
* |
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
* |
|
|
|
|
* Unless required by applicable law or agreed to in writing, software |
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS, |
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
|
|
|
* See the License for the specific language governing permissions and |
|
|
|
|
* limitations under the License. |
|
|
|
|
* |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
#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 <condition_variable> |
|
|
|
|
#include <memory> |
|
|
|
|
#include <mutex> |
|
|
|
|
#include <condition_variable> |
|
|
|
|
#include <sstream> |
|
|
|
|
#include "src/proto/grpc/testing/echo.grpc.pb.h" |
|
|
|
|
#include "test/cpp/util/string_ref_helper.h" |
|
|
|
|
|
|
|
|
|
namespace grpc { |
|
|
|
|
namespace testing { |
|
|
|
@ -22,14 +40,16 @@ class CallbackStreamingTestService |
|
|
|
|
EchoResponse* response, |
|
|
|
|
experimental::ServerCallbackRpcController* controller) override; |
|
|
|
|
|
|
|
|
|
experimental::ServerBidiReactor<EchoRequest, EchoResponse>* BidiStream() 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) |
|
|
|
|
EchoResponse* response, ClientContext* context, |
|
|
|
|
int num_msgs_to_send) |
|
|
|
|
: request_{request}, |
|
|
|
|
response_{response}, |
|
|
|
|
context_{context}, |
|
|
|
@ -89,7 +109,6 @@ class BidiClient |
|
|
|
|
bool done_ = false; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} // namespace testing
|
|
|
|
|
} // namespace grpc
|
|
|
|
|
#endif // TEST_CPP_MICROBENCHMARKS_CALLBACK_TEST_SERVICE_H
|
|
|
|
|