Short circuit functions if empty: array, map, string.

This commit is contained in:
Stephen Chung
2021-09-20 18:35:23 +08:00
parent 6e7fbfa66f
commit adbe9a292e
3 changed files with 352 additions and 137 deletions

View File

@@ -17,32 +17,56 @@ def_package!(crate:BasicMapPackage:"Basic object map utilities.", lib, {
mod map_functions {
#[rhai_fn(name = "has", pure)]
pub fn contains(map: &mut Map, prop: ImmutableString) -> bool {
map.contains_key(prop.as_str())
if map.is_empty() {
false
} else {
map.contains_key(prop.as_str())
}
}
#[rhai_fn(pure)]
pub fn len(map: &mut Map) -> INT {
map.len() as INT
}
pub fn clear(map: &mut Map) {
map.clear();
if !map.is_empty() {
map.clear();
}
}
pub fn remove(map: &mut Map, name: ImmutableString) -> Dynamic {
map.remove(name.as_str()).unwrap_or_else(|| ().into())
if !map.is_empty() {
map.remove(name.as_str()).unwrap_or_else(|| Dynamic::UNIT)
} else {
Dynamic::UNIT
}
}
#[rhai_fn(name = "mixin", name = "+=")]
pub fn mixin(map: &mut Map, map2: Map) {
map.extend(map2.into_iter());
if !map2.is_empty() {
map.extend(map2.into_iter());
}
}
#[rhai_fn(name = "+")]
pub fn merge(map1: Map, map2: Map) -> Map {
let mut map1 = map1;
map1.extend(map2.into_iter());
map1
if map2.is_empty() {
map1
} else if map1.is_empty() {
map2
} else {
let mut map1 = map1;
map1.extend(map2.into_iter());
map1
}
}
pub fn fill_with(map: &mut Map, map2: Map) {
map2.into_iter().for_each(|(key, value)| {
map.entry(key).or_insert(value);
});
if !map2.is_empty() {
if map.is_empty() {
*map = map2;
} else {
map2.into_iter().for_each(|(key, value)| {
map.entry(key).or_insert(value);
});
}
}
}
#[rhai_fn(name = "==", return_raw, pure)]
pub fn equals(
@@ -53,23 +77,22 @@ mod map_functions {
if map1.len() != map2.len() {
return Ok(false);
}
if map1.is_empty() {
return Ok(true);
}
let mut map2 = map2;
if !map1.is_empty() {
let mut map2 = map2;
for (m1, v1) in map1.iter_mut() {
if let Some(v2) = map2.get_mut(m1) {
let equals = ctx
.call_fn_dynamic_raw(OP_EQUALS, true, &mut [v1, v2])
.map(|v| v.as_bool().unwrap_or(false))?;
for (m1, v1) in map1.iter_mut() {
if let Some(v2) = map2.get_mut(m1) {
let equals = ctx
.call_fn_dynamic_raw(OP_EQUALS, true, &mut [v1, v2])
.map(|v| v.as_bool().unwrap_or(false))?;
if !equals {
if !equals {
return Ok(false);
}
} else {
return Ok(false);
}
} else {
return Ok(false);
}
}
@@ -88,11 +111,19 @@ mod map_functions {
pub mod indexing {
#[rhai_fn(pure)]
pub fn keys(map: &mut Map) -> Array {
map.keys().cloned().map(Into::<Dynamic>::into).collect()
if map.is_empty() {
Array::new()
} else {
map.keys().cloned().map(Into::<Dynamic>::into).collect()
}
}
#[rhai_fn(pure)]
pub fn values(map: &mut Map) -> Array {
map.values().cloned().collect()
if map.is_empty() {
Array::new()
} else {
map.values().cloned().collect()
}
}
}
}