hxrts-aura-composition 0.2.0

Aura Layer 3: Handler composition and effect system assembly
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

//! Trace handler adapter

use crate::adapters::collect_ops;
use crate::adapters::utils::{deserialize_operation_params, serialize_operation_result};
use crate::registry::{HandlerContext, HandlerError, RegistrableHandler};
use async_trait::async_trait;
use aura_core::effects::trace::{TraceEffects, TraceEvent, TraceSpanId};
use aura_core::{EffectType, ExecutionMode};
use aura_effects::trace::TraceHandler;
use std::sync::Arc;

/// Adapter for TraceHandler
pub struct TraceHandlerAdapter {
    handler: Arc<dyn TraceEffects>,
}

impl TraceHandlerAdapter {
    pub fn new(handler: TraceHandler) -> Self {
        Self {
            handler: Arc::new(handler),
        }
    }

    pub fn new_shared(handler: Arc<dyn TraceEffects>) -> Self {
        Self { handler }
    }
}

#[cfg_attr(target_arch = "wasm32", async_trait(?Send))]
#[cfg_attr(not(target_arch = "wasm32"), async_trait)]
impl RegistrableHandler for TraceHandlerAdapter {
    async fn execute_operation_bytes(
        &self,
        effect_type: EffectType,
        operation: &str,
        parameters: &[u8],
        _ctx: &HandlerContext,
    ) -> Result<Vec<u8>, HandlerError> {
        if effect_type != EffectType::Trace {
            return Err(HandlerError::UnsupportedEffect { effect_type });
        }

        match operation {
            "trace_event" => {
                let event: TraceEvent =
                    deserialize_operation_params(effect_type, operation, parameters)?;
                self.handler.trace_event(event).await;
                Ok(Vec::new())
            }
            "trace_span" => {
                let event: TraceEvent =
                    deserialize_operation_params(effect_type, operation, parameters)?;
                let span = self.handler.trace_span(event).await;
                serialize_operation_result(effect_type, operation, &span)
            }
            "trace_span_end" => {
                let span: TraceSpanId =
                    deserialize_operation_params(effect_type, operation, parameters)?;
                self.handler.trace_span_end(span).await;
                Ok(Vec::new())
            }
            _ => Err(HandlerError::UnknownOperation {
                effect_type,
                operation: operation.to_string(),
            }),
        }
    }

    fn supported_operations(&self, effect_type: EffectType) -> Vec<String> {
        collect_ops(effect_type, false)
    }

    fn supports_effect(&self, effect_type: EffectType) -> bool {
        effect_type == EffectType::Trace
    }

    fn execution_mode(&self) -> ExecutionMode {
        ExecutionMode::Production
    }
}