Merge pull request #17179 from yashykt/failhijackedrecv

Add interceptor methods to fail recv msg for hijacked rpcs and set recv message to nullptr on failure

1 files changed, 7 insertions, 2 deletions

diff --git a/include/grpcpp/impl/codegen/interceptor.h b/include/grpcpp/impl/codegen/interceptor.h
index 519c65c717..a57a3fccbb 100644
--- a/include/grpcpp/impl/codegen/interceptor.h
+++ b/include/grpcpp/impl/codegen/interceptor.h
@@ -168,8 +168,13 @@ class InterceptorBatchMethods {
   /// list.
   virtual std::unique_ptr<ChannelInterface> GetInterceptedChannel() = 0;
 
-  // On a hijacked RPC/ to-be hijacked RPC, this can be called to fail a SEND
-  // MESSAGE op
+  /// On a hijacked RPC, an interceptor can decide to fail a PRE_RECV_MESSAGE
+  /// op. This would be a signal to the reader that there will be no more
+  /// messages, or the stream has failed or been cancelled.
+  virtual void FailHijackedRecvMessage() = 0;
+
+  /// On a hijacked RPC/ to-be hijacked RPC, this can be called to fail a SEND
+  /// MESSAGE op
   virtual void FailHijackedSendMessage() = 0;
 };