propagate the McThriftContext wrapper for thriftHandlerCallback consi…

…stently Reviewed By: alikhtarov Differential Revision: D49243311 fbshipit-source-id: d170455b0cd818253b8e158b44f6fa5cc7b8ed2a
facebook · Sep 18, 2023 · 819b227 · 819b227
1 parent 565f39e
commit 819b227
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mcrouter/lib/network/McThriftContext.h b/mcrouter/lib/network/McThriftContext.h
@@ -74,6 +74,10 @@ class McThriftContext {
     return underlying_->getConnectionContext()->getHeader();
   }
 
+  apache::thrift::HandlerCallback<Reply>& getThriftHandlerCallback() {
+    return *underlying_;
+  }
+
  private:
   std::unique_ptr<apache::thrift::HandlerCallback<Reply>> underlying_;
 };