dbus_async/
handler.rs

1use crate::{DBus, DBusResult};
2use async_trait::async_trait;
3use dbus_message_parser::{message::Message, value::ObjectPath};
4use futures::{
5    channel::mpsc::{channel, Receiver},
6    lock::Mutex,
7    StreamExt,
8};
9use std::sync::Arc;
10
11/// A trait for the generic `Message` handler.
12#[async_trait]
13pub trait Handler: Send {
14    /// Handle the `Message`.
15    async fn handle(&mut self, dbus: &DBus, msg: Message) -> DBusResult<()>;
16}
17
18#[async_trait]
19pub trait Binder: Sized {
20    async fn bind(self, dbus: DBus, object_path: ObjectPath) -> DBusResult<()> {
21        let (sender, receiver) = channel(128);
22        dbus.add_method_call(object_path, sender)?;
23        self.bind_by_receiver(dbus, receiver).await
24    }
25
26    async fn bind_by_receiver(self, dbus: DBus, receiver: Receiver<Message>) -> DBusResult<()>;
27}
28
29// TODO: Wait until specialization https://github.com/rust-lang/rust/issues/31844 is finished to
30// define a default impl with Deletable
31#[async_trait]
32impl<T> Binder for T
33where
34    T: Handler,
35{
36    async fn bind_by_receiver(
37        mut self,
38        dbus: DBus,
39        mut receiver: Receiver<Message>,
40    ) -> DBusResult<()> {
41        while let Some(msg) = receiver.next().await {
42            self.handle(&dbus, msg).await?;
43        }
44        Ok(())
45    }
46}
47
48// TODO: Wait until specialization https://github.com/rust-lang/rust/issues/31844 is finished to
49// define a default impl with Deletable
50#[async_trait]
51impl<T> Binder for Arc<Mutex<T>>
52where
53    T: Handler,
54{
55    async fn bind_by_receiver(self, dbus: DBus, mut receiver: Receiver<Message>) -> DBusResult<()> {
56        while let Some(msg) = receiver.next().await {
57            let mut guard = self.lock().await;
58            guard.handle(&dbus, msg).await?;
59        }
60        Ok(())
61    }
62}