From e1f082a3ab012b0ef3b52b2b5b05513df75a4930 Mon Sep 17 00:00:00 2001 From: Vincent Ambo Date: Sun, 26 Feb 2023 18:28:49 +0300 Subject: feat(tvix/eval): add SharedThunkSet This is a ThunkSet wrapped to be shareable, which will be required once ThunkSets are embedded in futures. Change-Id: I5a067b7972ac86e4d354c75ef05c86b2284c1137 Reviewed-on: https://cl.tvl.fyi/c/depot/+/8144 Tested-by: BuildkiteCI Reviewed-by: raitobezarius --- tvix/eval/src/value/thunk.rs | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tvix/eval/src/value/thunk.rs b/tvix/eval/src/value/thunk.rs index 1be3370b2377..1c7ed615527c 100644 --- a/tvix/eval/src/value/thunk.rs +++ b/tvix/eval/src/value/thunk.rs @@ -488,3 +488,14 @@ impl ThunkSet { self.0.insert(ptr) } } + +#[derive(Default, Clone)] +pub struct SharedThunkSet(Rc>); + +impl SharedThunkSet { + /// Check whether the given thunk has already been seen. Will mark the thunk + /// as seen otherwise. + pub fn insert(&self, thunk: &Thunk) -> bool { + self.0.borrow_mut().insert(thunk) + } +} -- cgit 1.4.1