From b7cd7952c890f585ff876c622482534e5d58d4a4 Mon Sep 17 00:00:00 2001
From: sodo <djdisodo@gmail.com>
Date: Mon, 1 Jan 2024 21:18:30 +0900
Subject: avr: support sleep

---
 embassy-executor/src/arch/avr.rs | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'embassy-executor/src')

diff --git a/embassy-executor/src/arch/avr.rs b/embassy-executor/src/arch/avr.rs
index 2c715f297..fa6afe762 100644
--- a/embassy-executor/src/arch/avr.rs
+++ b/embassy-executor/src/arch/avr.rs
@@ -8,11 +8,16 @@ mod thread {
     use core::marker::PhantomData;
 
     pub use embassy_executor_macros::main_avr as main;
+    use portable_atomic::{AtomicBool, Ordering};
 
     use crate::{raw, Spawner};
 
+    static SIGNAL_WORK_THREAD_MODE: AtomicBool = AtomicBool::new(false);
+
     #[export_name = "__pender"]
-    fn __pender(_context: *mut ()) {}
+    fn __pender(_context: *mut ()) {
+        SIGNAL_WORK_THREAD_MODE.store(true, Ordering::SeqCst);
+    }
 
     /// avr Executor
     pub struct Executor {
@@ -52,7 +57,11 @@ mod thread {
 
             loop {
                 unsafe {
-                    self.inner.poll();
+                    if SIGNAL_WORK_THREAD_MODE.swap(false, Ordering::SeqCst) {
+                        self.inner.poll();
+                    } else {
+                        avr_device::asm::sleep();
+                    }
                 }
             }
         }
-- 
cgit