Files
ladybird/Libraries/LibWeb/IndexedDB/Internal/Database.h
Zaggy1024 c64d276653 LibWeb: Make IDB connections weakly-referenced by their database
We shouldn't be keeping these permanently alive.
2026-04-08 03:03:38 +02:00

94 lines
3.6 KiB
C++

/*
* Copyright (c) 2024-2025, stelar7 <dudedbz@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <LibGC/HeapVector.h>
#include <LibGC/Ptr.h>
#include <LibGC/RootVector.h>
#include <LibGC/Weak.h>
#include <LibWeb/DOM/EventTarget.h>
#include <LibWeb/IndexedDB/Internal/ObjectStore.h>
#include <LibWeb/StorageAPI/StorageKey.h>
namespace Web::IndexedDB {
// https://www.w3.org/TR/IndexedDB/#database-construct
class Database : public Bindings::PlatformObject {
WEB_NON_IDL_PLATFORM_OBJECT(Database, Bindings::PlatformObject);
GC_DECLARE_ALLOCATOR(Database);
public:
void set_version(u64 version) { m_version = version; }
u64 version() const { return m_version; }
String name() const { return m_name; }
void set_upgrade_transaction(GC::Ptr<IDBTransaction> transaction) { m_upgrade_transaction = transaction; }
[[nodiscard]] GC::Ptr<IDBTransaction> upgrade_transaction() { return m_upgrade_transaction; }
void associate(GC::Ref<IDBDatabase> connection);
void dissociate(IDBDatabase& connection);
using AssociatedConnections = GC::HeapVector<GC::Ref<IDBDatabase>>;
GC::Ref<AssociatedConnections> associated_connections_as_heap_vector();
GC::Ref<AssociatedConnections> associated_connections_as_heap_vector_except(IDBDatabase& connection);
GC::RootVector<GC::Ref<IDBDatabase>> associated_connections_as_root_vector();
ReadonlySpan<GC::Ref<ObjectStore>> object_stores() { return m_object_stores; }
GC::Ptr<ObjectStore> object_store_with_name(String const& name) const;
void add_object_store(GC::Ref<ObjectStore> object_store) { m_object_stores.append(object_store); }
void remove_object_store(GC::Ref<ObjectStore> object_store)
{
m_object_stores.remove_first_matching([&](auto& entry) { return entry == object_store; });
}
[[nodiscard]] static Vector<GC::Weak<Database>> for_key(StorageAPI::StorageKey const&);
[[nodiscard]] static Optional<Database&> for_key_and_name(StorageAPI::StorageKey const&, String const&);
[[nodiscard]] static ErrorOr<GC::Ref<Database>> create_for_key_and_name(JS::Realm&, StorageAPI::StorageKey const&, String const&);
[[nodiscard]] static ErrorOr<void> delete_for_key_and_name(StorageAPI::StorageKey const&, String const&);
static void for_each_database(AK::Function<void(Database&)> const& visitor);
[[nodiscard]] static GC::Ref<Database> create(JS::Realm&, String const&);
virtual ~Database();
void wait_for_connections_to_close(ReadonlySpan<GC::Ref<IDBDatabase>> connections, GC::Ref<GC::Function<void()>> after_all);
void check_pending_connection_wait();
protected:
explicit Database(IDBDatabase& database);
explicit Database(JS::Realm& realm, String name)
: PlatformObject(realm)
, m_name(move(name))
{
}
virtual void visit_edges(Visitor&) override;
private:
struct PendingConnectionWait {
Vector<GC::Weak<IDBDatabase>> connections;
GC::Ref<GC::Function<void()>> callback;
};
Vector<GC::Weak<IDBDatabase>> m_associated_connections;
Optional<PendingConnectionWait> m_pending_connection_wait;
// A database has a name which identifies it within a specific storage key.
String m_name;
// A database has a version. When a database is first created, its version is 0 (zero).
u64 m_version { 0 };
// A database has at most one associated upgrade transaction, which is either null or an upgrade transaction, and is initially null.
GC::Ptr<IDBTransaction> m_upgrade_transaction;
// A database has zero or more object stores which hold the data stored in the database.
Vector<GC::Ref<ObjectStore>> m_object_stores;
};
}