harn-vm 0.10.170

Async bytecode virtual machine for the Harn programming language
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

//! Shared poll boundary for native and interpreter execution scopes.

use super::{AmbientExecutionScope, RestoreGuard, Scoped};
use std::{
    future::Future,
    pin::Pin,
    task::{Context, Poll},
};

impl<F: Future> Future for Scoped<F> {
    type Output = F::Output;

    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<F::Output> {
        let this = self.project();
        this.scope
            .poll_scoped(*this.commit_on_ready, || this.inner.poll(cx))
    }
}

impl AmbientExecutionScope {
    pub(crate) fn poll_scoped<T>(
        &mut self,
        commit_on_ready: bool,
        poll: impl FnOnce() -> Poll<T>,
    ) -> Poll<T> {
        self.swap_in_place();
        let mut restore = RestoreGuard {
            scope: self,
            armed: true,
        };
        let result = poll();
        if result.is_ready() && commit_on_ready {
            restore.commit();
        }
        result
    }
}