yao 0.1.4

a fast, tiny, extensiable workflow engine
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

use crate::{plugin, ActPlugin, Engine, Message, Workflow};
use std::sync::{Arc, Mutex};

#[test]
fn plugin_register() {
    let engine = Engine::new();

    let plugin_count = engine.plugins.lock().unwrap().len();
    engine.register_plugin(&TestPlugin::new());
    assert_eq!(engine.plugins.lock().unwrap().len(), plugin_count + 1);
}

#[tokio::test]
async fn plugin_init() {
    let engine = Engine::new();

    let test_plugin = TestPlugin::new();
    engine.register_plugin(&test_plugin);
    plugin::init(&engine).await;
    assert_eq!(*test_plugin.is_init.lock().unwrap(), true);
}

#[derive(Debug, Clone)]
struct TestPlugin {
    is_init: Arc<Mutex<bool>>,
}

impl TestPlugin {
    fn new() -> Self {
        Self {
            is_init: Arc::new(Mutex::new(false)),
        }
    }
}

impl ActPlugin for TestPlugin {
    fn on_init(&self, engine: &Engine) {
        println!("TestPlugin");
        *self.is_init.lock().unwrap() = true;

        // engine.register_module("name", module);
        // engine.register_action("func", func);
        engine.on_workflow_start(|_w: &Workflow| {});
        engine.on_workflow_complete(|_w: &Workflow| {});

        engine.on_message(|_msg: &Message| {});
    }
}