fix: initial delay should take care of shutdown signal (#7)

Signed-off-by: tison <[email protected]>
fast · Dec 11, 2024 · e871589 · e871589
1 parent 85671fc
commit e871589
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/fastimer/src/schedule/arbitrary.rs b/fastimer/src/schedule/arbitrary.rs
@@ -66,7 +66,12 @@ pub trait ArbitraryDelayActionExt: ArbitraryDelayAction {
 
             if let Some(initial_delay) = initial_delay {
                 if initial_delay > Duration::ZERO {
-                    make_delay.delay(initial_delay).await;
+                    let is_shutdown = self.is_shutdown();
+                    let delay = make_delay.delay(initial_delay);
+                    if let Either::Left(()) = select(is_shutdown, delay).await {
+                        self.teardown();
+                        return;
+                    }
                 }
             }
 

diff --git a/fastimer/src/schedule/simple.rs b/fastimer/src/schedule/simple.rs
@@ -76,7 +76,12 @@ pub trait SimpleActionExt: SimpleAction {
 
             if let Some(initial_delay) = initial_delay {
                 if initial_delay > Duration::ZERO {
-                    make_delay.delay(initial_delay).await;
+                    let is_shutdown = self.is_shutdown();
+                    let delay = make_delay.delay(initial_delay);
+                    if let Either::Left(()) = select(is_shutdown, delay).await {
+                        self.teardown();
+                        return;
+                    };
                 }
             }
 
@@ -159,7 +164,12 @@ pub trait SimpleActionExt: SimpleAction {
             if let Some(initial_delay) = initial_delay {
                 if initial_delay > Duration::ZERO {
                     next = make_instant_from_now(initial_delay);
-                    make_delay.delay_util(next).await;
+                    let is_shutdown = self.is_shutdown();
+                    let delay = make_delay.delay_util(next);
+                    if let Either::Left(()) = select(is_shutdown, delay).await {
+                        self.teardown();
+                        return;
+                    };
                 }
             }