add request timeout #78

dtm-labs · May 14, 2024 · ba66e11 · ba66e11
1 parent 585d2aa
commit ba66e11
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 1 deletion.
diff --git a/src/DtmCommon/Imp/TransBase.cs b/src/DtmCommon/Imp/TransBase.cs
@@ -19,6 +19,8 @@ public class TransBase
 
         [JsonPropertyName("timeout_to_fail")]
         public long TimeoutToFail { get; set; }
+        [JsonPropertyName("request_timeout")]
+        public long RequestTimeout { get; set; }
 
         [JsonPropertyName("retry_interval")]
         public long RetryInterval { get; set; }

diff --git a/src/Dtmcli/Saga/Saga.cs b/src/Dtmcli/Saga/Saga.cs
@@ -70,7 +70,16 @@ public Saga SetTimeoutToFail(long timeoutToFail)
             this._transBase.TimeoutToFail = timeoutToFail;
             return this;
         }
-
+        /// <summary>
+        /// Set request timeout
+        /// </summary>
+        /// <param name="requestTimeout">request timeout</param>
+        /// <returns></returns>
+        public Saga SetRequestTimeout(long requestTimeout)
+        {
+            this._transBase.RequestTimeout = requestTimeout;
+            return this;
+        }
         /// <summary>
         /// Set retry interval for trans, unit is second
         /// </summary>

diff --git a/src/Dtmgrpc/DtmGImp/Utils.cs b/src/Dtmgrpc/DtmGImp/Utils.cs
@@ -149,6 +149,7 @@ public static dtmgpb.DtmRequest BuildDtmRequest(TransBase transBase)
                 TimeoutToFail = transBase.TimeoutToFail,
                 RetryInterval = transBase.RetryInterval,
                 RetryLimit = transBase.RetryLimit,
+                RequestTimeout = transBase.RequestTimeout,
             };
 
             if (transBase.BranchHeaders != null)

diff --git a/src/Dtmgrpc/Saga/SagaGrpc.cs b/src/Dtmgrpc/Saga/SagaGrpc.cs
@@ -74,6 +74,16 @@ public SagaGrpc SetTimeoutToFail(long timeoutToFail)
             this._transBase.TimeoutToFail = timeoutToFail;
             return this;
         }
+        /// <summary>
+        /// Set request timeout
+        /// </summary>
+        /// <param name="requestTimeout">request timeout</param>
+        /// <returns></returns>
+        public SagaGrpc SetRequestTimeout(long requestTimeout)
+        {
+            this._transBase.RequestTimeout = requestTimeout;
+            return this;
+        }
 
         /// <summary>
         /// Set retry interval for trans, unit is second