viceroy-lib 0.17.0

Viceroy implementation details.
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
105
106

use {
    crate::component::bindings::fastly::compute::{erl, types},
    crate::linking::ComponentCtx,
};

use wasmtime::component::Resource;

impl erl::Host for ComponentCtx {}

impl erl::HostRateCounter for ComponentCtx {
    fn open(&mut self, name: String) -> Result<Resource<String>, erl::OpenError> {
        let res = self.wasi_table.push(name).unwrap();

        Ok(res)
    }

    fn get_name(&mut self, rc: Resource<String>) -> String {
        self.wasi_table.get(&rc).unwrap().to_owned()
    }

    fn check_rate(
        &mut self,
        rc: Resource<String>,
        entry: String,
        delta: u32,
        window: u32,
        limit: u32,
        pb: Resource<String>,
        ttl: u32,
    ) -> Result<bool, types::Error> {
        let rc = self.wasi_table.get(&rc).unwrap();
        let pb = self.wasi_table.get(&pb).unwrap();
        crate::component::erl::check_rate(
            &mut self.session,
            rc,
            entry,
            delta,
            window,
            limit,
            pb,
            ttl,
        )
    }

    fn increment(
        &mut self,
        rc: Resource<String>,
        entry: String,
        delta: u32,
    ) -> Result<(), types::Error> {
        let rc = self.wasi_table.get(&rc).unwrap();
        crate::component::erl::ratecounter_increment(&mut self.session, rc, entry, delta)
    }

    fn lookup_rate(
        &mut self,
        rc: Resource<String>,
        entry: String,
        window: u32,
    ) -> Result<u32, types::Error> {
        let rc = self.wasi_table.get(&rc).unwrap();
        crate::component::erl::ratecounter_lookup_rate(&mut self.session, rc, entry, window)
    }

    fn lookup_count(
        &mut self,
        rc: Resource<String>,
        entry: String,
        duration: u32,
    ) -> Result<u32, types::Error> {
        let rc = self.wasi_table.get(&rc).unwrap();
        crate::component::erl::ratecounter_lookup_count(&mut self.session, rc, entry, duration)
    }

    fn drop(&mut self, ratecounter: Resource<String>) -> wasmtime::Result<()> {
        self.wasi_table.delete(ratecounter)?;
        Ok(())
    }
}

impl erl::HostPenaltyBox for ComponentCtx {
    fn open(&mut self, name: String) -> Result<Resource<String>, erl::OpenError> {
        let res = self.wasi_table.push(name).unwrap();

        Ok(res)
    }

    fn get_name(&mut self, pb: Resource<String>) -> String {
        self.wasi_table.get(&pb).unwrap().to_owned()
    }

    fn add(&mut self, pb: Resource<String>, entry: String, ttl: u32) -> Result<(), types::Error> {
        let pb = self.wasi_table.get(&pb).unwrap();
        crate::component::erl::penaltybox_add(&mut self.session, pb, entry, ttl)
    }

    fn has(&mut self, pb: Resource<String>, entry: String) -> Result<bool, types::Error> {
        let pb = self.wasi_table.get(&pb).unwrap();
        crate::component::erl::penaltybox_has(&mut self.session, pb, entry)
    }

    fn drop(&mut self, penaltybox: Resource<String>) -> wasmtime::Result<()> {
        self.wasi_table.delete(penaltybox)?;
        Ok(())
    }
}