SERVER-33299 Add support for timeouts to TransportLayerASIO Sessions

1 files changed, 10 insertions, 0 deletions

diff --git a/src/mongo/transport/session.h b/src/mongo/transport/session.h
index d25d32b9027..db80aa0814d 100644
--- a/src/mongo/transport/session.h
+++ b/src/mongo/transport/session.h
@@ -112,6 +112,16 @@ public:
     virtual Status sinkMessage(Message message) = 0;
     virtual void asyncSinkMessage(Message message, std::function<void(Status)> cb) = 0;
 
+    /**
+    * This should only be used to detect when the remote host has disappeared without
+    * notice. It does NOT work correctly for ensuring that operations complete or fail
+    * by some deadline.
+    *
+    * This timeout will only effect calls sourceMessage()/sinkMessage(). Async operations do not
+    * currently support timeouts.
+    */
+    virtual void setTimeout(boost::optional<Milliseconds> timeout) = 0;
+
     virtual const HostAndPort& remote() const = 0;
     virtual const HostAndPort& local() const = 0;