use std::future::{Future, ready};
use std::process::ExitCode;
use ruststream::memory::prelude::*;
use ruststream::runtime::cli::run_main;
use serde::{Deserialize, Serialize};
#[derive(Debug, PartialEq, Deserialize, Serialize, schemars::JsonSchema)]
struct Order {
id: u64,
customer: String,
}
impl OutgoingDestination for Order {
type Form = FixedName;
const DESTINATION: &'static str = "orders.placed";
}
impl MessageHeaders for Order {
type Contract = NoHeaders;
}
struct Accept;
impl Handle<Order> for Accept {
fn handle(
&self,
order: &Order,
_outs: &(),
_ctx: &mut Context<'_>,
) -> impl Future<Output = Result<(), HandlerOutcome>> {
tracing::info!(order = order.id, customer = %order.customer, "order accepted");
ready(Ok(()))
}
}
fn app() -> impl App {
RustStream::new(AppInfo::new("shop", "0.1.0")).with_broker(MemoryBroker::new(), |b| {
b.include(subscriber("orders.placed", Accept).build());
})
}
fn main() -> ExitCode {
run_main(app)
}
#[cfg(test)]
mod tests {
use std::error::Error;
use ruststream::memory::prelude::*;
use ruststream::testing::TestApp;
use super::{Order, app};
#[tokio::test]
async fn a_placed_order_is_accepted() -> Result<(), Box<dyn Error>> {
let tb = TestApp::start(app()).await?;
let order = Order {
id: 7,
customer: "ada@example.com".to_owned(),
};
tb.message(&order).publish().await?;
tb.broker::<MemoryBroker>()
.subscriber("orders.placed")
.assert_called_once()
.with(&order)
.settled(HandlerOutcome::ack());
Ok(())
}
}