maurice/snix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(tvix/eval): use im::Vector directly where possible

Browse source 
The conversion from im::Vector -> Vec is cheaper for NixList
construction (of course), so where possible we should make use of
that.

This updates most builtins dealing with lists to use Vector directly,
and marks the function constructing NixList from Vec as deprecated so
that we get appropriate warnings in places where it's still in use.

These places are currently inside of JSON serialisation logic which is
in flux right now, so lets leave them as-is until it's stabilised.

Change-Id: I037f12a2800f2576db4d9526bd935efd079163f0
Reviewed-on: https://cl.tvl.fyi/c/depot/+/7671
Reviewed-by: sterni <sternenseemann@systemli.org>
Tested-by: BuildkiteCI
This commit is contained in:
Vincent Ambo 2022-12-29 15:50:19 +03:00 • committed by tazjin
parent 6324f586c9
commit 610c44ec1e
3 changed files with 31 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tvix/eval/src/value/list.rs
View file

@ -27,14 +27,6 @@ impl TotalDisplay for NixList {
}
}
// TODO(tazjin): uses of this instance are likely inefficient and can be optimised.
// Eventually this instance should be removed.
impl From<Vec<Value>> for NixList {
fn from(vs: Vec<Value>) -> Self {
Self(Vector::from_iter(vs.into_iter()))
}
}
impl From<Vector<Value>> for NixList {
fn from(vs: Vector<Value>) -> Self {
Self(vs)
@ -57,7 +49,9 @@ mod arbitrary {
type Strategy = BoxedStrategy<Self>;
fn arbitrary_with(args: Self::Parameters) -> Self::Strategy {
any_with::<Vec<Value>>(args).prop_map(|v| v.into()).boxed()
any_with::<Vec<Value>>(args)
.prop_map(NixList::from_vec)
.boxed()
}
}
}
@ -79,7 +73,7 @@ impl NixList {
stack_slice.len(),
);
stack_slice.into()
NixList(Vector::from_iter(stack_slice.into_iter()))
}
pub fn iter(&self) -> vector::Iter<Value> {
@ -116,6 +110,11 @@ impl NixList {
pub fn into_inner(self) -> Vector<Value> {
self.0
}
#[deprecated(note = "callers should avoid constructing from Vec")]
pub fn from_vec(vs: Vec<Value>) -> Self {
Self(Vector::from_iter(vs.into_iter()))
}
}
impl IntoIterator for NixList {

7
tvix/eval/src/value/mod.rs
View file

@ -544,7 +544,7 @@ impl From<PathBuf> for Value {
impl From<Vec<Value>> for Value {
fn from(val: Vec<Value>) -> Self {
Self::List(NixList::from(val))
Self::List(NixList::from_vec(val))
}
}
@ -601,6 +601,7 @@ fn type_error(expected: &'static str, actual: &Value) -> ErrorKind {
#[cfg(test)]
mod tests {
use super::*;
use im::vector;
mod nix_eq {
use crate::observer::NoOpObserver;
@ -643,8 +644,8 @@ mod tests {
let mut observer = NoOpObserver {};
let mut vm = VM::new(Default::default(), Box::new(crate::DummyIO), &mut observer);
let v1 = Value::List(NixList::from(vec![Value::Integer(1)]));
let v2 = Value::List(NixList::from(vec![Value::Float(1.0)]));
let v1 = Value::List(NixList::from(vector![Value::Integer(1)]));
let v2 = Value::List(NixList::from(vector![Value::Float(1.0)]));
assert!(v1.nix_eq(&v2, &mut vm).unwrap())
}