diff --git a/src/common/procedure/src/local/runner.rs b/src/common/procedure/src/local/runner.rs index 1d1420f432b1..6509ee8fc0d1 100644 --- a/src/common/procedure/src/local/runner.rs +++ b/src/common/procedure/src/local/runner.rs @@ -1105,7 +1105,6 @@ mod tests { #[tokio::test] async fn test_rollback_exceed_max_retry_later() { - common_telemetry::init_default_ut_logging(); let exec_fn = |_| async { Err(Error::retry_later(MockError::new(StatusCode::Unexpected))) }.boxed(); let rollback_fn = move |_| { diff --git a/src/common/procedure/src/procedure.rs b/src/common/procedure/src/procedure.rs index d2d4a5b5f569..9c43aeb520da 100644 --- a/src/common/procedure/src/procedure.rs +++ b/src/common/procedure/src/procedure.rs @@ -159,6 +159,10 @@ impl Procedure for Box { (**self).execute(ctx).await } + async fn rollback(&mut self, ctx: &Context) -> Result<()> { + (**self).rollback(ctx).await + } + fn rollback_supported(&self) -> bool { (**self).rollback_supported() }