|
3 | 3 | using DotPulsar.Internal; |
4 | 4 | using Polly; |
5 | 5 | using Polly.Retry; |
| 6 | +using Polly.Timeout; |
6 | 7 |
|
7 | 8 | namespace DotPulsar.Extensions; |
8 | 9 |
|
@@ -37,7 +38,19 @@ public static ValueTask Process<TMessage>( |
37 | 38 | CancellationToken cancellationToken = default) |
38 | 39 | => Process(consumer, processor, resiliencePipeline, new ProcessingOptions(), failureHandler, cancellationToken); |
39 | 40 |
|
40 | | - public static ResiliencePipelineBuilder AddRetryProducer(this ResiliencePipelineBuilder pipelineBuilder, Action<RetryStrategyOptions>? configure = null) { |
| 41 | + public static ResiliencePipelineBuilder AddResilientProducer(this ResiliencePipelineBuilder pipelineBuilder, Action<RetryStrategyOptions>? configureRetry = null, Action<TimeoutStrategyOptions>? configureTimeout = null) { |
| 42 | + return pipelineBuilder.AddProducerRetry(configureRetry).AddProducerTimeout(configureTimeout); |
| 43 | + } |
| 44 | + |
| 45 | + public static ResiliencePipelineBuilder AddProducerTimeout(this ResiliencePipelineBuilder pipelineBuilder, Action<TimeoutStrategyOptions>? configure = null) { |
| 46 | + var options = new TimeoutStrategyOptions { |
| 47 | + Timeout = TimeSpan.FromSeconds(30) |
| 48 | + }; |
| 49 | + configure?.Invoke(options); |
| 50 | + return pipelineBuilder.AddTimeout(options); |
| 51 | + } |
| 52 | + |
| 53 | + public static ResiliencePipelineBuilder AddProducerRetry(this ResiliencePipelineBuilder pipelineBuilder, Action<RetryStrategyOptions>? configure = null) { |
41 | 54 | var options = new RetryStrategyOptions { |
42 | 55 | MaxRetryAttempts = 10, |
43 | 56 | Delay = TimeSpan.FromMilliseconds(100), |
@@ -76,7 +89,7 @@ public static IProducer<TMessage> CreateResilient<TMessage>(this IProducerBuilde |
76 | 89 | if (configurePipelineBuilder != null) { |
77 | 90 | configurePipelineBuilder(pipelineBuilder); |
78 | 91 | } else { |
79 | | - pipelineBuilder.AddRetryProducer(); |
| 92 | + pipelineBuilder.AddResilientProducer(); |
80 | 93 | } |
81 | 94 |
|
82 | 95 | return CreateResilient(producerBuilder, pipelineBuilder.Build()); |
|
0 commit comments