adding several features

This commit is contained in:
DioCrafts
2025-03-19 00:44:27 +01:00
parent 7157762540
commit d9bbd575d2
66 changed files with 12137 additions and 1055 deletions
+1
View File
@@ -0,0 +1 @@
pub mod storage_transaction;
@@ -0,0 +1,131 @@
use std::future::Future;
use std::pin::Pin;
use crate::common::errors::{DomainError, ErrorKind};
/// Tipo para operaciones y rollbacks asíncronos
type TransactionOp = Pin<Box<dyn Future<Output = Result<(), DomainError>> + Send>>;
/// Transacción para operaciones de almacenamiento
/// Permite definir un conjunto de operaciones y sus rollbacks correspondientes
pub struct StorageTransaction {
/// Operaciones a ejecutar
operations: Vec<Box<dyn FnOnce() -> TransactionOp + Send>>,
/// Operaciones de rollback para revertir cambios en caso de error
rollbacks: Vec<Box<dyn FnOnce() -> TransactionOp + Send>>,
/// Nombre de la transacción para logging
name: String,
}
impl StorageTransaction {
/// Crea una nueva transacción
pub fn new(name: &str) -> Self {
Self {
operations: Vec::new(),
rollbacks: Vec::new(),
name: name.to_string(),
}
}
/// Añade una operación a la transacción con su correspondiente rollback
pub fn add_operation<F, R>(&mut self, operation: F, rollback: R)
where
F: Future<Output = Result<(), DomainError>> + Send + 'static,
R: Future<Output = Result<(), DomainError>> + Send + 'static,
{
self.operations.push(Box::new(move || Box::pin(operation)));
self.rollbacks.push(Box::new(move || Box::pin(rollback)));
}
/// Añade una operación sin rollback (para limpieza o logging)
#[allow(dead_code)]
pub fn add_finalizer<F>(&mut self, finalizer: F)
where
F: Future<Output = Result<(), DomainError>> + Send + 'static,
{
// El rollback es una operación nula
let noop = async { Ok(()) };
self.operations.push(Box::new(move || Box::pin(finalizer)));
self.rollbacks.push(Box::new(move || Box::pin(noop)));
}
/// Ejecuta la transacción aplicando todas las operaciones en orden
/// Si alguna falla, ejecuta los rollbacks en orden inverso
pub async fn commit(mut self) -> Result<(), DomainError> {
tracing::debug!("Iniciando transacción: {}", self.name);
let mut completed_ops = Vec::new();
// Extraer operaciones para evitar problemas de propiedad
let operations = std::mem::take(&mut self.operations);
let transaction_name = self.name.clone();
// Ejecutar operaciones
for (i, op) in operations.into_iter().enumerate() {
match op().await {
Ok(()) => {
completed_ops.push(i);
tracing::trace!("Operación {} completada en transacción: {}", i, transaction_name);
}
Err(e) => {
tracing::error!("Error en operación {} de transacción {}: {}", i, transaction_name, e);
// Ejecutar rollbacks para las operaciones completadas en orden inverso
self.rollback(completed_ops).await?;
return Err(DomainError::new(
ErrorKind::InternalError,
"Transaction",
format!("Falló la transacción '{}': {}", transaction_name, e)
).with_source(e));
}
}
}
tracing::debug!("Transacción completada exitosamente: {}", transaction_name);
Ok(())
}
/// Ejecuta rollbacks para las operaciones completadas
async fn rollback(mut self, completed_ops: Vec<usize>) -> Result<(), DomainError> {
tracing::warn!("Iniciando rollback para transacción: {}", self.name);
let mut rollback_errors = Vec::new();
// Extraer rollbacks para evitar problemas de propiedad
let mut rollbacks = Vec::new();
std::mem::swap(&mut rollbacks, &mut self.rollbacks);
// Ejecutar rollbacks en orden inverso
for i in completed_ops.into_iter().rev() {
if i < rollbacks.len() {
// Tomar propiedad del rollback (obtener una referencia mutable)
if let Some(rb) = rollbacks.get_mut(i) {
// Intercambiar con una función vacía
let rollback = std::mem::replace(rb, Box::new(|| Box::pin(async { Ok(()) })));
if let Err(e) = rollback().await {
tracing::error!("Error en rollback de operación {} en transacción {}: {}",
i, self.name, e);
rollback_errors.push(e);
}
}
}
}
// Si hubo errores en el rollback, reportarlos
if !rollback_errors.is_empty() {
tracing::error!("Errores durante rollback de transacción {}: {} errores",
self.name, rollback_errors.len());
return Err(DomainError::new(
ErrorKind::InternalError,
"Transaction",
format!("Errores durante rollback de transacción '{}': {} errores",
self.name, rollback_errors.len())
));
}
tracing::info!("Rollback de transacción completado: {}", self.name);
Ok(())
}
}