chore: add WithClientTimeout to TopicConfiguration

momentohq · Sep 14, 2023 · 036296b · 036296b
1 parent 52a4310
commit 036296b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/Momento.Sdk/Config/ITopicConfiguration.cs b/src/Momento.Sdk/Config/ITopicConfiguration.cs
@@ -1,3 +1,4 @@
+using System;
 using Microsoft.Extensions.Logging;
 using Momento.Sdk.Config.Transport;
 
@@ -20,4 +21,12 @@ public interface ITopicConfiguration
     /// <param name="transportStrategy">This is responsible for configuring network tunables.</param>
     /// <returns>Configuration object with custom transport strategy provided</returns>
     public ITopicConfiguration WithTransportStrategy(ITopicTransportStrategy transportStrategy);
+
+    /// <summary>
+    /// Add the specified client timeout to an existing instance of TopicConfiguration object as an addiion to
+    /// the existing transport strategy.
+    /// </summary>
+    /// <param name="clientTimeout">The amount of time to wait before cancelling the request.</param>
+    /// <returns>TopicConfiguration object with client timeout provided</returns>
+    public ITopicConfiguration WithClientTimeout(TimeSpan clientTimeout);
 }
diff --git a/src/Momento.Sdk/Config/TopicConfiguration.cs b/src/Momento.Sdk/Config/TopicConfiguration.cs
@@ -1,3 +1,4 @@
+using System;
 using Microsoft.Extensions.Logging;
 using Momento.Sdk.Config.Transport;
 
@@ -29,6 +30,15 @@ public ITopicConfiguration WithTransportStrategy(ITopicTransportStrategy transpo
         return new TopicConfiguration(LoggerFactory, transportStrategy);
     }
 
+    /// <inheritdoc/>
+    public ITopicConfiguration WithClientTimeout(TimeSpan clientTimeout)
+    {
+        return new TopicConfiguration(
+            loggerFactory: LoggerFactory,
+            transportStrategy: TransportStrategy.WithClientTimeout(clientTimeout)
+        );
+    }
+
     /// <inheritdoc />
     public override bool Equals(object obj)
     {