Struct AsyncTracker 

pub struct AsyncTracker { /* private fields */ }

Async memory tracker for task-aware memory tracking.

Implementations

impl AsyncTracker

pub fn new() -> Self

pub fn set_current_task(task_id: u64)

pub fn clear_current_task()

pub fn get_current_task() -> Option<u64>

Get the current task ID from the thread-local storage.

Note: This only returns the manually set task ID. For automatic tokio task detection, use track_in_tokio_task().

pub fn enter_task(task_id: u64) -> TaskGuard

Enter a task context with automatic cleanup. Returns a TaskGuard that will clear the task ID when dropped.

pub fn with_task<F, T>(task_id: u64, f: F) -> T

where F: FnOnce() -> T,

Execute a closure within a task context with automatic cleanup. The task ID is automatically cleared after the closure completes, even if the closure panics.

pub fn track_task_start( &self, task_id: u64, name: String, _thread_id: ThreadId, ) -> Result<(), AsyncError>

pub fn track_task_start_with_tokio( &self, task_id: u64, tokio_task_id: u64, name: String, _thread_id: ThreadId, ) -> Result<(), AsyncError>

pub fn track_task_end(&self, task_id: u64) -> Result<(), AsyncError>

Track a task end.

pub async fn track_in_tokio_task<F, T>( &self, name: String, future: F, ) -> (u64, T)

where F: Future<Output = T>,

Execute an async block within a tokio task context.

This method automatically detects the tokio task ID and sets up tracking. When the future completes, the task is automatically marked as ended.

Arguments

• name - Task name for identification
• future - The async block to execute

Returns

A tuple of (unique_task_id, output). The unique_task_id is our internal ID (not the tokio task ID).

pub fn detect_zombie_tasks(&self) -> Vec<u64>

Detect zombie tasks.

A zombie task is a task that was started but never completed. These tasks may indicate memory leaks or improper task cleanup.

Returns

A vector of task IDs for zombie tasks.

pub fn zombie_task_stats(&self) -> (usize, usize)

Get statistics about zombie tasks.

Optimized to acquire the lock only once.

pub fn track_allocation_auto( &self, ptr: usize, size: usize, var_name: Option<String>, type_name: Option<String>, )

pub fn track_allocation(&self, ptr: usize, size: usize, task_id: u64)

Track an allocation associated with a task.

pub fn track_allocation_with_location( &self, ptr: usize, size: usize, task_id: u64, var_name: Option<String>, type_name: Option<String>, source_location: Option<SourceLocation>, )

Track an allocation with source location.

pub fn track_deallocation(&self, ptr: usize)

Track a deallocation associated with a task.

pub fn get_stats(&self) -> AsyncStats

Get current statistics.

pub fn snapshot(&self) -> AsyncSnapshot

Take a snapshot of current state.

pub fn get_task_profile(&self, task_id: u64) -> Option<TaskMemoryProfile>

Get task memory profile

pub fn get_all_profiles(&self) -> Vec<TaskMemoryProfile>

Get all task profiles

pub fn is_initialized(&self) -> bool

Check if tracker is initialized

pub fn set_initialized(&self)

Mark tracker as initialized

pub fn analyze_task( &self, task_id: u64, task_type: TaskType, ) -> Option<TaskReport>

Generate task efficiency report

pub fn get_resource_rankings(&self) -> Vec<ResourceRanking>

Get resource rankings for all tasks

Trait Implementations

impl Default for AsyncTracker

fn default() -> Self

Returns the “default value” for a type.

impl Drop for AsyncTracker

fn drop(&mut self)

Executes the destructor for this type.