summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibJS/Runtime/WeakMap.h
blob: 9e15860c684f2fbc9cdc03fd267410d7556131ca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/*
 * Copyright (c) 2021, Idan Horowitz <idan.horowitz@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/HashMap.h>
#include <LibJS/Runtime/GlobalObject.h>
#include <LibJS/Runtime/Object.h>
#include <LibJS/Runtime/WeakContainer.h>

namespace JS {

class WeakMap final
    : public Object
    , public WeakContainer {
    JS_OBJECT(WeakMap, Object);

public:
    static WeakMap* create(GlobalObject&);

    explicit WeakMap(Object& prototype);
    virtual ~WeakMap() override;

    HashMap<Cell*, Value> const& values() const { return m_values; };
    HashMap<Cell*, Value>& values() { return m_values; };

    virtual void remove_swept_cells(Badge<Heap>, Span<Cell*>) override;

private:
#ifdef JS_TRACK_ZOMBIE_CELLS
    virtual void did_become_zombie() override
    {
        deregister();
    }
#endif

    void visit_edges(Visitor&) override;

    HashMap<Cell*, Value> m_values; // This stores Cell pointers instead of Object pointers to aide with sweeping
};

}