get_rbac_registrations.rs

//! Get RBAC registrations by Catalyst ID.

use std::sync::Arc;

use scylla::{
    prepared_statement::PreparedStatement, statement::Consistency,
    transport::iterator::TypedRowStream, DeserializeRow, SerializeRow, Session,
};
use tracing::error;

use crate::db::{
    index::{
        queries::{PreparedQueries, PreparedSelectQuery},
        session::CassandraSession,
    },
    types::{DbCatalystId, DbSlot, DbTransactionId, DbTxnIndex, DbUuidV4},
};

/// Get registrations by Catalyst ID query.
const QUERY: &str = include_str!("../cql/get_rbac_registrations_catalyst_id.cql");

/// Get registrations by Catalyst ID query params.
#[derive(SerializeRow)]
#[allow(dead_code)]
pub(crate) struct QueryParams {
    /// A Catalyst ID.
    pub catalyst_id: DbCatalystId,
}

/// Get registrations by Catalyst ID query.
#[allow(dead_code)]
#[derive(DeserializeRow)]
pub(crate) struct Query {
    /// Registration transaction id.
    pub txn_id: DbTransactionId,
    /// A block slot number.
    pub slot_no: DbSlot,
    /// A transaction index.
    pub txn_index: DbTxnIndex,
    /// A previous  transaction id.
    pub prv_txn_id: Option<DbTransactionId>,
    /// A registration purpose.
    pub purpose: DbUuidV4,
}

impl Query {
    /// Prepares a query.
    pub(crate) async fn prepare(session: Arc<Session>) -> anyhow::Result<PreparedStatement> {
        PreparedQueries::prepare(session, QUERY, Consistency::All, true)
            .await
            .inspect_err(
                |e| error!(error=%e, "Failed to prepare get RBAC registrations by Catalyst ID query"),
            )
    }

    /// Executes a get registrations by Catalyst ID query.
    #[allow(dead_code)]
    pub(crate) async fn execute(
        session: &CassandraSession, params: QueryParams,
    ) -> anyhow::Result<TypedRowStream<Query>> {
        session
            .execute_iter(PreparedSelectQuery::RbacRegistrationsByCatalystId, params)
            .await?
            .rows_stream::<Query>()
            .map_err(Into::into)
    }
}