onion-vm 0.1.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

use arc_gc::gc::GC;

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

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>) -> Result<StepResult, RuntimeError> {
        if self.runnables.is_empty() {
            return Ok(StepResult::Return(OnionPair::new_static(
                &OnionObject::Boolean(true).stabilize(),
                &OnionObject::Undefined(Some("All runnables completed".to_string())).stabilize(),
            )));
        }

        let mut i = 0;
        while i < self.runnables.len() {
            match self.runnables[i].step(gc) {
                Ok(step_result) => {
                    match step_result {
                        StepResult::Continue => {
                            i += 1; // 继续处理下一个runnable
                        }
                        StepResult::NewRunnable(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
                        }
                        StepResult::Return(_) => {
                            // 移除已完成的runnable
                            self.runnables.remove(i);
                        }
                        StepResult::Error(error) => {
                            return Err(RuntimeError::StepError(error.to_string()));
                        }
                    }
                }
                Err(e) => {
                    return Err(RuntimeError::StepError(e.to_string()));
                }
            }
        }

        // 如果所有runnable都需要继续，返回Continue
        Ok(StepResult::Continue)
    }
    fn receive(
        &mut self,
        step_result: StepResult,
        _gc: &mut GC<OnionObjectCell>,
    ) -> Result<(), RuntimeError> {
        match step_result {
            StepResult::Error(err) => Err(err),
            _ => Err(RuntimeError::DetailedError(
                "Unexpected step result in AsyncScheduler".to_string(),
            )),
        }
    }

    fn copy(&self, gc: &mut GC<OnionObjectCell>) -> Box<dyn Runnable> {
        Box::new(AsyncScheduler {
            runnables: self.runnables.iter().map(|r| r.copy(gc)).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
        }))
    }
}