riftenlabs-indexer/src/db/cauldron/mempool.rs

// Copyright (C) 2024-2026 Whiterun LLC
//
// This software is licensed under the GNU Affero General Public License (AGPL), version 3.0 or later.
// A copy of the license can be found in the LICENSE file or at https://www.gnu.org/licenses/agpl-3.0.html

use std::collections::HashSet;

use anyhow::Result;
use bitcoincash::Txid;
use rusqlite::Connection;

use crate::db::blob::{FromBlob, ToBlob};

pub fn load_mempool(conn: &Connection) -> Result<HashSet<Txid>> {
    let mut stmt = conn.prepare("SELECT txid FROM tx WHERE blockhash is NULL")?;
    let txid_iter = stmt.query_map([], |row| row.get(0))?;

    let mut txids: HashSet<Txid> = HashSet::new();
    for txid_res in txid_iter {
        let txid_blob: Vec<u8> = txid_res?;
        let txid = Txid::from_blob(&txid_blob).expect("invalid txid in db");
        txids.insert(txid);
    }
    Ok(txids)
}

pub fn delete_mempool_txs<'a, I>(db_tx: &Connection, txids: I) -> Result<bool>
where
    I: IntoIterator<Item = &'a Txid>,
{
    let txid_blobs: Vec<Vec<u8>> = txids.into_iter().map(|txid| txid.to_blob()).collect();
    let placeholders = txid_blobs
        .iter()
        .map(|_| "?")
        .collect::<Vec<_>>()
        .join(", ");
    let query = format!("DELETE FROM tx WHERE txid IN ({placeholders}) AND blockhash is NULL");

    let params: Vec<&dyn rusqlite::ToSql> = txid_blobs
        .iter()
        .map(|s| s as &dyn rusqlite::ToSql)
        .collect();

    let mut stmt = db_tx.prepare(&query)?;
    let rows_deleted = stmt.execute(rusqlite::params_from_iter(params))?;

    Ok(rows_deleted != 0)
}