Use stream rpc to ensure all related requests go to a single server.

1 files changed, 10 insertions, 1 deletions

diff --git a/test/cpp/util/proto_reflection_descriptor_database.h b/test/cpp/util/proto_reflection_descriptor_database.h
index bf94654c3d..f37135e681 100644
--- a/test/cpp/util/proto_reflection_descriptor_database.h
+++ b/test/cpp/util/proto_reflection_descriptor_database.h
@@ -80,9 +80,18 @@ class ProtoReflectionDescriptorDatabase
   }
 
  private:
+  typedef ClientReaderWriter<
+      grpc::reflection::v1alpha::DescriptorDatabaseRequest,
+      grpc::reflection::v1alpha::DescriptorDatabaseResponse>
+      ClientStream;
+
   const google::protobuf::FileDescriptorProto ParseFileDescriptorProtoResponse(
-      reflection::v1alpha::FileDescriptorProtoResponse* response);
+      const std::string& byte_fd_proto);
+
+  const std::shared_ptr<ClientStream> GetStream();
 
+  std::shared_ptr<ClientStream> stream_;
+  grpc::ClientContext ctx_;
   std::unique_ptr<grpc::reflection::v1alpha::ServerReflection::Stub> stub_;
   std::unordered_set<string> known_files_;
   std::unordered_set<string> missing_symbols_;