Function expand

flow-derive/src/actor.rs:17–108

(input: DeriveInput)

use syn::{spanned::Spanned, DeriveInput};

pub fn expand(input: DeriveInput) -> TokenStream {
    let ident = input.ident;
    let is_local = attr(&input.attrs, "local").is_some();
    let spawn_func = if is_local {
        lit::ident("spawn_local")
    } else {
        lit::ident("spawn")
    };

    fn send_empty_f((_, ident, ty): ExtractParams) -> TokenStream {
        if match_last_ty(ty, "Vec") {
            quote_spanned! {ident.span()=>
                for chan in &self.#ident {
                    chan.send_any(flow_rs::envelope::DummyEnvelope{}.seal()).await.ok();
                }
            }
        } else if match_last_ty(ty, "HashMap") {
            quote_spanned! {ident.span()=>
                for chan in self.#ident.values() {
                    chan.send_any(flow_rs::envelope::DummyEnvelope{}.seal()).await.ok();
                }
            }
        } else if match_last_ty(ty, "DynPorts") {
            quote_spanned! (ident.span()=>
                for chan in self.#ident.cache().values() {
                    chan.send_any(flow_rs::envelope::DummyEnvelope{}.seal()).await.ok();
                }
            )
        } else {
            quote_spanned! (ident.span()=>self.#ident.send_any(flow_rs::envelope::DummyEnvelope{}.seal()).await.ok();)
        }
    }
    fn min_empty_n_f((_, ident, ty): ExtractParams) -> TokenStream {
        if match_last_ty(ty, "Vec") {
            quote_spanned! {ident.span()=>
                for chan in &self.#ident {
                    min_empty_n = std::cmp::min(chan.empty_n(), min_empty_n)
                }
            }
        } else if match_last_ty(ty, "HashMap") {
            quote_spanned! {ident.span()=>
                for chan in self.#ident.values() {
                    min_empty_n = std::cmp::min(chan.empty_n(), min_empty_n)
                }
            }
        } else if match_last_ty(ty, "DynPorts") {
            quote_spanned! (ident.span()=>
                for chan in self.#ident.cache().values() {
                    min_empty_n = std::cmp::min(chan.empty_n(), min_empty_n)
                }
            )
        } else {
            quote_spanned! (ident.span()=> min_empty_n = std::cmp::min(self.#ident.empty_n(), min_empty_n);)
        }
    }

    let send_empty = extract_ports(&input.data, type_name::OUT_T, send_empty_f);
    let recv_empty_n = extract_ports(&input.data, type_name::IN_T, min_empty_n_f);