arcon_codegen 0.1.4

Code generation for Arcon 
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

use proc_macro2::{Ident, Span, TokenStream};

pub fn system(
    addr: &str,
    kompact_extra_options: Option<TokenStream>,
    kompact_connections: Option<TokenStream>,
    termination: Option<TokenStream>,
) -> TokenStream {
    quote! {
        let mut cfg = KompactConfig::default();
        let sock_addr = #addr.parse().expect("Failed to parse SocketAddr");
        cfg.system_components(DeadletterBox::new, NetworkConfig::new(sock_addr).build());

        #kompact_extra_options

        let system = KompactSystem::new(cfg).expect("Failed to create KompactSystem");

        // Connect Components, Create ActorPaths, Create Tasks

        #kompact_connections

        // enable optional termination for testing
        #termination
    }
}

pub fn await_termination(system_name: &str) -> TokenStream {
    let system = Ident::new(&system_name, Span::call_site());
    quote! {
        #system.await_termination();
    }
}