selium-kernel 1.0.0-alpha.6

Streaming compute fabric
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

use std::{
    any::{Any, TypeId},
    collections::HashMap,
    num::TryFromIntError,
    sync::Arc,
};

use thiserror::Error;

use crate::registry::RegistryError;

pub mod drivers;
pub mod futures;
pub mod guest_async;
pub mod guest_data;
pub mod mailbox;
pub mod operation;
pub mod registry;
pub mod session;

pub struct Kernel {
    capabilities: HashMap<TypeId, Arc<dyn Any>>,
}

#[derive(Default)]
pub struct KernelBuilder {
    capabilities: HashMap<TypeId, Arc<dyn Any>>,
}

#[derive(Error, Debug)]
pub enum KernelError {
    #[error("Linker error")]
    LinkerError(#[from] wasmtime::Error),
    #[error("Could not access guest memory")]
    MemoryAccess(#[from] wasmtime::MemoryAccessError),
    #[error("Guest did not reserve enough memory for this call")]
    MemoryCapacity,
    #[error("Could not retrieve guest memory from `Caller`")]
    MemoryMissing,
    #[error("Could not convert int to usize")]
    IntConvert(#[from] TryFromIntError),
    #[error("Invalid resource handle provided by guest")]
    InvalidHandle,
    #[error("Registry error")]
    Registry(#[from] RegistryError),
    #[error("Driver error: {0}")]
    Driver(String),
}

impl Kernel {
    pub fn build() -> KernelBuilder {
        KernelBuilder::default()
    }

    pub fn get<C: 'static>(&self) -> Option<&C> {
        self.capabilities
            .get(&TypeId::of::<C>())
            .and_then(|cap| cap.downcast_ref::<C>())
    }
}

impl KernelBuilder {
    pub fn add_capability<C: 'static>(&mut self, capability: Arc<C>) -> Arc<C> {
        self.capabilities
            .insert(TypeId::of::<C>(), capability.clone());
        capability
    }

    pub fn build(self) -> Result<Kernel, KernelError> {
        Ok(Kernel {
            capabilities: self.capabilities,
        })
    }
}