87 lines
2.1 KiB
Rust
87 lines
2.1 KiB
Rust
|
use async_trait::async_trait;
|
||
|
|
use messagebus::{
|
||
|
|
derive::{Error as MbError, Message},
|
||
|
|
error, AsyncHandler, Bus, Message, TypeTag,
|
||
|
|
};
|
||
|
|
use serde_derive::{Deserialize, Serialize};
|
||
|
|
use thiserror::Error;
|
||
|
|
|
||
|
|
#[derive(Debug, Error, MbError)]
|
||
|
|
enum Error {
|
||
|
|
#[error("Error({0})")]
|
||
|
|
Error(anyhow::Error),
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<M: Message> From<error::Error<M>> for Error {
|
||
|
|
fn from(err: error::Error<M>) -> Self {
|
||
|
|
Self::Error(err.into())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Serialize, Deserialize, Message)]
|
||
|
|
#[message(shared, clone)]
|
||
|
|
#[type_tag("Msg")]
|
||
|
|
struct Msg {
|
||
|
|
test1: u32,
|
||
|
|
test2: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Serialize, Deserialize, Message)]
|
||
|
|
#[message(shared, clone)]
|
||
|
|
#[type_tag("MsgResponse")]
|
||
|
|
struct MsgResponse {
|
||
|
|
test1: u32,
|
||
|
|
test2: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
struct TmpReceiver;
|
||
|
|
|
||
|
|
#[async_trait]
|
||
|
|
impl AsyncHandler<Msg> for TmpReceiver {
|
||
|
|
type Error = Error;
|
||
|
|
type Response = MsgResponse;
|
||
|
|
|
||
|
|
async fn handle(&self, msg: Msg, _bus: &Bus) -> Result<Self::Response, Self::Error> {
|
||
|
|
Ok(MsgResponse {
|
||
|
|
test1: msg.test1 * 2,
|
||
|
|
test2: msg.test2 + ", World!",
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[tokio::test]
|
||
|
|
async fn test() {
|
||
|
|
let (b, poller) = Bus::build()
|
||
|
|
.register_shared_message::<Msg>()
|
||
|
|
.register(TmpReceiver)
|
||
|
|
.subscribe_async::<Msg>(8, Default::default())
|
||
|
|
.done()
|
||
|
|
.build();
|
||
|
|
|
||
|
|
let type_tag: TypeTag = "Msg".into();
|
||
|
|
let message = br#"{"test1":12,"test2":"Hello"}"#;
|
||
|
|
|
||
|
|
let json = &mut serde_json::Deserializer::from_slice(message);
|
||
|
|
let mut de = <dyn erased_serde::Deserializer>::erase(json);
|
||
|
|
|
||
|
|
let val = b
|
||
|
|
.request_deserialize(type_tag, &mut de, Default::default())
|
||
|
|
.await
|
||
|
|
.unwrap();
|
||
|
|
|
||
|
|
let mut buff: Vec<u8> = Vec::new();
|
||
|
|
let json = &mut serde_json::Serializer::new(&mut buff);
|
||
|
|
let mut ser = <dyn erased_serde::Serializer>::erase(json);
|
||
|
|
val.as_shared_ref()
|
||
|
|
.unwrap()
|
||
|
|
.erased_serialize(&mut ser)
|
||
|
|
.unwrap();
|
||
|
|
|
||
|
|
assert_eq!(val.type_tag(), TypeTag::from("MsgResponse"));
|
||
|
|
assert_eq!(buff.as_slice(), br#"{"test1":24,"test2":"Hello, World!"}"#);
|
||
|
|
|
||
|
|
b.flush().await;
|
||
|
|
b.close().await;
|
||
|
|
poller.await;
|
||
|
|
}
|