From 69ef0e75c605a2e208860685abb28f60d63cc033 Mon Sep 17 00:00:00 2001
From: nadia <nadia__@bk.ru>
Date: Tue, 21 May 2024 15:28:40 +0500
Subject: [PATCH] fix: get generic grpc token\metadata provider

---
 src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs      | 4 ++--
 src/Sitko.Core.Grpc.Client/IGrpcMetadataProviderFactory.cs | 7 +++++--
 src/Sitko.Core.Grpc.Client/IGrpcTokenProviderFactory.cs    | 7 +++++--
 3 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs b/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs
index f6097e1e8..5116b4b76 100644
--- a/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs
+++ b/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs
@@ -80,7 +80,7 @@ public GrpcClientModuleOptions<TClient> AddMetadataProvider<TMetadataProvider>()
         configureServicesActions.Add(services =>
         {
             services.AddTransient<IGrpcMetadataProviderFactory<TClient>, GrpcMetadataProviderFactory<TClient, TMetadataProvider>>();
-            services.TryAddTransient<IGrpcMetadataProvider, TMetadataProvider>();
+            services.AddTransient<IGrpcMetadataProvider, TMetadataProvider>();
         });
         return this;
     }
@@ -91,7 +91,7 @@ public GrpcClientModuleOptions<TClient> AddTokenAuth<TTokenProvider>()
         configureServicesActions.Add(services =>
         {
             services.AddTransient<IGrpcTokenProviderFactory<TClient>, GrpcTokenProviderFactory<TClient, TTokenProvider>>();
-            services.TryAddTransient<IGrpcTokenProvider, TTokenProvider>();
+            services.AddTransient<IGrpcTokenProvider, TTokenProvider>();
         });
         return this;
     }
diff --git a/src/Sitko.Core.Grpc.Client/IGrpcMetadataProviderFactory.cs b/src/Sitko.Core.Grpc.Client/IGrpcMetadataProviderFactory.cs
index ca11839a8..281781fb6 100644
--- a/src/Sitko.Core.Grpc.Client/IGrpcMetadataProviderFactory.cs
+++ b/src/Sitko.Core.Grpc.Client/IGrpcMetadataProviderFactory.cs
@@ -8,9 +8,12 @@ internal interface IGrpcMetadataProviderFactory<TClient> where TClient : ClientB
     public IGrpcMetadataProvider GetProvider(IServiceProvider serviceProvider);
 }
 
-internal class GrpcMetadataProviderFactory<TClient, TMetadataProvider> : IGrpcMetadataProviderFactory<TClient> where TClient : ClientBase<TClient>
+internal class GrpcMetadataProviderFactory<TClient, TMetadataProvider> : IGrpcMetadataProviderFactory<TClient>
+    where TClient : ClientBase<TClient>
     where TMetadataProvider : class, IGrpcMetadataProvider
 
 {
-    public IGrpcMetadataProvider GetProvider(IServiceProvider serviceProvider) => serviceProvider.GetService<TMetadataProvider>()!;
+    public IGrpcMetadataProvider GetProvider(IServiceProvider serviceProvider) =>
+        serviceProvider.GetServices<IGrpcMetadataProvider>().OfType<TMetadataProvider>().FirstOrDefault() ??
+        throw new InvalidDataException($"Service {typeof(TMetadataProvider)} is not register");
 }
diff --git a/src/Sitko.Core.Grpc.Client/IGrpcTokenProviderFactory.cs b/src/Sitko.Core.Grpc.Client/IGrpcTokenProviderFactory.cs
index d4d8bf508..a5842c0b6 100644
--- a/src/Sitko.Core.Grpc.Client/IGrpcTokenProviderFactory.cs
+++ b/src/Sitko.Core.Grpc.Client/IGrpcTokenProviderFactory.cs
@@ -8,9 +8,12 @@ public interface IGrpcTokenProviderFactory<TClient> where TClient : ClientBase<T
     public IGrpcTokenProvider GetProvider(IServiceProvider serviceProvider);
 }
 
-internal class GrpcTokenProviderFactory<TClient, TTokenProvider> : IGrpcTokenProviderFactory<TClient> where TClient : ClientBase<TClient>
+internal class GrpcTokenProviderFactory<TClient, TTokenProvider> : IGrpcTokenProviderFactory<TClient>
+    where TClient : ClientBase<TClient>
     where TTokenProvider : class, IGrpcTokenProvider
 
 {
-    public IGrpcTokenProvider GetProvider(IServiceProvider serviceProvider) => serviceProvider.GetService<TTokenProvider>()!;
+    public IGrpcTokenProvider GetProvider(IServiceProvider serviceProvider) =>
+        serviceProvider.GetServices<IGrpcTokenProvider>().OfType<TTokenProvider>().FirstOrDefault() ??
+        throw new InvalidDataException($"Service {typeof(TTokenProvider)} is not register");
 }