feat: Add overload on AddAtcRestClient with a IContractSerializer

atc-net · Jul 14, 2024 · 656980d · 656980d
1 parent 05afd47
commit 656980d
Showing 1 changed file with 21 additions and 5 deletions.
diff --git a/src/Atc.Rest.Client/Options/ServiceCollectionExtensions.cs b/src/Atc.Rest.Client/Options/ServiceCollectionExtensions.cs
@@ -7,7 +7,8 @@ public static IServiceCollection AddAtcRestClient<TOptions>(
         this IServiceCollection services,
         string clientName,
         TOptions options,
-        Action<IHttpClientBuilder>? httpClientBuilder = default)
+        Action<IHttpClientBuilder>? httpClientBuilder = default,
+        IContractSerializer? contractSerializer = null)
         where TOptions : AtcRestClientOptions, new()
     {
         services.AddSingleton(options);
@@ -23,7 +24,14 @@ public static IServiceCollection AddAtcRestClient<TOptions>(
 
         // Register utilities
         services.AddSingleton<IHttpMessageFactory, HttpMessageFactory>();
-        services.AddSingleton<IContractSerializer, DefaultJsonContractSerializer>();
+        if (contractSerializer is null)
+        {
+            services.AddSingleton<IContractSerializer, DefaultJsonContractSerializer>();
+        }
+        else
+        {
+            services.AddSingleton(contractSerializer);
+        }
 
         return services;
     }
@@ -34,9 +42,10 @@ public static IServiceCollection AddAtcRestClient(
         string clientName,
         Uri baseAddress,
         TimeSpan timeout,
-        Action<IHttpClientBuilder>? httpClientBuilder = default)
+        Action<IHttpClientBuilder>? httpClientBuilder = default,
+        IContractSerializer? contractSerializer = null)
     {
-        var clientBuilder = services.AddHttpClient(clientName, (s, c) =>
+        var clientBuilder = services.AddHttpClient(clientName, (_, c) =>
         {
             c.BaseAddress = baseAddress;
             c.Timeout = timeout;
@@ -46,7 +55,14 @@ public static IServiceCollection AddAtcRestClient(
 
         // Register utilities
         services.AddSingleton<IHttpMessageFactory, HttpMessageFactory>();
-        services.AddSingleton<IContractSerializer, DefaultJsonContractSerializer>();
+        if (contractSerializer is null)
+        {
+            services.AddSingleton<IContractSerializer, DefaultJsonContractSerializer>();
+        }
+        else
+        {
+            services.AddSingleton(contractSerializer);
+        }
 
         return services;
     }