stak_module/
hot_reload.rs

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
//! Hot-reloaded modules.

use crate::{Guard, Module};
use core::ops::Deref;

/// A hot-reloaded module.
pub struct HotReloadModule {
    module: hmr::Module,
}

impl HotReloadModule {
    /// Creates a hot-reloaded module.
    pub const fn new(path: &'static str) -> Self {
        Self {
            module: hmr::Module::new(path),
        }
    }
}

impl Module<'static> for HotReloadModule {
    type Guard = HotReloadGuard;

    fn bytecode(&'static self) -> Self::Guard {
        HotReloadGuard(self.module.load())
    }
}

/// A read guard against a hot-reloaded module.
pub struct HotReloadGuard(hmr::Guard);

impl Deref for HotReloadGuard {
    type Target = [u8];

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl Guard for HotReloadGuard {}