Feat: Add clear old txs cron

src/cron/clear_transactions.rs

@@ -0,0 +1,14 @@
+use crate::{database::{queries::{QueryContext, delete_txs}}};
+
+use super::CronJobError;
+use crate::cron::ValidatorCronError;
+
+pub async fn clear_old_transactions<Context>(ctx: &Context) -> Result<(), CronJobError>
+where
+  Context: QueryContext 
+{
+  let epoch = ctx.current_epoch();
+  delete_txs(ctx, epoch, 40).await
+    .map(|amount| print!( "Deleted {} transactions from epoch {} to {}", amount, epoch - 40, epoch))
+    .map_err(|err| CronJobError::ValidatorError(ValidatorCronError::from(err)))
+}

src/cron/error.rs

@@ -21,3 +21,9 @@ impl From<anyhow::Error> for ValidatorCronError {
         ValidatorCronError::AddressNotFound
     }
 }
+
+impl From<diesel::result::Error> for ValidatorCronError {
+    fn from(_err: diesel::result::Error) -> ValidatorCronError {
+        ValidatorCronError::TxNotFound
+    }
+}

src/cron/mod.rs

@@ -5,6 +5,7 @@ mod error;
 mod slasher;
 mod transactions;
 mod validate;
+mod clear_transactions;
 
 use crate::{
     context,
@@ -56,6 +57,12 @@ where
             validate::validate_transactions,
             30
         ),
+        create_cron(
+            &ctx,
+            "clear old transactions",
+            clear_transactions::clear_old_transactions,
+            180
+        )
     );
 }
 

src/database/queries.rs

@@ -10,6 +10,8 @@ use crate::database::schema::transactions::dsl::*;
 use crate::database::schema::{bundle, transactions};
 use crate::state::ValidatorStateAccess;
 
+use super::models::Epoch;
+
 pub trait QueryContext: ValidatorStateAccess {
     fn get_db_connection(&self) -> PooledConnection<ConnectionManager<PgConnection>>;
     fn current_epoch(&self) -> u128;
@@ -80,3 +82,15 @@ where
         .filter(transactions::id.eq(tx_id))
         .first::<Transaction>(&conn)
 }
+
+pub async fn delete_txs<Context>(ctx: &Context, current_epoch: u128, epoch_amount: u128) -> Result<usize, Error>
+where
+    Context: QueryContext,
+{
+    let epochs : Vec<Epoch> = (0..epoch_amount).map(|i| Epoch(current_epoch - i)).collect();
+    let conn = ctx.get_db_connection();
+    let txs = transactions
+        .filter(transactions::epoch.ne_all(epochs));
+    diesel::delete(txs)
+        .execute(&conn)
+}