feat: add interface registration

VirtoCommerce · May 6, 2024 · fd86907 · fd86907
1 parent 7a7da96
commit fd86907
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 3 deletions.
diff --git a/src/VirtoCommerce.ElasticAppSearch.Core/Services/IElasticAppSearchRetryPolicySelector.cs b/src/VirtoCommerce.ElasticAppSearch.Core/Services/IElasticAppSearchRetryPolicySelector.cs
@@ -0,0 +1,9 @@
+using System.Net.Http;
+using Polly.Retry;
+
+namespace VirtoCommerce.ElasticAppSearch.Core.Services;
+
+public interface IElasticAppSearchRetryPolicySelector
+{
+    AsyncRetryPolicy<HttpResponseMessage> GetRetryPolicy();
+}
diff --git a/src/VirtoCommerce.ElasticAppSearch.Core/VirtoCommerce.ElasticAppSearch.Core.csproj b/src/VirtoCommerce.ElasticAppSearch.Core/VirtoCommerce.ElasticAppSearch.Core.csproj
@@ -9,5 +9,6 @@
   <ItemGroup>
     <PackageReference Include="Microsoft.SourceLink.GitHub" Version="8.0.0" PrivateAssets="All" />
     <PackageReference Include="VirtoCommerce.SearchModule.Core" Version="3.800.0" />
+    <PackageReference Include="Microsoft.Extensions.Http.Polly" Version="8.0.0" />
   </ItemGroup>
 </Project>
diff --git a/src/VirtoCommerce.ElasticAppSearch.Data/Services/ElasticAppSearchPolicySelector.cs b/src/VirtoCommerce.ElasticAppSearch.Data/Services/ElasticAppSearchPolicySelector.cs
@@ -6,10 +6,11 @@
 using Polly.Extensions.Http;
 using Polly.Retry;
 using VirtoCommerce.ElasticAppSearch.Core.Models;
+using VirtoCommerce.ElasticAppSearch.Core.Services;
 
 namespace VirtoCommerce.ElasticAppSearch.Data.Services;
 
-public class ElasticAppSearchPolicySelector
+public class ElasticAppSearchPolicySelector : IElasticAppSearchRetryPolicySelector
 {
     private readonly ElasticAppSearchOptions _options;
     private readonly ILogger<ElasticAppSearchApiClient> _logger;

diff --git a/src/VirtoCommerce.ElasticAppSearch.Web/Module.cs b/src/VirtoCommerce.ElasticAppSearch.Web/Module.cs
@@ -43,7 +43,7 @@ public void Initialize(IServiceCollection serviceCollection)
             serviceCollection.AddSingleton<ElasticAppSearchProvider>();
             serviceCollection.AddSingleton<ISearchFacetsQueryBuilder, SearchFacetsQueryBuilder>();
             serviceCollection.AddSingleton<IAggregationsResponseBuilder, AggregationsResponseBuilder>();
-            serviceCollection.AddSingleton<ElasticAppSearchPolicySelector>();
+            serviceCollection.AddSingleton<IElasticAppSearchRetryPolicySelector, ElasticAppSearchPolicySelector>();
 
             serviceCollection.AddHttpClient(ModuleConstants.ModuleName, (serviceProvider, httpClient) =>
                 {
@@ -71,7 +71,7 @@ public void Initialize(IServiceCollection serviceCollection)
                 })
                 .AddPolicyHandler((serviceProvider, _) =>
                 {
-                    var policySelector = serviceProvider.GetRequiredService<ElasticAppSearchPolicySelector>();
+                    var policySelector = serviceProvider.GetRequiredService<IElasticAppSearchRetryPolicySelector>();
 
                     return policySelector.GetRetryPolicy();
                 });