feat(cat-gateway): impl Query for UpdateTxoSpentQuery

Author: saibatizoku

catalyst-gateway/bin/src/db/index/queries/mod.rs

@@ -250,7 +250,8 @@ async fn prepare_queries(
         StakeRegistrationInsertQuery,
         Cip36Insert,
         Cip36InvalidInsert,
-        Cip36ForVoteKeyInsert
+        Cip36ForVoteKeyInsert,
+        UpdateTxoSpentQuery
     );
     Ok(queries)
 }
@@ -276,8 +277,7 @@ impl PreparedQueries {
             cip36_registration_error_insert_queries,
             cip36_registration_for_vote_key_insert_queries,
         ) = Cip36InsertQuery::prepare_batch(&session, cfg).await?;
-        let txo_spent_update_queries =
-            UpdateTxoSpentQuery::prepare_batch(session.clone(), cfg).await?;
+        let txo_spent_update_queries = UpdateTxoSpentQuery::prepare_batch(&session, cfg).await?;
         let txo_by_stake_address_query = GetTxoByStakeAddressQuery::prepare(session.clone()).await;
         let txi_by_txn_hash_query = GetTxiByTxnHashesQuery::prepare(session.clone()).await;
         let (

catalyst-gateway/bin/src/db/index/queries/staked_ada/update_txo_spent.rs

@@ -1,14 +1,16 @@
 //! Update the TXO Spent column to optimize future queries.
 
-use std::sync::Arc;
+use std::{fmt, sync::Arc};
 
 use scylla::{SerializeRow, Session};
 use tracing::error;
 
 use crate::{
     db::{
         index::{
-            queries::{FallibleQueryResults, PreparedQueries, PreparedQuery, SizedBatch},
+            queries::{
+                FallibleQueryResults, PreparedQueries, PreparedQuery, Query, QueryKind, SizedBatch,
+            },
             session::CassandraSession,
         },
         types::{DbSlot, DbStakeAddress, DbTxnIndex, DbTxnOutputOffset},
@@ -37,10 +39,27 @@ pub(crate) struct UpdateTxoSpentQueryParams {
 /// Update TXO spent query.
 pub(crate) struct UpdateTxoSpentQuery;
 
+impl Query for UpdateTxoSpentQuery {
+    /// Prepare Batch of Insert TXI Index Data Queries
+    async fn prepare_query(
+        session: &Arc<Session>, cfg: &cassandra_db::EnvVars,
+    ) -> anyhow::Result<QueryKind> {
+        Self::prepare_batch(session, cfg)
+            .await
+            .map(QueryKind::Batch)
+    }
+}
+
+impl fmt::Display for UpdateTxoSpentQuery {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{UPDATE_TXO_SPENT_QUERY}")
+    }
+}
+
 impl UpdateTxoSpentQuery {
     /// Prepare a batch of update TXO spent queries.
     pub(crate) async fn prepare_batch(
-        session: Arc<Session>, cfg: &cassandra_db::EnvVars,
+        session: &Arc<Session>, cfg: &cassandra_db::EnvVars,
     ) -> anyhow::Result<SizedBatch> {
         PreparedQueries::prepare_batch(
             session.clone(),