feat: 使用线程处理 MQTT 发送。

src/main.rs

@@ -3,9 +3,11 @@ use esp_idf_svc::eventloop::{Background, EspBackgroundEventLoop, EspEventLoop};
 use esp_idf_sys::{self as _};
 use log::*;
 use std::{
+    borrow::Borrow,
     env,
+    sync::{Arc, Mutex},
     thread::{self, sleep},
-    time::Duration, borrow::Borrow,
+    time::Duration,
 };
 
 mod beep;
@@ -19,7 +21,7 @@ mod wifi;
 use crate::{
     beep::{ringtone, Beep},
     dc_out_controller::{DcOutController, DcOutControllerState, DC_OUT_STATE_EVENT_LOOP},
-    message_queue::{MqDto, MQ_EVENTLOOP},
+    message_queue::MqDto,
     voltage_detection::{VoltageDetectionWorker, VOLTAGE_EVENTLOOP},
 };
 use crate::{
@@ -41,13 +43,6 @@ fn main() {
     let peripherals = esp_idf_hal::peripherals::Peripherals::take().unwrap();
 
     let blink_pin = peripherals.pins.gpio5;
-    let beep_pin = peripherals.pins.gpio6;
-    let ledc_timer0 = peripherals.ledc.timer0;
-    let ledc_channel0 = peripherals.ledc.channel0;
-    let dc_out_ctl_pin = peripherals.pins.gpio3;
-    let i2c0 = peripherals.i2c0;
-    let sda_pin = peripherals.pins.gpio4;
-    let scl_pin = peripherals.pins.gpio10;
 
     info!("About to start a background event loop");
     match EspBackgroundEventLoop::new(&Default::default()) {
@@ -87,51 +82,48 @@ fn main() {
 
     sleep(Duration::from_millis(100));
 
-    let mut mq = MessageQueue::new();
+    let mut _mq = MessageQueue::new();
-    mq.init().unwrap();
 
     let _mq_subscription;
-    match mq.watch() {
+    match _mq.watch() {
         Err(err) => {
             error!("Can not watch MessageQueue. {}", err);
         }
         Ok(subscription) => _mq_subscription = subscription,
     }
 
+    let _mq_tx_for_voltage = _mq.tx.clone();
+    let _mq_tx_for_dc_out_state = _mq.tx.clone();
+
     let _voltage_subscription;
     if let Some(voltage_event_loop) = unsafe { VOLTAGE_EVENTLOOP.as_mut() } {
-        if let Some(mq_event_loop) = unsafe { MQ_EVENTLOOP.as_mut() } {
         _voltage_subscription = voltage_event_loop
             .subscribe(move |message: &VoltageDetectionWorker| {
                 if let Ok(json_str) = serde_json::to_string(&message) {
-                        let result = mq_event_loop.post(
+                    match _mq_tx_for_voltage.lock() {
-                            &MqDto {
+                        Ok(tx) => {
-                                topic: "voltage",
+                            let result = tx.send(MqDto {
-                                message: json_str.as_bytes(),
+                                topic: "voltage".to_string(),
-                            },
+                                message: json_str,
-                            None,
+                            });
-                        );
                             match result {
-                            Ok(success) => {
+                                Ok(()) => {
-                                if !success {
+                                    warn!("send voltage to mq message failed.")
-                                    warn!("post voltage to mq event failed.")
+                                }
+                                Err(err) => warn!("send voltage to mq message failed. {}", err),
                             }
                         }
-                            Err(err) => warn!("post voltage to mq event failed. {}", err),
+                        Err(err) => warn!("send voltage to mq message failed. {}", err),
                     }
                 }
             })
             .expect(" Listening Event Loop Failed");
-        } else {
-            panic!("MQ_EVENTLOOP is undefined!");
-        }
     } else {
         panic!("VOLTAGE_EVENTLOOP is undefined!");
     }
 
     let _dc_out_state_subscription;
     if let Some(dc_state_event_loop) = unsafe { DC_OUT_STATE_EVENT_LOOP.as_mut() } {
-        if let Some(mq_event_loop) = unsafe { MQ_EVENTLOOP.as_mut() } {
         _dc_out_state_subscription = dc_state_event_loop
             .subscribe(move |message: &DcOutControllerState| {
                 info!("Event Loop Value");
@@ -147,26 +139,24 @@ fn main() {
                     }
                     _ => beep.play(ringtone::SILENCE).expect("Can not beep."),
                 }
-                    let result = mq_event_loop.post(
+                match _mq_tx_for_dc_out_state.lock() {
-                        &MqDto {
+                    Ok(tx) => {
-                            topic: "dc_out_state",
+                        let result = tx.send(MqDto {
-                            message: message.to_json().as_bytes(),
+                            topic: "dc_out_state".to_string(),
-                        },
+                            message: message.to_json(),
-                        None,
+                        });
-                    );
+
                         match result {
-                        Ok(success) => {
+                            Ok(_) => {
-                            if !success {
+                                info!("send dc_out_state message success.")
-                                warn!("post dc_out_state event failed.")
+                            }
+                            Err(err) => warn!("send dc_out_state message failed. {}", err),
                         }
                     }
-                        Err(err) => warn!("post dc_out_state event failed. {}", err),
+                    Err(err) => warn!("send dc_out_state to mq message failed. {}", err),
                 }
             })
             .expect(" Listening Event Loop Failed");
-        } else {
-            panic!("MQ_EVENTLOOP is undefined!");
-        }
     } else {
         panic!("DC_OUT_STATE_EVENT_LOOP is undefined!");
     }

src/message_queue.rs

@@ -1,38 +1,32 @@
-use std::thread;
+use std::{
+    sync::{
+        mpsc::{self, Receiver, Sender},
+        Arc, Mutex,
+    },
+    thread::{self, spawn},
+    time::Duration,
+};
 
 use anyhow::Result;
-use embedded_svc::{
+use embedded_svc::mqtt::client::{utils::ConnState, Client, Connection, MessageImpl, Publish, QoS};
-    event_bus::EventBus,
-    mqtt::client::{utils::ConnState, Client, Connection, MessageImpl, Publish, QoS},
-};
 use esp_idf_svc::{
-    eventloop::{
+    eventloop::{Background, EspEventLoop},
-        Background, EspBackgroundEventLoop, EspEventFetchData, EspEventLoop, EspEventPostData,
-        EspSubscription, EspTypedEventDeserializer, EspTypedEventSerializer, EspTypedEventSource,
-        User,
-    },
     mqtt::client::{EspMqttClient, MqttClientConfiguration},
 };
-use esp_idf_sys::{c_types, EspError};
+use esp_idf_sys::EspError;
 use log::*;
-use serde::{Deserialize, Serialize};
-
-pub static mut MQ_EVENTLOOP: Option<EspEventLoop<esp_idf_svc::eventloop::User<Background>>> = None;
 
 pub struct MessageQueue {
-    pub client: Option<EspMqttClient<ConnState<MessageImpl, EspError>>>,
+    pub tx: Arc<Mutex<Sender<MqDto>>>,
-    voltage_subscription: Option<EspSubscription<User<Background>>>,
+    pub rx: Arc<Mutex<Receiver<MqDto>>>,
 }
 
-impl MessageQueue {
+pub struct MessageQueueWatcher {
-    pub fn new() -> MessageQueue {
+    pub client: EspMqttClient<ConnState<MessageImpl, EspError>>,
-        return MessageQueue {
+}
-            client: None,
-            voltage_subscription: None,
-        };
-    }
 
-    pub fn init(&mut self) -> Result<()> {
+impl MessageQueueWatcher {
+    pub fn new() -> Result<Self> {
         let conf = MqttClientConfiguration {
             client_id: Some("rust-esp32-std-demo"),
             crt_bundle_attach: Some(esp_idf_sys::esp_crt_bundle_attach),
@@ -74,14 +68,11 @@ impl MessageQueue {
 
         info!("Published a hello message to topic \"esp32-c3-rust-wifi-demo/ping\"");
 
-        self.client = Some(client);
+        anyhow::Ok(Self { client })
-        anyhow::Ok(())
     }
 
     pub fn publish(&mut self, topic: &str, bytes: &[u8]) -> Result<u32> {
         self.client
-            .as_mut()
-            .unwrap()
             .publish(
                 format!("{}/{}", "ups_0_2", topic).as_str(),
                 QoS::AtMostOnce,
@@ -90,56 +81,35 @@ impl MessageQueue {
             )
             .map_err(|err| anyhow::anyhow!("publish message to queue was failed!. {}", err))
     }
+}
 
-    pub fn watch(mut self) -> anyhow::Result<EspSubscription<User<Background>>> {
+impl MessageQueue {
-        match EspBackgroundEventLoop::new(&Default::default()) {
+    pub fn new() -> Self {
-            Ok(eventloop) => unsafe { MQ_EVENTLOOP = Some(eventloop) },
+        let (tx, rx) = mpsc::channel();
-            Err(err) => anyhow::bail!("Init Event Loop failed. {:?}", err),
+        return MessageQueue {
+            tx: Arc::new(Mutex::new(tx)),
+            rx: Arc::new(Mutex::new(rx)),
+        };
     }
-        if let Some(eventloop) = unsafe { MQ_EVENTLOOP.as_mut() } {
+
-            let subscription = eventloop
+    pub fn watch(&mut self) -> anyhow::Result<()> {
-                .subscribe(move |dto: &MqDto| {
+        let mut watcher = MessageQueueWatcher::new()
-                    info!(
+            .map_err(|err| anyhow::anyhow!("Create MessageQueueWatcher failed. {}", err))?;
-                        "publish data: {:?}",
+        let rx = self.rx.to_owned();
-                        unsafe {
+        // let (tx, rx) = mpsc::channel::<MqDto>();
-                            String::from_utf8_unchecked(dto.message.to_vec())
+        spawn(move || loop {
-                        }
+            if let Ok(dto) = rx.lock().unwrap().recv_timeout(Duration::from_secs(1)) {
-                    );
+                if let Err(err) = watcher.publish(dto.topic.as_str(), dto.message.as_bytes()) {
-                    if let Err(err) = self.publish(dto.topic, dto.message) {
                     warn!("Can not publish message to MQTT. {}", err);
                 }
-                })
-                .map_err(|err| anyhow::anyhow!(" Listening Event Loop Failed. {}", err))?;
-            // self.voltage_subscription = Some(subscription);
-            return anyhow::Ok(subscription);
             }
-        anyhow::bail!("MQ_EVENTLOOP is Not Defined.")
+        });
+        anyhow::Ok(())
     }
 }
 
-#[derive(Copy, Debug, Clone, Serialize, Deserialize)]
+#[derive(Debug, Clone)]
-pub struct MqDto<'a, 'b> {
+pub struct MqDto {
-    pub topic: &'a str,
+    pub message: String,
-    pub message: &'b [u8],
+    pub topic: String,
-}
-
-impl EspTypedEventSource for MqDto<'_, '_> {
-    fn source() -> *const c_types::c_char {
-        b"MqDto\0".as_ptr() as *const _
-    }
-}
-
-impl EspTypedEventSerializer<MqDto<'_, '_>> for MqDto<'_, '_> {
-    fn serialize<R>(event: &MqDto, f: impl for<'a> FnOnce(&'a EspEventPostData) -> R) -> R {
-        f(&unsafe { EspEventPostData::new(Self::source(), Self::event_id(), event) })
-    }
-}
-
-impl<'b, 'c> EspTypedEventDeserializer<MqDto<'b, 'c>> for MqDto<'b, 'c> {
-    fn deserialize<R>(
-        data: &EspEventFetchData,
-        f: &mut impl for<'a> FnMut(&'a MqDto<'b, 'c>) -> R,
-    ) -> R {
-        f(unsafe { data.as_payload() })
-    }
 }