added periodic stopwatch return

kougen · Mar 25, 2024 · fe4f67c · fe4f67c
1 parent 749df72
commit fe4f67c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/_src/Implementation/Time/DefaultStopwatch.cs b/_src/Implementation/Time/DefaultStopwatch.cs
@@ -105,11 +105,17 @@ public void Reset()
             }
         }
 
-        public void PeriodicOperation(int periodInMilliseconds, ITickListener listener, CancellationToken cancellationToken)
+        public IPeriodicStopwatch PeriodicOperation(int periodInMilliseconds, ITickListener listener, CancellationToken cancellationToken)
         {
             var periodicStopwatch = new PeriodicStopwatch(this, periodInMilliseconds, listener, cancellationToken);
             _periodicStopwatches.Add(periodicStopwatch);
             periodicStopwatch.Start();
+            return periodicStopwatch;
+        }
+
+        public IPeriodicStopwatch PeriodicOperation(int periodInMilliseconds, ITickListener listener)
+        {
+            return PeriodicOperation(periodInMilliseconds, listener, _cancellationToken);
         }
 
         public IPeriodicStopwatchFactory GetPeriodicStopwatchFactory()

diff --git a/_src/Infrastructure/Time/IStopwatch.cs b/_src/Infrastructure/Time/IStopwatch.cs
@@ -10,7 +10,8 @@ public interface IStopwatch : IAdjustableStopwatch, IDisposable
     {
         void Wait(int periodInMilliseconds, ITickListener listener);
         Task WaitAsync(int periodInMilliseconds, ITickListener listener);
-        void PeriodicOperation(int periodInMilliseconds, ITickListener listener, CancellationToken cancellationToken);
+        IPeriodicStopwatch PeriodicOperation(int periodInMilliseconds, ITickListener listener, CancellationToken cancellationToken);
+        IPeriodicStopwatch PeriodicOperation(int periodInMilliseconds, ITickListener listener);
         IPeriodicStopwatchFactory GetPeriodicStopwatchFactory();
         void RegisterStopwatch(IPeriodicStopwatch stopwatch);
         void UnregisterStopwatch(IPeriodicStopwatch stopwatch);