lmflow 0.3.3

Dataflow-graph compute engine: a pure-Rust scheduler behind a stable C ABI
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

#![allow(dead_code)]

use std::sync::Once;

use lmflow::{register_kernel, Graph, Kernel, KernelContract, KernelCtx};

#[derive(Default)]
struct TestPassThrough;

impl Kernel for TestPassThrough {
    fn get_contract(contract: &mut KernelContract) {
        contract.input_any(0);
        contract.output_any(0);
    }

    fn process(&mut self, context: &mut KernelCtx) -> lmflow::Result<()> {
        context.forward(0, 0)
    }
}

#[derive(Default)]
struct TestSink;

impl Kernel for TestSink {
    fn get_contract(contract: &mut KernelContract) {
        contract.input_any(0);
    }

    fn process(&mut self, context: &mut KernelCtx) -> lmflow::Result<()> {
        context.counter_add("sink.packets", 1);
        Ok(())
    }

    fn close(&mut self, context: &mut KernelCtx) -> lmflow::Result<()> {
        context.counter_add("sink.closed", 1);
        Ok(())
    }
}

pub fn register_test_kernels() {
    static ONCE: Once = Once::new();
    ONCE.call_once(|| {
        register_kernel::<TestPassThrough>("PassThrough").unwrap();
        register_kernel::<TestSink>("Sink").unwrap();
    });
}

pub fn graph_from_yaml(text: &str) -> lmflow::Result<Graph> {
    register_test_kernels();
    Graph::from_yaml(text)
}