Support multi-stream synchronous client tests

pull/1470/head
Vijay Pai 10 years ago
parent 6dc5502871
commit 25aa12fa75
  1. 6
      test/cpp/qps/client_sync.cc

@ -97,10 +97,10 @@ class SynchronousUnaryClient GRPC_FINAL : public SynchronousClient {
class SynchronousStreamingClient GRPC_FINAL : public SynchronousClient { class SynchronousStreamingClient GRPC_FINAL : public SynchronousClient {
public: public:
SynchronousStreamingClient(const ClientConfig& config): SynchronousStreamingClient(const ClientConfig& config):
SynchronousClient(config) { SynchronousClient(config), context_(num_threads_) {
for (size_t thread_idx=0;thread_idx<num_threads_;thread_idx++){ for (size_t thread_idx=0;thread_idx<num_threads_;thread_idx++){
auto* stub = channels_[thread_idx % channels_.size()].get_stub(); auto* stub = channels_[thread_idx % channels_.size()].get_stub();
stream_ = stub->StreamingCall(&context_); stream_ = stub->StreamingCall(&context_[thread_idx]);
} }
StartThreads(num_threads_); StartThreads(num_threads_);
} }
@ -122,7 +122,7 @@ class SynchronousStreamingClient GRPC_FINAL : public SynchronousClient {
return false; return false;
} }
private: private:
grpc::ClientContext context_; std::vector<grpc::ClientContext> context_;
std::unique_ptr<grpc::ClientReaderWriter<SimpleRequest, std::unique_ptr<grpc::ClientReaderWriter<SimpleRequest,
SimpleResponse>> stream_; SimpleResponse>> stream_;
}; };

Loading…
Cancel
Save