rsiot/components_config/master_device/device/tasks/
input_request.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
use tokio::sync::{broadcast, mpsc};

use crate::message::{Message, MsgDataBound};

use super::{
    set_address_and_send_to_fieldbus::set_address_and_send_to_fieldbus, AddressBound, Buffer,
    BufferBound, RequestResponseBound,
};

pub struct InputRequest<TMsg, TRequest, TBuffer, TAddress> {
    pub address: TAddress,
    pub buffer: Buffer<TBuffer>,
    pub ch_rx_msgbus_to_device: broadcast::Receiver<Message<TMsg>>,
    pub ch_tx_device_to_fieldbus: mpsc::Sender<TRequest>,
    pub fn_msgs_to_buffer: fn(&Message<TMsg>, &mut TBuffer),
    pub fn_buffer_to_request: fn(&TBuffer) -> Vec<TRequest>,
}

impl<TMsg, TRequest, TBuffer, TAddress> InputRequest<TMsg, TRequest, TBuffer, TAddress>
where
    TMsg: MsgDataBound,
    TRequest: RequestResponseBound<TAddress>,
    TBuffer: BufferBound,
    TAddress: AddressBound,
{
    pub async fn spawn(mut self) -> super::Result<()> {
        while let Ok(msg) = self.ch_rx_msgbus_to_device.recv().await {
            let requests = {
                let mut buffer = self.buffer.lock().await;
                let buffer_old = buffer.clone();
                (self.fn_msgs_to_buffer)(&msg, &mut buffer);
                if *buffer == buffer_old {
                    continue;
                }
                (self.fn_buffer_to_request)(&buffer)
            };

            set_address_and_send_to_fieldbus(
                requests,
                self.address,
                &self.ch_tx_device_to_fieldbus,
            )
            .await;
        }

        Ok(())
    }
}