optimize code

GreptimeTeam · Sep 4, 2024 · ebbcba2 · ebbcba2
1 parent 8e9ce54
commit ebbcba2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/common/procedure/src/local/runner.rs b/src/common/procedure/src/local/runner.rs
@@ -1104,7 +1104,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
@@ -159,6 +159,10 @@ impl<T: Procedure + ?Sized> Procedure for Box<T> {
         (**self).execute(ctx).await
     }
 
+    async fn rollback(&mut self, ctx: &Context) -> Result<()> {
+        (**self).rollback(ctx).await
+    }
+
     fn rollback_supported(&self) -> bool {
         (**self).rollback_supported()
     }