metrics.rs

use prometheus::{register_int_counter, register_int_gauge, IntCounter, IntGauge};
use sqlx::PgPool;
use std::sync::LazyLock;
use tokio::{task::JoinHandle, time::sleep};
use tracing::{error, info};

pub(crate) static VERIFY_PROOF_SUCCESS_COUNTER: LazyLock<IntCounter> = LazyLock::new(|| {
    register_int_counter!(
        "coprocessor_txn_sender_verify_proof_success_counter",
        "Number of successful verify or reject proof txns in transaction-sender"
    )
    .unwrap()
});

pub(crate) static VERIFY_PROOF_FAIL_COUNTER: LazyLock<IntCounter> = LazyLock::new(|| {
    register_int_counter!(
        "coprocessor_txn_sender_verify_proof_fail_counter",
        "Number of failed verify or reject proof txns requests in transaction-sender"
    )
    .unwrap()
});

pub(crate) static ADD_CIPHERTEXT_MATERIAL_SUCCESS_COUNTER: LazyLock<IntCounter> =
    LazyLock::new(|| {
        register_int_counter!(
            "coprocessor_txn_sender_add_ciphertext_material_success_counter",
            "Number of successful add ciphertext material txns in transaction-sender"
        )
        .unwrap()
    });

pub(crate) static ADD_CIPHERTEXT_MATERIAL_FAIL_COUNTER: LazyLock<IntCounter> =
    LazyLock::new(|| {
        register_int_counter!(
            "coprocessor_txn_sender_add_ciphertext_material_fail_counter",
            "Number of failed add ciphertext material txns requests in transaction-sender"
        )
        .unwrap()
    });

pub(crate) static ADD_CIPHERTEXT_MATERIAL_UNSENT: LazyLock<IntGauge> = LazyLock::new(|| {
    register_int_gauge!(
        "coprocessor_add_ciphertext_material_unsent_gauge",
        "Number of unsent add ciphertext material transactions"
    )
    .unwrap()
});

pub(crate) static VERIFY_PROOF_RESP_UNSENT_TXN: LazyLock<IntGauge> = LazyLock::new(|| {
    register_int_gauge!(
        "coprocessor_verify_proof_resp_unsent_txn_gauge",
        "Number of unsent verify proof response transactions"
    )
    .unwrap()
});

pub(crate) static VERIFY_PROOF_PENDING: LazyLock<IntGauge> = LazyLock::new(|| {
    register_int_gauge!(
        "coprocessor_verify_proof_pending_gauge",
        "Number of pending verify proof requests"
    )
    .unwrap()
});

pub fn spawn_gauge_update_routine(period: std::time::Duration, db_pool: PgPool) -> JoinHandle<()> {
    tokio::spawn(async move {
        loop {
            match sqlx::query_scalar(
                "SELECT COUNT(*) FROM ciphertext_digest WHERE txn_is_sent = FALSE",
            )
            .fetch_one(&db_pool)
            .await
            {
                Ok(count) => {
                    info!(unsent_add_ciphertext_material_count = %count, "Fetched unsent add ciphertext material count");
                    ADD_CIPHERTEXT_MATERIAL_UNSENT.set(count);
                }
                Err(e) => {
                    error!(error = %e, "Failed to fetch unsent add ciphertext material count");
                }
            }

            match sqlx::query_scalar("SELECT COUNT(*) FROM verify_proofs WHERE verified IS NULL")
                .fetch_one(&db_pool)
                .await
            {
                Ok(count) => {
                    info!(verify_proof_pending = %count, "Fetched pending verify proofs count");
                    VERIFY_PROOF_PENDING.set(count);
                }
                Err(e) => {
                    error!(error = %e, "Failed to fetch pending verify proofs count");
                }
            }

            match sqlx::query_scalar(
                "SELECT COUNT(*) FROM verify_proofs WHERE verified IS NOT NULL",
            )
            .fetch_one(&db_pool)
            .await
            {
                Ok(count) => {
                    info!(verify_proof_resp_unsent_txn = %count, "Fetched unsent verify proof response count");
                    VERIFY_PROOF_RESP_UNSENT_TXN.set(count);
                }
                Err(e) => {
                    error!(error = %e, "Failed to fetch unsent verify proof response count");
                }
            }

            sleep(period).await;
        }
    })
}