Skip to content
Snippets Groups Projects
Commit 6dd49a5c authored by David Garcia Quintas's avatar David Garcia Quintas
Browse files

Added some methods to context inspector. Also minor tweak to server context IsCancelled() method

parent e4109866
No related branches found
No related tags found
No related merge requests found
...@@ -97,7 +97,7 @@ class ServerContext { ...@@ -97,7 +97,7 @@ class ServerContext {
void AddInitialMetadata(const grpc::string& key, const grpc::string& value); void AddInitialMetadata(const grpc::string& key, const grpc::string& value);
void AddTrailingMetadata(const grpc::string& key, const grpc::string& value); void AddTrailingMetadata(const grpc::string& key, const grpc::string& value);
bool IsCancelled(); bool IsCancelled() const;
const std::multimap<grpc::string, grpc::string>& client_metadata() { const std::multimap<grpc::string, grpc::string>& client_metadata() {
return client_metadata_; return client_metadata_;
......
...@@ -144,7 +144,7 @@ void ServerContext::AddTrailingMetadata(const grpc::string& key, ...@@ -144,7 +144,7 @@ void ServerContext::AddTrailingMetadata(const grpc::string& key,
trailing_metadata_.insert(std::make_pair(key, value)); trailing_metadata_.insert(std::make_pair(key, value));
} }
bool ServerContext::IsCancelled() { bool ServerContext::IsCancelled() const {
return completion_op_ && completion_op_->CheckCancelled(cq_); return completion_op_ && completion_op_->CheckCancelled(cq_);
} }
......
...@@ -62,5 +62,14 @@ InteropContextInspector::InteropContextInspector( ...@@ -62,5 +62,14 @@ InteropContextInspector::InteropContextInspector(
const ::grpc::ServerContext& context) const ::grpc::ServerContext& context)
: context_(context) {} : context_(context) {}
std::shared_ptr<const AuthContext> InteropContextInspector::GetAuthContext()
const {
return context_.auth_context();
}
bool InteropContextInspector::IsCancelled() const {
return context_.IsCancelled();
}
} // namespace testing } // namespace testing
} // namespace grpc } // namespace grpc
...@@ -48,7 +48,9 @@ class InteropContextInspector { ...@@ -48,7 +48,9 @@ class InteropContextInspector {
public: public:
InteropContextInspector(const ::grpc::ServerContext& context); InteropContextInspector(const ::grpc::ServerContext& context);
// Inspector methods, able to peek inside ServerContext go here. // Inspector methods, able to peek inside ServerContext, follow.
std::shared_ptr<const AuthContext> GetAuthContext() const;
bool IsCancelled() const;
private: private:
const ::grpc::ServerContext& context_; const ::grpc::ServerContext& context_;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment