teo_runtime/model/relation/
decorator.rs

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