Function collectDiskSizeInfo

src/function/table/disk_size_info.cpp:138–259

}

static std::vector<DiskSizeEntry> collectDiskSizeInfo(const ClientContext* context) {
    std::vector<DiskSizeEntry> entries;
    std::unordered_map<std::string, idx_t> entryLookup;
    auto storageManager = StorageManager::Get(*context);
    auto catalog = Catalog::Get(*context);
    auto dataFH = storageManager->getDataFH();

    // Handle in-memory databases
    if (storageManager->isInMemory()) {
        entries.push_back({"info", "in_memory_database", 0, 0});
        return entries;
    }

    auto pageManager = dataFH->getPageManager();

    // 1. Database header (always 1 page at index 0)
    entries.push_back({"header", "database_header", 1, LBUG_PAGE_SIZE});

    // 2. Get catalog and metadata page ranges from database header
    auto databaseHeader = DatabaseHeader::readDatabaseHeader(*dataFH->getFileInfo());
    if (databaseHeader.has_value()) {
        entries.push_back({"catalog", "catalog", databaseHeader->catalogPageRange.numPages,
            databaseHeader->catalogPageRange.numPages * LBUG_PAGE_SIZE});

        entries.push_back({"metadata", "metadata", databaseHeader->metadataPageRange.numPages,
            databaseHeader->metadataPageRange.numPages * LBUG_PAGE_SIZE});
    }

    // 3. Count table data pages
    auto nodeTableEntries =
        catalog->getNodeTableEntries(&transaction::DUMMY_CHECKPOINT_TRANSACTION);
    auto relGroupEntries = catalog->getRelGroupEntries(&transaction::DUMMY_CHECKPOINT_TRANSACTION);

    for (const auto tableEntry : nodeTableEntries) {
        auto& nodeTable = storageManager->getTable(tableEntry->getTableID())->cast<NodeTable>();
        uint64_t tablePages = 0;
        auto numNodeGroups = nodeTable.getNumNodeGroups();
        for (auto i = 0ul; i < numNodeGroups; i++) {
            tablePages += countNodeGroupPages(nodeTable.getNodeGroup(i));
        }
        entries.push_back(
            {"node_table", tableEntry->getName(), tablePages, tablePages * LBUG_PAGE_SIZE});

        for (const auto& indexHolder : nodeTable.getIndexes()) {
            if (!indexHolder.isLoaded()) {
                continue;
            }
            const auto* index = indexHolder.getIndex();
            const auto indexInfo = index->getIndexInfo();
            for (const auto& storageEntry : index->getStorageEntries()) {
                if (storageEntry.pageRange.startPageIdx == INVALID_PAGE_IDX ||
                    storageEntry.pageRange.numPages == 0) {
                    continue;
                }
                appendOrIncrementEntry(entries, entryLookup, "index",
                    tableEntry->getName() + "." + indexInfo.name + ":" + storageEntry.component,
                    storageEntry.pageRange.numPages);
            }