stasis-rs 0.1.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
32

use std::sync::{Arc, RwLock};

use async_trait::async_trait;

use crate::domain::errors::{Result, StasisError};
use crate::domain::runtime::cluster_node::ClusterControlEvent;
use crate::ports::outbound::runtime::cluster_control_event_sink::ClusterControlEventSink;

#[derive(Clone, Default)]
pub struct InMemoryClusterControlEventSink {
    events: Arc<RwLock<Vec<ClusterControlEvent>>>,
}

impl InMemoryClusterControlEventSink {
    pub fn events(&self) -> Result<Vec<ClusterControlEvent>> {
        let events = self.events.read().map_err(|_| {
            StasisError::PortFailure("cluster event sink lock poisoned".to_string())
        })?;
        Ok(events.clone())
    }
}

#[async_trait]
impl ClusterControlEventSink for InMemoryClusterControlEventSink {
    async fn emit(&self, event: ClusterControlEvent) -> Result<()> {
        let mut events = self.events.write().map_err(|_| {
            StasisError::PortFailure("cluster event sink lock poisoned".to_string())
        })?;
        events.push(event);
        Ok(())
    }
}