ladybird/Libraries/LibJS/Runtime/SetIterator.cpp
Aliaksandr Kalenik 646457099c LibJS: Avoid IteratorRecord GC-allocation in GetIterator instruction
With this change, `GetIterator` no longer GC-allocates an
`IteratorRecord`. Instead, it stores the iterator record fields in
bytecode registers. This avoids per-iteration allocations in patterns
like: `for (let [x] of array) {}`.

`IteratorRecord` now inherits from `IteratorRecordImpl`, which holds the
iteration state. This allows the existing iteration helpers
(`iterator_next()`, `iterator_step()`, etc.) operate on both the
GC-allocated and the register-backed forms.

Microbenchmarks:
1.1x array-destructuring-assignment-rest.js
1.226x array-destructuring-assignment.js
2025-11-02 20:05:47 +01:00

74 lines
1.9 KiB
C++

/*
* Copyright (c) 2021, Idan Horowitz <idan.horowitz@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <LibJS/Runtime/Array.h>
#include <LibJS/Runtime/NativeFunction.h>
#include <LibJS/Runtime/SetIterator.h>
namespace JS {
GC_DEFINE_ALLOCATOR(SetIterator);
GC::Ref<SetIterator> SetIterator::create(Realm& realm, Set& set, Object::PropertyKind iteration_kind)
{
return realm.create<SetIterator>(set, iteration_kind, realm.intrinsics().set_iterator_prototype());
}
SetIterator::SetIterator(Set& set, Object::PropertyKind iteration_kind, Object& prototype)
: Object(ConstructWithPrototypeTag::Tag, prototype)
, m_set(set)
, m_iteration_kind(iteration_kind)
, m_iterator(static_cast<Set const&>(set).begin())
{
}
void SetIterator::visit_edges(Cell::Visitor& visitor)
{
Base::visit_edges(visitor);
visitor.visit(m_set);
}
BuiltinIterator* SetIterator::as_builtin_iterator_if_next_is_not_redefined(Value next_method)
{
if (next_method.is_object()) {
auto const& next_function = next_method.as_object();
if (next_function.is_native_function()) {
auto const& native_function = static_cast<NativeFunction const&>(next_function);
if (native_function.is_set_prototype_next_builtin())
return this;
}
}
return nullptr;
}
ThrowCompletionOr<void> SetIterator::next(VM& vm, bool& done, Value& value)
{
if (m_done) {
done = true;
value = js_undefined();
return {};
}
if (m_iterator == m_set->end()) {
m_done = true;
done = true;
value = js_undefined();
return {};
}
VERIFY(m_iteration_kind != Object::PropertyKind::Key);
value = (*m_iterator).key;
++m_iterator;
if (m_iteration_kind == Object::PropertyKind::Value) {
return {};
}
value = Array::create_from(*vm.current_realm(), { value, value });
return {};
}
}