Option to not set http client in AddCogniteClient

cognitedata · Jun 10, 2020 · 0d95454 · 0d95454
1 parent 7edc0c5
commit 0d95454
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/ExtractorUtils/Cognite/CogniteUtils.cs b/ExtractorUtils/Cognite/CogniteUtils.cs
@@ -260,11 +260,21 @@ static IAsyncPolicy<HttpResponseMessage> GetTimeoutPolicy()
         /// Cognite API (enabled in debug mode)</param>
         /// <param name="setMetrics">If true, a <see cref="IMetrics"/> metrics collector is created and used by the client
         /// to report metrics on the number and duration of API requests</param>
-        public static void AddCogniteClient(this IServiceCollection services, string appId, bool setLogger = false, bool setMetrics = false)
+        /// <param name="setHttpClient">Default true. If false <see cref="Client.Builder"/> is not added to the
+        /// <see cref="ServiceCollection"/>. If this is false it must be added before this method is called.</param>
+        public static void AddCogniteClient(this IServiceCollection services,
+                                            string appId,
+                                            bool setLogger = false,
+                                            bool setMetrics = false,
+                                            bool setHttpClient = true)
         {
-            services.AddHttpClient<Client.Builder>(c => c.Timeout = Timeout.InfiniteTimeSpan)
-                .AddPolicyHandler((provider, message) => { return GetRetryPolicy(provider.GetRequiredService<ILogger<Client>>()); })
-                .AddPolicyHandler(GetTimeoutPolicy());
+            if (setHttpClient)
+            {
+                services.AddHttpClient<Client.Builder>(c => c.Timeout = Timeout.InfiniteTimeSpan)
+                    .AddPolicyHandler((provider, message) => { return GetRetryPolicy(provider.GetRequiredService<ILogger<Client>>()); })
+                    .AddPolicyHandler(GetTimeoutPolicy());
+            }
+
             services.AddHttpClient<Authenticator>();
             services.AddSingleton<IMetrics, CdfMetricCollector>();
             services.AddTransient(provider => {