Address comments: revert all changes except adding two new metadata keys

"census" and "census-bin"

1 files changed, 10 insertions, 0 deletions

diff --git a/include/grpc++/client_context.h b/include/grpc++/client_context.h
index 9d64715803..1f46d8e417 100644
--- a/include/grpc++/client_context.h
+++ b/include/grpc++/client_context.h
@@ -37,6 +37,7 @@
 /// side.
 /// - Control call settings such as compression and authentication.
 /// - Initial and trailing metadata coming from the server.
+/// - Get performace metrics (ie, census).
 ///
 /// Context settings are only relevant to the call they are invoked with, that
 /// is to say, they aren't sticky. Some of these settings, such as the
@@ -63,6 +64,8 @@
 #include <grpc/support/log.h>
 #include <grpc/support/time.h>
 
+struct census_context;
+
 namespace grpc {
 
 class Channel;
@@ -265,6 +268,12 @@ class ClientContext {
   /// \return The call's peer URI.
   grpc::string peer() const;
 
+  /// Get and set census context.
+  void set_census_context(struct census_context* ccp) { census_context_ = ccp; }
+  struct census_context* census_context() const {
+    return census_context_;
+  }
+
   /// Send a best-effort out-of-band cancel. The call could be in any stage.
   /// e.g. if it is already finished, it may still return success.
   ///
@@ -314,6 +323,7 @@ class ClientContext {
   grpc::string authority_;
   std::shared_ptr<CallCredentials> creds_;
   mutable std::shared_ptr<const AuthContext> auth_context_;
+  struct census_context* census_context_;
   std::multimap<grpc::string, grpc::string> send_initial_metadata_;
   std::multimap<grpc::string_ref, grpc::string_ref> recv_initial_metadata_;
   std::multimap<grpc::string_ref, grpc::string_ref> trailing_metadata_;