onion-vm 0.2.2

Virtual machine runtime for the Onion programming language with async execution and garbage collection
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use arc_gc::gc::GC;

use crate::{
    lambda::runnable::{Runnable, RuntimeError, StepResult},
    types::{
        object::{OnionObject, OnionObjectCell},
        pair::OnionPair,
    },
    unwrap_step_result,
};

pub struct AsyncScheduler {
    pub(crate) runnables: Vec<Box<dyn Runnable>>,
}

impl AsyncScheduler {
    pub fn new(runnables: Vec<Box<dyn Runnable>>) -> Self {
        AsyncScheduler { runnables }
    }
}

impl Runnable for AsyncScheduler {
    fn step(&mut self, gc: &mut GC<OnionObjectCell>) -> StepResult {
        if self.runnables.is_empty() {
            return StepResult::Return(
                OnionPair::new_static(
                    &OnionObject::Boolean(true).stabilize(),
                    &OnionObject::Undefined(Some("All runnables completed".to_string().into()))
                        .stabilize(),
                )
                .into(),
            );
        }

        let mut i = 0;
        while i < self.runnables.len() {
            match self.runnables[i].step(gc) {
                StepResult::Continue => {
                    i += 1; // 继续处理下一个runnable
                }
                StepResult::NewRunnable(new_runnable) => {
                    self.runnables.push(new_runnable);
                    unwrap_step_result!(self.runnables[i].receive(
                        &StepResult::Return(
                            OnionObject::Undefined(Some("Task Launched".to_string().into()))
                                .stabilize()
                                .into(),
                        ),
                        gc,
                    ));
                    i += 1; // 继续处理下一个runnable
                }
                StepResult::ReplaceRunnable(new_runnable) => {
                    // self.runnables.push(new_runnable);
                    // self.runnables[i].receive(
                    //     StepResult::Return(
                    //         OnionObject::Undefined(Some("Task Launched".to_string())).stabilize(),
                    //     ),
                    //     gc,
                    // )?;
                    // i += 1; // 继续处理下一个runnable
                    self.runnables[i] = new_runnable;
                    i += 1; // 继续处理下一个runnable
                }
                StepResult::Return(_) => {
                    // 移除已完成的runnable
                    self.runnables.remove(i);
                }
                e => return e,
            }
        }

        // 如果所有runnable都需要继续，返回Continue
        StepResult::Continue
    }
    fn receive(
        &mut self,
        _step_result: &StepResult,
        _gc: &mut GC<OnionObjectCell>,
    ) -> Result<(), RuntimeError> {
        Err(RuntimeError::DetailedError(
            "AsyncScheduler does not support receive".to_string().into(),
        ))
    }

    fn copy(&self) -> Box<dyn Runnable> {
        Box::new(AsyncScheduler {
            runnables: self.runnables.iter().map(|r| r.copy()).collect(),
        })
    }

    fn format_context(&self) -> Result<serde_json::Value, RuntimeError> {
        let mut tasks_json_array = serde_json::Value::Array(vec![]);
        for runnable in &self.runnables {
            let frame_json = runnable.format_context()?;
            tasks_json_array.as_array_mut().unwrap().push(frame_json);
        }
        // {type: "AsyncScheduler", tasks: tasks_json_array}
        Ok(serde_json::json!({
            "type": "AsyncScheduler",
            "tasks": tasks_json_array
        }))
    }
}