buildkit_llb/serialization/
mod.rs

1use std::collections::BTreeMap;
2
3mod id;
4mod operation;
5mod output;
6
7pub(crate) use self::id::OperationId;
8pub(crate) use self::operation::Operation;
9pub(crate) use self::output::Node;
10
11pub(crate) type Result<T> = std::result::Result<T, ()>;
12
13#[derive(Default)]
14pub struct Context {
15    inner: BTreeMap<u64, Node>,
16}
17
18impl Context {
19    #[allow(clippy::map_entry)]
20    pub(crate) fn register<'a>(&'a mut self, op: &dyn Operation) -> Result<&'a Node> {
21        let id = **op.id();
22
23        if !self.inner.contains_key(&id) {
24            let node = op.serialize(self)?;
25            self.inner.insert(id, node);
26        }
27
28        Ok(self.inner.get(&id).unwrap())
29    }
30
31    #[cfg(test)]
32    pub(crate) fn registered_nodes_iter(&self) -> impl Iterator<Item = &Node> {
33        self.inner.iter().map(|pair| pair.1)
34    }
35
36    pub(crate) fn into_registered_nodes(self) -> impl Iterator<Item = Node> {
37        self.inner.into_iter().map(|pair| pair.1)
38    }
39}
buildkit_llb/serialization/mod.rs

buildkit_llb/serialization/
mod.rs
