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

//! generate_fn_macro contains a simple macro that is used to generate a closure function
//! that we use for task building.
//! (use this macro with the caveat that variables captured inside the block can be reused,
//! if you need to create closures dynamically,
//! refer to generate_closure_template in convenience mod).

/// Create a closure that return a DelayTaskHandel by macro.

#[macro_export]
macro_rules! create_async_fn_body {
    ($async_body:block) => {
        move |context: TaskContext| {
            let f = async move {
                let future_inner = async move { $async_body };
                future_inner.await;

                context.finishe_task(None).await;
            };
            let handle = async_spawn(f);
            create_delay_task_handler(handle)
        }
    };

    (($($capture_variable:ident),+)$async_body:block) => {

        move |context: TaskContext| {

            $(
                concat_idents::concat_idents!(variable_ref_name = $capture_variable, "_ref" {
                    let variable_ref_name = $capture_variable.clone();
                });
            )+
            let f = async move {
                let future_inner = async move { $async_body };
                future_inner.await;

                context.finishe_task(None).await;
            };
            let handle = async_spawn(f);
            create_delay_task_handler(handle)
        }
    }
}

cfg_tokio_support!(
    /// Create a closure that return a DelayTaskHandel by macro.

    #[macro_export]
    macro_rules! create_async_fn_tokio_body {
        ($async_body:block) => {
            |context| {
                let handle = tokio_async_spawn(async move {
                    let future_inner = async move { $async_body };
                    future_inner.await;

                    context.finishe_task().await;
                });
                create_delay_task_handler(handle)
            }

            (($($capture_variable:ident),+)$async_body:block) => {

                move |context: TaskContext| {

                    $(
                        concat_idents::concat_idents!(variable_ref_name = $capture_variable, "_ref" {
                            let variable_ref_name = $capture_variable.clone();
                        });
                    )+
                    let f = async move {
                        let future_inner = async move { $async_body };
                        future_inner.await;

                        context.finishe_task().await;
                    };
                    let handle = async_spawn(f);
                    create_delay_task_handler(handle)
                }
            }
        };
    }
);