teo_runtime/model/relation/
decorator.rs1use 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}