teo_runtime/model/
decorator.rs

1use std::sync::Arc;
2use educe::Educe;
3use serde::Serialize;
4use crate::arguments::Arguments;
5use teo_result::Result;
6use crate::model;
7
8pub trait Call: Send + Sync {
9    fn call(&self, args: Arguments, model_builder: &model::Builder) -> Result<()>;
10}
11
12impl<F> Call for F where
13    F: Fn(Arguments, &model::Builder) -> Result<()> + Send + Sync {
14    fn call(&self, args: Arguments, model_builder: &model::Builder) -> Result<()> {
15        self(args, model_builder)
16    }
17}
18
19#[derive(Educe, Clone)]
20#[educe(Debug)]
21pub struct Decorator {
22    inner: Arc<Inner>,
23}
24
25#[derive(Educe, Serialize)]
26#[educe(Debug)]
27struct Inner {
28    path: Vec<String>,
29    #[educe(Debug(ignore))] #[serde(skip)]
30    call: Arc<dyn Call>,
31}
32
33impl Decorator {
34
35    pub fn new<T>(path: Vec<String>, call: T) -> Self where T: Call + 'static {
36        Self {
37            inner: Arc::new(Inner {
38                path,
39                call: Arc::new(call),
40            }),
41        }
42    }
43
44    pub fn path(&self) -> &Vec<String> {
45        &self.inner.path
46    }
47
48    pub fn call(&self) -> &dyn Call {
49        self.inner.call.as_ref()
50    }
51}
52
53impl Serialize for Decorator {
54    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error> where S: serde::Serializer {
55        self.inner.serialize(serializer)
56    }
57}