735 lines
29 KiB
Rust
735 lines
29 KiB
Rust
use std::hash::{Hash, Hasher};
|
||
|
||
use oxide_kernel::shape_forge::EMPTY_SHAPE_ID;
|
||
use oxide_types::object::JsObject;
|
||
use oxide_types::value::JsValue;
|
||
|
||
use oxide_runtime_api::{NativeResult, VmHost};
|
||
|
||
macro_rules! native_try {
|
||
($expr:expr) => {
|
||
match $expr {
|
||
Ok(value) => value,
|
||
Err(err) => return NativeResult::Err(err),
|
||
}
|
||
};
|
||
}
|
||
|
||
/// Set/Map 的键包装:用 SameValue 语义比较(NaN 视为相同、±0 视为相同),
|
||
/// 非 double 值按原始位比较。
|
||
#[derive(Clone, Copy)]
|
||
pub struct SetKey(pub JsValue);
|
||
|
||
/// 把 int/double 数值统一归一化为可比较的位模式:NaN 用专属哨兵、±0 归 +0,
|
||
/// 使 `0`(int) 与 `0.0`/`-0.0`(double) 视为同一键,且 NaN 与任何数值都不同。
|
||
fn numeric_key_bits(val: JsValue) -> Option<u64> {
|
||
let d = if val.is_int() {
|
||
val.as_int() as f64
|
||
} else if val.is_double() {
|
||
val.as_double()
|
||
} else {
|
||
return None;
|
||
};
|
||
if d.is_nan() {
|
||
// 哨兵必须避开 0 与所有合法 f64 位模式碰撞(f64 位模式永不为 1)。
|
||
Some(1)
|
||
} else if d == 0.0 {
|
||
Some(0.0f64.to_bits())
|
||
} else {
|
||
Some(d.to_bits())
|
||
}
|
||
}
|
||
|
||
impl PartialEq for SetKey {
|
||
fn eq(&self, other: &Self) -> bool {
|
||
if let (Some(a), Some(b)) = (numeric_key_bits(self.0), numeric_key_bits(other.0)) {
|
||
return a == b;
|
||
}
|
||
// SAFETY: JsValue 是 8 字节 NaN-box Copy 值;此处以原始位定义非数值值的同一性。
|
||
unsafe { std::mem::transmute::<JsValue, u64>(self.0) == std::mem::transmute::<JsValue, u64>(other.0) }
|
||
}
|
||
}
|
||
|
||
impl Eq for SetKey {}
|
||
|
||
impl Hash for SetKey {
|
||
fn hash<H: Hasher>(&self, state: &mut H) {
|
||
if let Some(bits) = numeric_key_bits(self.0) {
|
||
bits.hash(state);
|
||
} else {
|
||
// SAFETY: JsValue 是 8 字节 NaN-box Copy 值;哈希原始位与上面的相等判定一致。
|
||
unsafe { std::mem::transmute::<JsValue, u64>(self.0).hash(state) }
|
||
}
|
||
}
|
||
}
|
||
|
||
pub(crate) type SetInner = indexmap::IndexSet<SetKey>;
|
||
|
||
/// 取出 Set 对象 native-data 槽中存储的 `IndexSet` 指针。
|
||
///
|
||
/// # 调用方维护的安全性契约
|
||
///
|
||
/// 指针在 Set `JsObject` 存活期间有效:`JsObject` 分配于当前 `Epoch` arena,
|
||
/// native builtin 执行期间不会调用 `Epoch::reset()`。持有分配的
|
||
/// `Box<IndexSet>` 由 `new_set_inner()` 创建,Set 生命周期内不释放。
|
||
/// native 调用为单线程,同一 Set 对象同时至多存在一个活 `*mut` 别名。
|
||
fn get_set_inner<H: VmHost>(vm: &mut H, this_val: JsValue) -> Result<*mut SetInner, JsValue> {
|
||
if !this_val.is_object() {
|
||
return Err(crate::error::create_type_error(vm, "called on non-Set object"));
|
||
}
|
||
let set_ptr = this_val.as_js_object_ptr();
|
||
if set_ptr.is_null() {
|
||
return Err(crate::error::create_type_error(vm, "Set internal state invalid"));
|
||
}
|
||
// SAFETY: set_ptr 是当前 Epoch bump 分配的 JsObject 的非空、对齐指针,
|
||
// 本调用期间有效。
|
||
let set_obj = unsafe { &*set_ptr };
|
||
if !set_obj.is_set() {
|
||
return Err(crate::error::create_type_error(vm, "Set.prototype.add called on incompatible receiver"));
|
||
}
|
||
// SAFETY: native_data 持有 `alloc_set` 写入的裸指针,即有效的
|
||
// 堆分配 `Box<IndexSet<SetKey>>`;IndexSet 至多要求 8 字节对齐,
|
||
// 全局分配器满足该要求。
|
||
let inner_ptr = set_obj.native_data() as *mut SetInner;
|
||
if inner_ptr.is_null() {
|
||
return Err(crate::error::create_type_error(vm, "Set internal state invalid"));
|
||
}
|
||
Ok(inner_ptr)
|
||
}
|
||
|
||
fn new_set_inner() -> *mut SetInner {
|
||
Box::into_raw(Box::new(SetInner::new()))
|
||
}
|
||
|
||
fn alloc_set<H: VmHost>(vm: &mut H) -> *mut JsObject {
|
||
let set_proto = vm.session().builtin_world().set_proto.as_ptr() as *mut JsObject;
|
||
let mut obj = JsObject::new_empty(EMPTY_SHAPE_ID, JsValue::from_js_object(set_proto));
|
||
obj.set_set(true);
|
||
let inner = new_set_inner();
|
||
obj.set_native_data(inner as *mut u8);
|
||
vm.alloc_object(obj)
|
||
}
|
||
|
||
/// 收集 Set 中作为对象引用的元素(GC 根边),供跨 epoch 遍历/重写时追踪。
|
||
pub fn set_native_edges(obj: &JsObject) -> Vec<JsValue> {
|
||
if !obj.is_set() {
|
||
return Vec::new();
|
||
}
|
||
let inner = obj.native_data() as *const SetInner;
|
||
if inner.is_null() {
|
||
return Vec::new();
|
||
}
|
||
unsafe {
|
||
(*inner)
|
||
.iter()
|
||
.map(|key| key.0)
|
||
.filter(|value: &JsValue| value.is_object())
|
||
.collect()
|
||
}
|
||
}
|
||
|
||
/// 克隆 Set 的 native 数据到新对象,用 `rewrite` 改写其中的对象引用。
|
||
pub fn clone_set_native_with_rewrite<F>(src: &JsObject, dst: &mut JsObject, mut rewrite: F)
|
||
where
|
||
F: FnMut(JsValue) -> JsValue,
|
||
{
|
||
if !src.is_set() {
|
||
return;
|
||
}
|
||
let inner = src.native_data() as *const SetInner;
|
||
if inner.is_null() {
|
||
dst.set_native_data(std::ptr::null_mut());
|
||
return;
|
||
}
|
||
let mut cloned = SetInner::new();
|
||
unsafe {
|
||
for key in (*inner).iter() {
|
||
let new_key = if key.0.is_object() { SetKey(rewrite(key.0)) } else { *key };
|
||
cloned.insert(new_key);
|
||
}
|
||
}
|
||
dst.set_native_data(Box::into_raw(Box::new(cloned)) as *mut u8);
|
||
}
|
||
|
||
/// 原地重写 Set 的 native 数据,用 `rewrite` 改写其中的对象引用。
|
||
pub fn rewrite_set_native<F>(obj: &mut JsObject, mut rewrite: F)
|
||
where
|
||
F: FnMut(JsValue) -> JsValue,
|
||
{
|
||
if !obj.is_set() {
|
||
return;
|
||
}
|
||
let inner = obj.native_data() as *mut SetInner;
|
||
if inner.is_null() {
|
||
return;
|
||
}
|
||
unsafe {
|
||
let mut rewritten = SetInner::with_capacity((*inner).len());
|
||
for key in (*inner).iter() {
|
||
let new_key = if key.0.is_object() { SetKey(rewrite(key.0)) } else { *key };
|
||
rewritten.insert(new_key);
|
||
}
|
||
*inner = rewritten;
|
||
}
|
||
}
|
||
|
||
/// 释放 Set 的 native 数据(IndexSet),返回释放的字节数供泄漏统计。
|
||
pub fn drop_set_native(obj: &mut JsObject) -> u64 {
|
||
if !obj.is_set() {
|
||
return 0;
|
||
}
|
||
let inner = obj.native_data() as *mut SetInner;
|
||
if inner.is_null() {
|
||
return 0;
|
||
}
|
||
unsafe {
|
||
let boxed: Box<SetInner> = Box::from_raw(inner);
|
||
let bytes = std::mem::size_of::<SetInner>() + boxed.capacity() * std::mem::size_of::<SetKey>();
|
||
drop(boxed);
|
||
obj.set_native_data(std::ptr::null_mut());
|
||
bytes as u64
|
||
}
|
||
}
|
||
|
||
/// `Set` 构造函数:创建带空 IndexSet native 数据的 Set 对象,若提供可迭代实参
|
||
/// 则按插入序逐个 add。
|
||
///
|
||
/// # 步骤
|
||
/// 1. 校验 NewTarget:`this` 的原型须是 Set.prototype(普通调用 `Set()` 抛 TypeError)。
|
||
/// 2. 创建空 Set。
|
||
/// 3. 取 adder = Get(set, "add"),要求可调用(否则 TypeError)。
|
||
/// 4. 对可迭代实参逐元素调用 adder(异常时先 IteratorClose 再透传)。
|
||
///
|
||
/// # 边界与前提
|
||
/// - 无实参或实参为 null/undefined 时返回空 Set,不触碰 adder。
|
||
pub fn set_constructor<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let is_new_call = this_val.is_object() && {
|
||
let set_proto = vm.session().builtin_world().set_proto.as_ptr() as *mut JsObject;
|
||
// 沿原型链查找 Set.prototype:`new Set()` 直接命中,子类 `super()` 经
|
||
// 子类 prototype 链命中;普通调用(global/undefined)不命中。
|
||
let this_ptr = this_val.as_js_object_ptr();
|
||
if this_ptr.is_null() {
|
||
false
|
||
} else {
|
||
let mut proto = unsafe { &*this_ptr }.proto();
|
||
let mut found = false;
|
||
for _ in 0..16 {
|
||
if !proto.is_object() {
|
||
break;
|
||
}
|
||
let proto_ptr = proto.as_js_object_ptr();
|
||
if proto_ptr.is_null() {
|
||
break;
|
||
}
|
||
if std::ptr::eq(proto_ptr, set_proto) {
|
||
found = true;
|
||
break;
|
||
}
|
||
proto = unsafe { &*proto_ptr }.proto();
|
||
}
|
||
found
|
||
}
|
||
};
|
||
if !is_new_call {
|
||
return NativeResult::Err(crate::error::create_type_error(vm, "Set must be called with new"));
|
||
}
|
||
|
||
let set_obj = alloc_set(vm);
|
||
let set_val = JsValue::from_js_object(set_obj);
|
||
|
||
if args.len() > 1 {
|
||
let iterable = vm.reg(args[1]);
|
||
if !iterable.is_undefined() && !iterable.is_null() {
|
||
let set_ref = unsafe { &*set_obj };
|
||
let add_si = vm.kernel_core().perm_interner().intern("add").0;
|
||
let adder = match vm.ordinary_get(set_ref, add_si, set_val) {
|
||
Ok(v) => v,
|
||
Err(err) => return NativeResult::Err(crate::iterator::engine_error(vm, &err)),
|
||
};
|
||
if !crate::iterator::is_callable(adder) {
|
||
return NativeResult::Err(crate::error::create_type_error(vm, "Set.add is not callable"));
|
||
}
|
||
if let Err(err) = crate::iterator::iterate_elements(vm, iterable, |vm, elem| {
|
||
match vm.call_function_sync(adder, set_val, &[elem]) {
|
||
Ok(_) => Ok(()),
|
||
Err(err) => Err(crate::iterator::engine_error(vm, &err)),
|
||
}
|
||
}) {
|
||
return NativeResult::Err(err);
|
||
}
|
||
}
|
||
}
|
||
|
||
NativeResult::Ok(set_val)
|
||
}
|
||
|
||
/// `Set.prototype.add(value)`:插入元素,返回 this。
|
||
pub fn set_add<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let val = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
unsafe {
|
||
(*inner).insert(SetKey(val));
|
||
}
|
||
NativeResult::Ok(this_val)
|
||
}
|
||
|
||
/// `Set.prototype.has(value)`:元素是否存在。
|
||
pub fn set_has<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let val = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let found = unsafe { (*inner).contains(&SetKey(val)) };
|
||
NativeResult::Ok(JsValue::bool(found))
|
||
}
|
||
|
||
/// `Set.prototype.delete(value)`:删除元素并返回是否删除成功。
|
||
pub fn set_delete<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let val = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let removed = unsafe { (*inner).shift_remove(&SetKey(val)) };
|
||
NativeResult::Ok(JsValue::bool(removed))
|
||
}
|
||
|
||
/// `Set.prototype.clear()`:清空全部元素,返回 undefined。
|
||
pub fn set_clear<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
unsafe {
|
||
(*inner).clear();
|
||
}
|
||
NativeResult::Ok(JsValue::undefined())
|
||
}
|
||
|
||
/// `Set.prototype.forEach(callbackfn, thisArg)`:按插入序对每个值调用回调,
|
||
/// 回调参数为 `(value, value, set)`。迭代期间新增的元素也会被访问。
|
||
pub fn set_for_each<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let callback = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
if !crate::iterator::is_callable(callback) {
|
||
return NativeResult::Err(crate::error::create_type_error(vm, "callback is not a function"));
|
||
}
|
||
let this_arg = if args.len() > 2 { vm.reg(args[2]) } else { JsValue::undefined() };
|
||
// 按下标迭代:每次回调后重新读当前下标(支持迭代期间插入)。
|
||
let mut index = 0usize;
|
||
loop {
|
||
let value = unsafe { (*inner).get_index(index).map(|elem| elem.0) };
|
||
let Some(value) = value else { break };
|
||
index += 1;
|
||
if let Err(err) = vm.call_function_sync(callback, this_arg, &[value, value, this_val]) {
|
||
return NativeResult::Err(crate::iterator::engine_error(vm, &err));
|
||
}
|
||
}
|
||
NativeResult::Ok(JsValue::undefined())
|
||
}
|
||
|
||
/// `Set.prototype.size` getter:返回元素数量。
|
||
pub fn set_size<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
NativeResult::Ok(JsValue::float(unsafe { (*inner).len() } as f64))
|
||
}
|
||
|
||
/// `Set.prototype.entries()`:返回按插入序迭代 `[value, value]` 对的迭代器。
|
||
pub fn set_entries<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let _inner = native_try!(get_set_inner(vm, this_val));
|
||
NativeResult::Ok(crate::iterator::make_mode_iterator(
|
||
vm,
|
||
this_val,
|
||
crate::iterator::set_entries_iter_next::<H> as *const (),
|
||
))
|
||
}
|
||
|
||
/// `Set.prototype.values()`:返回按插入序迭代值的迭代器。
|
||
pub fn set_values<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let _inner = native_try!(get_set_inner(vm, this_val));
|
||
NativeResult::Ok(crate::iterator::make_mode_iterator(
|
||
vm,
|
||
this_val,
|
||
crate::iterator::set_values_iter_next::<H> as *const (),
|
||
))
|
||
}
|
||
|
||
/// `Set.prototype.keys()`:别名 `values()`(Set 无独立键),返回同样的迭代器。
|
||
pub fn set_keys<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let _inner = native_try!(get_set_inner(vm, this_val));
|
||
// Set 的 keys() 是 values() 的别名——返回同样的逐元素迭代器。
|
||
NativeResult::Ok(crate::iterator::make_mode_iterator(
|
||
vm,
|
||
this_val,
|
||
crate::iterator::set_values_iter_next::<H> as *const (),
|
||
))
|
||
}
|
||
|
||
/// SetRecord 协议:读取参数对象的 `size`/`has`/`keys` 并校验。
|
||
///
|
||
/// # 步骤
|
||
/// 1. 参数非对象抛 TypeError。
|
||
/// 2. `size` 经 ToNumber 转换,为 NaN 抛 TypeError。
|
||
/// 3. `has`/`keys` 须可调用,否则抛 TypeError。
|
||
///
|
||
/// # 返回值
|
||
/// `(对象, size, has, keys)`。
|
||
fn get_set_record<H: VmHost>(vm: &mut H, other: JsValue) -> Result<(JsValue, f64, JsValue, JsValue), JsValue> {
|
||
if !other.is_object() {
|
||
return Err(crate::error::create_type_error(vm, "argument is not an object"));
|
||
}
|
||
let obj = unsafe { &*other.as_js_object_ptr() };
|
||
let size_si = vm.kernel_core().perm_interner().intern("size").0;
|
||
let raw_size = vm
|
||
.ordinary_get(obj, size_si, other)
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
let num_size = match oxide_runtime_api::to_number_full(raw_size, vm) {
|
||
Ok(n) => n,
|
||
Err(err) => return Err(crate::error::create_type_error(vm, &err)),
|
||
};
|
||
if num_size.is_nan() {
|
||
return Err(crate::error::create_type_error(vm, "size must be a number"));
|
||
}
|
||
let has_si = vm.kernel_core().perm_interner().intern("has").0;
|
||
let has = vm
|
||
.ordinary_get(obj, has_si, other)
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
if !crate::iterator::is_callable(has) {
|
||
return Err(crate::error::create_type_error(vm, "has must be callable"));
|
||
}
|
||
let keys_si = vm.kernel_core().perm_interner().intern("keys").0;
|
||
let keys = vm
|
||
.ordinary_get(obj, keys_si, other)
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
if !crate::iterator::is_callable(keys) {
|
||
return Err(crate::error::create_type_error(vm, "keys must be callable"));
|
||
}
|
||
Ok((other, num_size, has, keys))
|
||
}
|
||
|
||
/// 把 -0.0 归一化为 +0.0(集合运算把 `-0𝔽` 当 `+0𝔽` 处理)。
|
||
fn normalize_neg_zero(value: JsValue) -> JsValue {
|
||
if value.is_double() {
|
||
let d = value.as_double();
|
||
if d == 0.0 && d.is_sign_negative() {
|
||
return JsValue::float(0.0);
|
||
}
|
||
}
|
||
value
|
||
}
|
||
|
||
/// 用一个已填充的 `SetInner` 创建普通 Set 对象(原型为 Set.prototype)。
|
||
fn alloc_set_with_inner<H: VmHost>(vm: &mut H, inner: SetInner) -> JsValue {
|
||
let set_proto = vm.session().builtin_world().set_proto.as_ptr() as *mut JsObject;
|
||
let mut obj = JsObject::new_empty(EMPTY_SHAPE_ID, JsValue::from_js_object(set_proto));
|
||
obj.set_set(true);
|
||
obj.set_native_data(Box::into_raw(Box::new(inner)) as *mut u8);
|
||
JsValue::from_js_object(vm.alloc_object(obj))
|
||
}
|
||
|
||
/// 回调控制流:`Continue` 继续迭代,`Stop` 提前终止并 IteratorClose。
|
||
enum KeysFlow {
|
||
Continue,
|
||
Stop,
|
||
}
|
||
|
||
/// 调用 SetRecord 的 `keys` 方法取得迭代器并逐元素回调。
|
||
///
|
||
/// # 步骤
|
||
/// 1. `Call(keys, set)` 得迭代器对象(非对象抛 TypeError)。
|
||
/// 2. 逐次 `next` 读取 `{done, value}` 交给 `f`。
|
||
/// 3. 迭代抛错或 `f` 返回 `Stop` 时先 IteratorClose 再返回。
|
||
///
|
||
/// # 返回值
|
||
/// `Ok(Continue)` 迭代耗尽;`Ok(Stop)` 回调提前终止;`Err` 透传异常。
|
||
fn iterate_record_keys<H: VmHost, F>(
|
||
vm: &mut H, set: JsValue, keys_method: JsValue, mut f: F,
|
||
) -> Result<KeysFlow, JsValue>
|
||
where
|
||
F: FnMut(&mut H, JsValue) -> Result<KeysFlow, JsValue>,
|
||
{
|
||
let iter = match vm.call_function_sync(keys_method, set, &[]) {
|
||
Ok(v) => v,
|
||
Err(err) => return Err(crate::iterator::engine_error(vm, &err)),
|
||
};
|
||
if !iter.is_object() {
|
||
return Err(crate::error::create_type_error(vm, "keys() result is not an object"));
|
||
}
|
||
let next_si = vm.kernel_core().perm_interner().intern("next").0;
|
||
let done_si = vm.kernel_core().perm_interner().intern("done").0;
|
||
let value_si = vm.kernel_core().perm_interner().intern("value").0;
|
||
let mut flow = KeysFlow::Continue;
|
||
let run: Result<(), JsValue> = (|| {
|
||
loop {
|
||
let iter_obj = unsafe { &*iter.as_js_object_ptr() };
|
||
let next_fn = vm
|
||
.ordinary_get(iter_obj, next_si, iter)
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
let result = vm
|
||
.call_function_sync(next_fn, iter, &[])
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
if !result.is_object() {
|
||
return Err(crate::error::create_type_error(vm, "iterator result is not an object"));
|
||
}
|
||
let result_obj = unsafe { &*result.as_js_object_ptr() };
|
||
let done = vm
|
||
.ordinary_get(result_obj, done_si, result)
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
if oxide_runtime_api::to_boolean(done) {
|
||
break;
|
||
}
|
||
let value = vm
|
||
.ordinary_get(result_obj, value_si, result)
|
||
.map_err(|e| crate::iterator::engine_error(vm, &e))?;
|
||
flow = f(vm, value)?;
|
||
if matches!(flow, KeysFlow::Stop) {
|
||
break;
|
||
}
|
||
}
|
||
Ok(())
|
||
})();
|
||
if run.is_err() || matches!(flow, KeysFlow::Stop) {
|
||
crate::iterator::close_iterator(vm, iter);
|
||
}
|
||
run.map(|()| flow)
|
||
}
|
||
|
||
/// `Set.prototype.union(other)`:返回 this 与 other 的并集(新 Set,顺序为 this 在前)。
|
||
///
|
||
/// # 步骤
|
||
/// 1. 取 SetRecord,复制 this 的 [[SetData]]。
|
||
/// 2. 经 other 的 `keys()` 迭代追加缺失元素,`-0𝔽` 归一化为 `+0𝔽`。
|
||
///
|
||
/// # 注意事项
|
||
/// 只调用 other 的 `keys`,不调用 `has`。
|
||
pub fn set_union<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, _size, _has, rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
let mut result: SetInner = unsafe { (*inner).clone() };
|
||
let flow = match iterate_record_keys(vm, rec_set, rec_keys, |_vm, value| {
|
||
result.insert(SetKey(normalize_neg_zero(value)));
|
||
Ok(KeysFlow::Continue)
|
||
}) {
|
||
Ok(f) => f,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
debug_assert!(matches!(flow, KeysFlow::Continue));
|
||
NativeResult::Ok(alloc_set_with_inner(vm, result))
|
||
}
|
||
|
||
/// `Set.prototype.intersection(other)`:返回 this 与 other 的交集(新 Set)。
|
||
///
|
||
/// # 步骤
|
||
/// 1. 取 SetRecord。
|
||
/// 2. this 元素数 ≤ other 时:遍历 this,保留 other 的 `has(e)` 为真的元素(顺序为 this)。
|
||
/// 3. 否则遍历 other 的 `keys()`,保留 this 中含有的元素(顺序为 other)。
|
||
pub fn set_intersection<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, rec_size, rec_has, rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
let this_size = unsafe { (*inner).len() } as f64;
|
||
let mut result = SetInner::new();
|
||
if this_size <= rec_size {
|
||
let entries: Vec<JsValue> = unsafe { (*inner).iter().map(|k| k.0).collect() };
|
||
for e in entries {
|
||
let in_other = match vm.call_function_sync(rec_has, rec_set, &[e]) {
|
||
Ok(v) => v,
|
||
Err(err) => return NativeResult::Err(crate::iterator::engine_error(vm, &err)),
|
||
};
|
||
if oxide_runtime_api::to_boolean(in_other) {
|
||
result.insert(SetKey(e));
|
||
}
|
||
}
|
||
} else {
|
||
match iterate_record_keys(vm, rec_set, rec_keys, |_vm, value| {
|
||
let value = normalize_neg_zero(value);
|
||
if unsafe { (*inner).contains(&SetKey(value)) } {
|
||
result.insert(SetKey(value));
|
||
}
|
||
Ok(KeysFlow::Continue)
|
||
}) {
|
||
Ok(_) => {}
|
||
Err(err) => return NativeResult::Err(err),
|
||
}
|
||
}
|
||
NativeResult::Ok(alloc_set_with_inner(vm, result))
|
||
}
|
||
|
||
/// `Set.prototype.difference(other)`:返回 this 中不在 other 的元素(新 Set)。
|
||
///
|
||
/// # 步骤
|
||
/// 1. 取 SetRecord。
|
||
/// 2. this 元素数 ≤ other 时:遍历 this,保留 other 的 `has(e)` 为假的元素。
|
||
/// 3. 否则复制 this,经 other 的 `keys()` 逐个移除存在的元素。
|
||
pub fn set_difference<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, rec_size, rec_has, rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
let this_size = unsafe { (*inner).len() } as f64;
|
||
let mut result = SetInner::new();
|
||
if this_size <= rec_size {
|
||
let entries: Vec<JsValue> = unsafe { (*inner).iter().map(|k| k.0).collect() };
|
||
for e in entries {
|
||
let in_other = match vm.call_function_sync(rec_has, rec_set, &[e]) {
|
||
Ok(v) => v,
|
||
Err(err) => return NativeResult::Err(crate::iterator::engine_error(vm, &err)),
|
||
};
|
||
if !oxide_runtime_api::to_boolean(in_other) {
|
||
result.insert(SetKey(e));
|
||
}
|
||
}
|
||
} else {
|
||
unsafe {
|
||
for k in (*inner).iter() {
|
||
result.insert(*k);
|
||
}
|
||
}
|
||
match iterate_record_keys(vm, rec_set, rec_keys, |_vm, value| {
|
||
result.shift_remove(&SetKey(normalize_neg_zero(value)));
|
||
Ok(KeysFlow::Continue)
|
||
}) {
|
||
Ok(_) => {}
|
||
Err(err) => return NativeResult::Err(err),
|
||
}
|
||
}
|
||
NativeResult::Ok(alloc_set_with_inner(vm, result))
|
||
}
|
||
|
||
/// `Set.prototype.symmetricDifference(other)`:返回对称差集(新 Set)。
|
||
///
|
||
/// # 步骤
|
||
/// 1. 取 SetRecord,复制 this 的 [[SetData]]。
|
||
/// 2. 经 other 的 `keys()`:元素已在结果中则移除,否则追加。
|
||
pub fn set_symmetric_difference<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, _size, _has, rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
let mut result: SetInner = unsafe { (*inner).clone() };
|
||
match iterate_record_keys(vm, rec_set, rec_keys, |_vm, value| {
|
||
let key = SetKey(normalize_neg_zero(value));
|
||
if result.shift_remove(&key) {
|
||
// 已在结果中(交集部分)→ 移除。
|
||
} else {
|
||
result.insert(key);
|
||
}
|
||
Ok(KeysFlow::Continue)
|
||
}) {
|
||
Ok(_) => {}
|
||
Err(err) => return NativeResult::Err(err),
|
||
}
|
||
NativeResult::Ok(alloc_set_with_inner(vm, result))
|
||
}
|
||
|
||
/// `Set.prototype.isSubsetOf(other)`:this 的每个元素都在 other 中。
|
||
/// this 元素数 > other 的 size 时直接返回 false;否则遍历 this 调 other 的 `has`。
|
||
pub fn set_is_subset_of<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, rec_size, rec_has, _rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
if (unsafe { (*inner).len() }) as f64 > rec_size {
|
||
return NativeResult::Ok(JsValue::bool(false));
|
||
}
|
||
let entries: Vec<JsValue> = unsafe { (*inner).iter().map(|k| k.0).collect() };
|
||
for e in entries {
|
||
let in_other = match vm.call_function_sync(rec_has, rec_set, &[e]) {
|
||
Ok(v) => v,
|
||
Err(err) => return NativeResult::Err(crate::iterator::engine_error(vm, &err)),
|
||
};
|
||
if !oxide_runtime_api::to_boolean(in_other) {
|
||
return NativeResult::Ok(JsValue::bool(false));
|
||
}
|
||
}
|
||
NativeResult::Ok(JsValue::bool(true))
|
||
}
|
||
|
||
/// `Set.prototype.isSupersetOf(other)`:other 的每个元素都在 this 中。
|
||
/// this 元素数 < other 的 size 时直接返回 false;否则经 other 的 `keys()` 逐元素
|
||
/// 检查,提前发现不在时终止并 IteratorClose。
|
||
pub fn set_is_superset_of<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, rec_size, _has, rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
if ((unsafe { (*inner).len() }) as f64) < rec_size {
|
||
return NativeResult::Ok(JsValue::bool(false));
|
||
}
|
||
let flow = match iterate_record_keys(vm, rec_set, rec_keys, |_vm, value| {
|
||
if !unsafe { (*inner).contains(&SetKey(normalize_neg_zero(value))) } {
|
||
Ok(KeysFlow::Stop)
|
||
} else {
|
||
Ok(KeysFlow::Continue)
|
||
}
|
||
}) {
|
||
Ok(f) => f,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
NativeResult::Ok(JsValue::bool(matches!(flow, KeysFlow::Continue)))
|
||
}
|
||
|
||
/// `Set.prototype.isDisjointFrom(other)`:this 与 other 无公共元素。
|
||
/// this 元素数 ≤ other 时遍历 this 调 other 的 `has`;否则遍历 other 的 `keys()`。
|
||
pub fn set_is_disjoint_from<H: VmHost>(vm: &mut H, args: &[u8]) -> NativeResult {
|
||
let this_val = vm.reg(if args.is_empty() { 0 } else { args[0] });
|
||
let inner = native_try!(get_set_inner(vm, this_val));
|
||
let other = vm.reg(if args.len() > 1 { args[1] } else { 0 });
|
||
let (rec_set, rec_size, rec_has, rec_keys) = match get_set_record(vm, other) {
|
||
Ok(r) => r,
|
||
Err(err) => return NativeResult::Err(err),
|
||
};
|
||
let this_size = unsafe { (*inner).len() } as f64;
|
||
let disjoint = if this_size <= rec_size {
|
||
let entries: Vec<JsValue> = unsafe { (*inner).iter().map(|k| k.0).collect() };
|
||
let mut disjoint = true;
|
||
for e in entries {
|
||
let in_other = match vm.call_function_sync(rec_has, rec_set, &[e]) {
|
||
Ok(v) => v,
|
||
Err(err) => return NativeResult::Err(crate::iterator::engine_error(vm, &err)),
|
||
};
|
||
if oxide_runtime_api::to_boolean(in_other) {
|
||
disjoint = false;
|
||
break;
|
||
}
|
||
}
|
||
disjoint
|
||
} else {
|
||
match iterate_record_keys(vm, rec_set, rec_keys, |_vm, value| {
|
||
if unsafe { (*inner).contains(&SetKey(normalize_neg_zero(value))) } {
|
||
Ok(KeysFlow::Stop)
|
||
} else {
|
||
Ok(KeysFlow::Continue)
|
||
}
|
||
}) {
|
||
Ok(KeysFlow::Continue) => true,
|
||
Ok(KeysFlow::Stop) => false,
|
||
Err(err) => return NativeResult::Err(err),
|
||
}
|
||
};
|
||
NativeResult::Ok(JsValue::bool(disjoint))
|
||
}
|