Method writePartMetrics

src/Server/ServerPrometheusMetricsWriter.cpp:419–571

}

void ServerPrometheusMetricsWriter::writePartMetrics(WriteBuffer & wb)
{
    if (context->getServerType() == ServerType::cnch_server && send_part_metrics)
    {
        auto curr_ts = context->getTimestamp();
        auto cnch_catalog = context->getCnchCatalog();

        if (!cnch_catalog)
        {
            LOG_WARNING(&Poco::Logger::get("ServerPrometheusMetricsWriter"), "Cannot get catalog for part metrics");
        }
        else
        {
            auto db_models = cnch_catalog->getAllDataBases();

            std::string parts_num_key(PART_METRICS_PREFIX);
            parts_num_key.append(PARTS_NUMBER_LABEL);
            std::string parts_size_key(PART_METRICS_PREFIX);
            parts_size_key.append(PARTS_SIZE_LABEL);
            std::string rows_count_key(PART_METRICS_PREFIX);
            rows_count_key.append(ROWS_COUNT_LABEL);

            for (auto & db_model : db_models)
            {
                auto & db_name = db_model.name();

                auto tables = cnch_catalog->getTablesInDB(db_name);
                for (auto & table_name : tables)
                {
                    DB::StoragePtr storage = cnch_catalog->tryGetTable(*context, db_name, table_name, TxnTimestamp{context->getTimestamp()});
                    auto * cnch_table = dynamic_cast<StorageCnchMergeTree *>(storage.get());
                    if (!cnch_table)
                        continue;

                    Catalog::PartitionMap partitions;
                    cnch_catalog->getPartitionsFromMetastore(*cnch_table, partitions, nullptr);

                    for (auto & partition : partitions)
                    {
                        WriteBufferFromOwnString out;
                        partition.second->partition_ptr->serializeText(*cnch_table, out, format_settings);
                        auto partition_name = out.str();
                        auto partition_id = partition.first;

                        auto all_parts = cnch_catalog->getServerDataPartsInPartitions(storage, {partition_id}, curr_ts, nullptr);
                        auto visible_parts = CnchPartsHelper::calcVisibleParts(all_parts, false);
                        auto current_visible = visible_parts.cbegin();

                        /// Total parts
                        UInt64 total_parts_num = all_parts.size();
                        UInt64 total_parts_size = 0;
                        UInt64 total_rows_count = 0;
                        /// Visible parts
                        UInt64 visible_parts_num = visible_parts.size();
                        UInt64 visible_parts_size = 0;
                        UInt64 visible_rows_count = 0;
                        /// Invisible parts
                        UInt64 invisible_parts_num = total_parts_num - visible_parts_num;