Update internal usage of the CircuitBreaker

akkadotnet · Mar 23, 2020 · bfa8ebb · bfa8ebb
1 parent 7ba587d
commit bfa8ebb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/src/contrib/persistence/Akka.Persistence.Sql.Common/Journal/BatchingSqlJournal.cs b/src/contrib/persistence/Akka.Persistence.Sql.Common/Journal/BatchingSqlJournal.cs
@@ -558,6 +558,7 @@ protected BatchingSqlJournal(BatchingSqlJournalSetup setup)
             _serialization = Context.System.Serialization;
             Log = Context.GetLogger();
             _circuitBreaker = CircuitBreaker.Create(
+                Context.System.Scheduler,
                 maxFailures: Setup.CircuitBreakerSettings.MaxFailures,
                 callTimeout: Setup.CircuitBreakerSettings.CallTimeout,
                 resetTimeout: Setup.CircuitBreakerSettings.ResetTimeout);

diff --git a/src/core/Akka.Persistence/Journal/AsyncWriteJournal.cs b/src/core/Akka.Persistence/Journal/AsyncWriteJournal.cs
@@ -54,6 +54,7 @@ protected AsyncWriteJournal()
             CanPublish = extension.Settings.Internal.PublishPluginCommands;
             var config = extension.ConfigFor(Self);
             _breaker = new CircuitBreaker(
+                Context.System.Scheduler,
                 config.GetInt("circuit-breaker.max-failures", 0),
                 config.GetTimeSpan("circuit-breaker.call-timeout", null),
                 config.GetTimeSpan("circuit-breaker.reset-timeout", null));

diff --git a/src/core/Akka.Persistence/Snapshot/SnapshotStore.cs b/src/core/Akka.Persistence/Snapshot/SnapshotStore.cs
@@ -38,6 +38,7 @@ protected SnapshotStore()
             _publish = extension.Settings.Internal.PublishPluginCommands;
             var config = extension.ConfigFor(Self);
             _breaker = CircuitBreaker.Create(
+                Context.System.Scheduler,
                 config.GetInt("circuit-breaker.max-failures", 0),
                 config.GetTimeSpan("circuit-breaker.call-timeout", null),
                 config.GetTimeSpan("circuit-breaker.reset-timeout", null));