Minor improvements (clippy, API)

vercel · Jun 9, 2024 · a43455d · a43455d
1 parent 144e0be
commit a43455d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/crates/turbo-tasks-memory/src/memory_backend.rs b/crates/turbo-tasks-memory/src/memory_backend.rs
@@ -560,7 +560,7 @@ impl Backend for MemoryBackend {
                         .first()
                         .cloned()
                         .expect("No arguments for trait call");
-                    let stats = Arc::clone(&stats);
+                    let stats = Arc::clone(stats);
                     turbo_tasks.run_once(Box::pin(async move {
                         let function_id =
                             PersistentTaskType::resolve_trait_method(trait_type, name, this)

diff --git a/crates/turbo-tasks-memory/src/task_statistics.rs b/crates/turbo-tasks-memory/src/task_statistics.rs
@@ -11,11 +11,12 @@ pub struct TaskStatisticsApi {
 }
 
 impl TaskStatisticsApi {
-    pub fn enable(&self) {
+    pub fn enable(&self) -> &Arc<TaskStatistics> {
         // ignore error: potentially already initialized, that's okay
         let _ = self.inner.set(Arc::new(TaskStatistics {
             inner: DashMap::new(),
         }));
+        self.inner.get().unwrap()
     }
 
     pub fn is_enabled(&self) -> bool {
@@ -55,12 +56,7 @@ impl TaskStatistics {
         task_function_id: FunctionId,
         func: impl Fn(&mut TaskFunctionStatistics),
     ) {
-        func(
-            self.inner
-                .entry(task_function_id)
-                .or_insert(TaskFunctionStatistics::default())
-                .value_mut(),
-        )
+        func(self.inner.entry(task_function_id).or_default().value_mut())
     }
 }