extract InstanceEnv::console_timer_end

Author: Centril

crates/core/src/host/instance_env.rs

@@ -2,6 +2,7 @@ use super::scheduler::{get_schedule_from_row, ScheduleError, Scheduler};
 use crate::database_logger::{BacktraceProvider, LogLevel, Record};
 use crate::db::relational_db::{MutTx, RelationalDB};
 use crate::error::{DBError, DatastoreError, IndexError, NodesError};
+use crate::host::wasm_common::TimingSpan;
 use crate::replica_context::ReplicaContext;
 use core::mem;
 use parking_lot::{Mutex, MutexGuard};
@@ -189,6 +190,22 @@ impl InstanceEnv {
         );
     }
 
+    /// End a console timer by logging the span at INFO level.
+    pub fn console_timer_end(&self, span: &TimingSpan, function: Option<&str>, bt: &dyn BacktraceProvider) -> RtResult<u32> {
+        let elapsed = span.start.elapsed();
+        let message = format!("Timing span {:?}: {:?}", &span.name, elapsed);
+
+        let record = Record {
+            ts: chrono::Utc::now(),
+            target: None,
+            filename: None,
+            line_number: None,
+            function,
+            message: &message,
+        };
+        self.console_log(LogLevel::Info, &record, bt);
+    }
+
     /// Project `cols` in `row_ref` encoded in BSATN to `buffer`
     /// and return the full length of the BSATN.
     ///

crates/core/src/host/wasmtime/wasm_instance_env.rs

@@ -1199,23 +1199,11 @@ impl WasmInstanceEnv {
                 return Ok(errno::NO_SUCH_CONSOLE_TIMER.get().into());
             };
 
-            let elapsed = span.start.elapsed();
-            let message = format!("Timing span {:?}: {:?}", &span.name, elapsed);
             let function = caller.data().log_record_function();
-
-            let record = Record {
-                ts: chrono::Utc::now(),
-                target: None,
-                filename: None,
-                line_number: None,
-                function,
-                message: &message,
-            };
-            caller.data().instance_env.console_log(
-                crate::database_logger::LogLevel::Info,
-                &record,
-                &caller.as_context(),
-            );
+            caller
+                .data()
+                .instance_env
+                .console_timer_end(&span, function, &caller.as_context());
             Ok(0)
         })
     }