momento-functions-bytes 0.24.0

off-guest buffer management
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

/// A buffer of bytes, which may be inline or on the host.
///
/// Some bulk data processing Functions may choose to pass data straight from a
/// request or response body to another request or response body. To improve
/// performance on these kinds of Functions, `Data` avoids copying the buffer
/// into your function's memory.
#[derive(Debug)]
pub struct Data {
    data: Location,
}

impl Data {
    /// Turn the Data into a plain `Vec<u8>`.
    ///
    /// If the data is buffered on the host, this will read the buffer completely
    /// into your function's memory. You should use this with caution!
    ///
    /// For small buffers, this is inconsequential. For larger buffers, this may
    /// cause your function to run out of memory or to run slowly.
    pub fn into_bytes(self) -> Vec<u8> {
        self.data.into_bytes()
    }
}

impl From<crate::wit::momento::bytes::bytes::Data> for Data {
    fn from(value: crate::wit::momento::bytes::bytes::Data) -> Self {
        match value {
            crate::wit::momento::bytes::bytes::Data::Value(buffer) => Self {
                data: Location::Inline { buffer },
            },
            crate::wit::momento::bytes::bytes::Data::Buffer(resource) => Self {
                data: Location::OnHost { resource },
            },
        }
    }
}

impl From<Vec<u8>> for Data {
    fn from(value: Vec<u8>) -> Self {
        Self {
            data: Location::Inline { buffer: value },
        }
    }
}

impl From<&[u8]> for Data {
    fn from(value: &[u8]) -> Self {
        Self {
            data: Location::Inline {
                buffer: value.to_vec(),
            },
        }
    }
}

impl From<String> for Data {
    fn from(value: String) -> Self {
        Self {
            data: Location::Inline {
                buffer: value.into_bytes(),
            },
        }
    }
}

impl From<&str> for Data {
    fn from(value: &str) -> Self {
        Self {
            data: Location::Inline {
                buffer: value.as_bytes().to_vec(),
            },
        }
    }
}

enum Location {
    Inline {
        buffer: Vec<u8>,
    },
    OnHost {
        resource: crate::wit::momento::bytes::bytes::Buffer,
    },
}
impl Location {
    fn into_bytes(self) -> Vec<u8> {
        match self {
            Location::Inline { buffer } => buffer,
            Location::OnHost { resource } => {
                let mut buffer = Vec::with_capacity(resource.remaining() as usize);
                while let Some(chunk) = resource.read(resource.remaining().max(16384)) {
                    buffer.extend(chunk);
                }
                buffer
            }
        }
    }
}

impl std::fmt::Debug for Location {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Location::Inline { buffer } => f
                .debug_struct("Inline")
                .field("length", &buffer.len())
                .finish(),
            Location::OnHost { resource } => f
                .debug_struct("OnHost")
                .field("remaining", &resource.remaining())
                .finish(),
        }
    }
}

impl From<Data> for crate::wit::momento::bytes::bytes::Data {
    fn from(data: Data) -> Self {
        match data.data {
            Location::Inline { buffer } => Self::Value(buffer),
            Location::OnHost { resource } => Self::Buffer(resource),
        }
    }
}