Merge pull request #29 from stulzq/main

add http request timeout option #21

Author: catcherwong

src/Dtmcli/DtmOptions.cs

@@ -1,11 +1,23 @@
-namespace Dtmcli
+using System;
+
+namespace Dtmcli
 {
     public class DtmOptions
     {
         public string DtmUrl { get; set; }
 
+        /// <summary>
+        /// dtm server http request timeout in milliseconds, default 100,000 milliseconds(100s)
+        /// </summary>
+        public int DtmHttpTimeout { get; set; } = 100 * 1000;
+
+        /// <summary>
+        /// branch http request timeout in milliseconds, default 100,000 milliseconds(100s)
+        /// </summary>
+        public int BranchHttpTimeout { get; set; } = 100 * 1000;
+
         public string DBType { get; set; } = "mysql";
 
         public string BarrierTableName { get; set; } = "dtm_barrier.barrier";
     }
-}
+}

src/Dtmcli/ServiceCollectionExtensions.cs

@@ -16,31 +16,40 @@ public static IServiceCollection AddDtmcli(this IServiceCollection services, Act
             services.AddOptions();
             services.Configure(setupAction);
 
-            return AddDtmcliCore(services);
+            var op = new DtmOptions();
+            setupAction(op);
+
+            return AddDtmcliCore(services, op);
         }
 
         public static IServiceCollection AddDtmcli(this IServiceCollection services, IConfiguration configuration, string sectionName = "dtm")
         {
             services.Configure<DtmOptions>(configuration.GetSection(sectionName));
+            
+            var op=configuration.GetSection(sectionName).Get<DtmOptions>() ?? new DtmOptions();
 
-            return AddDtmcliCore(services);
+            return AddDtmcliCore(services, op);
         }
 
-        private static IServiceCollection AddDtmcliCore(IServiceCollection services)
+        private static IServiceCollection AddDtmcliCore(IServiceCollection services,DtmOptions options)
         {
-            AddHttpClient(services);
+            AddHttpClient(services, options);
             AddDtmCore(services);
 
             return services;
         }
 
-        private static void AddHttpClient(IServiceCollection services)
+        private static void AddHttpClient(IServiceCollection services,DtmOptions options)
         {
             services.AddHttpClient(Constant.DtmClientHttpName, client =>
             {
                 client.DefaultRequestHeaders.Add("Accept", "application/json");
+                client.Timeout = TimeSpan.FromMilliseconds(options.DtmHttpTimeout);
+            });
+            services.AddHttpClient(Constant.BranchClientHttpName, client =>
+            {
+                client.Timeout = TimeSpan.FromMilliseconds(options.BranchHttpTimeout);
             });
-            services.AddHttpClient(Constant.BranchClientHttpName);
         }
 
         private static void AddDtmCore(IServiceCollection services)