summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibJS/Runtime/WeakContainer.h
blob: 493f4fde51d2c3f120ff2b98199ffcdee1811710 (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
/*
 * Copyright (c) 2021, Idan Horowitz <idan.horowitz@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <LibJS/Heap/Heap.h>

namespace JS {

class WeakContainer {
public:
    explicit WeakContainer(Heap& heap)
        : m_heap(heap)
    {
        m_heap.did_create_weak_container({}, *this);
    }
    virtual ~WeakContainer()
    {
        deregister();
    }

    virtual void remove_sweeped_cells(Badge<Heap>, Vector<Cell*>&) = 0;

protected:
    void deregister()
    {
        if (!m_registered)
            return;
        m_heap.did_destroy_weak_container({}, *this);
        m_registered = false;
    }

private:
    bool m_registered { true };
    Heap& m_heap;
};

}