rskit-provider 0.2.0-alpha.3

Provider traits (request-response, stream, sink, duplex) with a tower bridge
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

//! Convenience constructors for building providers from plain async functions.

use std::future::Future;
use std::marker::PhantomData;

use rskit_errors::AppResult;

use crate::traits::{Provider, RequestResponse, Sink};

// ─── request_response_fn ─────────────────────────────────────────────────────

struct FnRR<I, O, F> {
    name: &'static str,
    f: F,
    _ph: PhantomData<fn(I) -> O>,
}

#[async_trait::async_trait]
impl<I, O, F, Fut> Provider for FnRR<I, O, F>
where
    I: Send + 'static,
    O: Send + 'static,
    F: Fn(I) -> Fut + Send + Sync + 'static,
    Fut: Future<Output = AppResult<O>> + Send + 'static,
{
    fn name(&self) -> &'static str {
        self.name
    }
}

#[async_trait::async_trait]
impl<I, O, F, Fut> RequestResponse<I, O> for FnRR<I, O, F>
where
    I: Send + 'static,
    O: Send + 'static,
    F: Fn(I) -> Fut + Send + Sync + 'static,
    Fut: Future<Output = AppResult<O>> + Send + 'static,
{
    async fn execute(&self, input: I) -> AppResult<O> {
        (self.f)(input).await
    }
}

/// Create a [`RequestResponse`] provider from an async function.
pub fn request_response_fn<I, O, F, Fut>(name: &'static str, f: F) -> impl RequestResponse<I, O>
where
    I: Send + 'static,
    O: Send + 'static,
    F: Fn(I) -> Fut + Send + Sync + 'static,
    Fut: Future<Output = AppResult<O>> + Send + 'static,
{
    FnRR {
        name,
        f,
        _ph: PhantomData,
    }
}

// ─── sink_fn ─────────────────────────────────────────────────────────────────

struct FnSink<I, F> {
    name: &'static str,
    f: F,
    _ph: PhantomData<fn(I)>,
}

#[async_trait::async_trait]
impl<I, F, Fut> Provider for FnSink<I, F>
where
    I: Send + 'static,
    F: Fn(I) -> Fut + Send + Sync + 'static,
    Fut: Future<Output = AppResult<()>> + Send + 'static,
{
    fn name(&self) -> &'static str {
        self.name
    }
}

impl<I, F, Fut> Sink<I> for FnSink<I, F>
where
    I: Send + 'static,
    F: Fn(I) -> Fut + Send + Sync + 'static,
    Fut: Future<Output = AppResult<()>> + Send + 'static,
{
    async fn send(&self, input: I) -> AppResult<()> {
        (self.f)(input).await
    }
}

/// Create a [`Sink`] provider from an async function.
pub fn sink_fn<I, F, Fut>(name: &'static str, f: F) -> impl Sink<I>
where
    I: Send + 'static,
    F: Fn(I) -> Fut + Send + Sync + 'static,
    Fut: Future<Output = AppResult<()>> + Send + 'static,
{
    FnSink {
        name,
        f,
        _ph: PhantomData,
    }
}