use std::sync::Arc;
use aion::Engine;
use aion_core::{ActivityId, RunId, WorkflowId};
use super::bridge::OutboxDeliveryCallback;
use crate::error::ServerError;
pub struct ServerOutboxDeliveryCallback {
engine: Arc<Engine>,
}
impl ServerOutboxDeliveryCallback {
#[must_use]
pub fn new(engine: Arc<Engine>) -> Self {
Self { engine }
}
}
impl OutboxDeliveryCallback for ServerOutboxDeliveryCallback {
fn deliver_completion(
&self,
workflow_id: &WorkflowId,
activity_id: &ActivityId,
run_id: Option<&RunId>,
result: String,
) -> Result<bool, ServerError> {
self.engine
.runtime()
.deliver_outbox_completion(
self.engine.registry(),
workflow_id,
activity_id,
run_id,
result,
)
.map_err(ServerError::from)
}
fn deliver_failure(
&self,
workflow_id: &WorkflowId,
activity_id: &ActivityId,
run_id: Option<&RunId>,
reason: String,
) -> Result<bool, ServerError> {
self.engine
.runtime()
.deliver_outbox_failure(
self.engine.registry(),
workflow_id,
activity_id,
run_id,
reason,
)
.map_err(ServerError::from)
}
}