vin-macros 6.0.0

Procedural macros that power vin
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

use proc_macro2::TokenStream as TokenStream2;
use syn::Ident;
use quote::quote;

use crate::actor::names::*;
use crate::actor::handles_attr::*;

pub fn form_vin_hidden_struct(
    name: &Ident, 
    handles_attrs: &Vec<HandlesAttr>
) -> TokenStream2 {
    // ===== Form mailbox
    // Mailbox fields to inject into the actor
    let (fulls, shorts) = form_message_names(handles_attrs);
    let mailbox_fields = fulls.iter().zip(shorts.iter()).map(|(full, short)| {
        quote! { 
            #short: (
                ::vin::async_channel::Sender<::vin::vin_core::detail::WrappedMessage<#full>>, 
                ::vin::async_channel::Receiver<::vin::vin_core::detail::WrappedMessage<#full>>
            )
        }
    }).collect::<Vec<_>>();

    let mailbox_inits = handles_attrs.iter().zip(shorts.iter()).map(|(attr, short)| {
        let max = attr.max_messages_at_once;
        quote! { let #short = ::vin::async_channel::bounded(#max) }
    }).collect::<Vec<_>>();

    // Mailbox struct to inject into the actor
    let mailbox_struct_name = form_mailbox_name(name);
    let mailbox_struct = quote! {
        struct #mailbox_struct_name {
            #(#mailbox_fields),*
        }

        impl Default for #mailbox_struct_name {
            fn default() -> Self {
                #(#mailbox_inits;)*

                Self {
                    #(#shorts),*
                }
            }
        }
    };

    // ===== Form final output
    let hidden_struct_name = form_hidden_struct_name(name);
    quote! {
        #mailbox_struct

        struct #hidden_struct_name {
            mailbox: #mailbox_struct_name,
            state: ::vin::crossbeam::atomic::AtomicCell<::vin::vin_core::State>,
            close: ::vin::tokio::sync::Notify,
            id: ::vin::vin_core::ActorId,
        }

        impl Default for #hidden_struct_name {
            fn default() -> Self {
                Self {
                    mailbox: Default::default(),
                    state: Default::default(),
                    close: Default::default(),
                    id: "none".into(),
                }
            }
        }
    }
}