stasis-rs 0.2.0

Durable AI orchestration framework with runtime jobs, lineage, and memory integration
Documentation 
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

use async_trait::async_trait;
use tokio::sync::mpsc::{self, UnboundedReceiver, UnboundedSender};

use crate::domain::errors::{Result, StasisError};
use crate::domain::runtime::outbox::OutboxEvent;
use crate::ports::outbound::runtime::event_publisher::EventPublisher;

#[derive(Clone)]
pub struct TokioChannelEventPublisher {
    tx: UnboundedSender<OutboxEvent>,
}

impl TokioChannelEventPublisher {
    pub fn new(tx: UnboundedSender<OutboxEvent>) -> Self {
        Self { tx }
    }

    pub fn channel() -> (Self, UnboundedReceiver<OutboxEvent>) {
        let (tx, rx) = mpsc::unbounded_channel();
        (Self::new(tx), rx)
    }
}

#[async_trait]
impl EventPublisher for TokioChannelEventPublisher {
    async fn publish(&self, event: &OutboxEvent) -> Result<()> {
        self.tx
            .send(event.clone())
            .map_err(|e| StasisError::PortFailure(format!("publish to tokio channel bus: {e}")))
    }
}