Struct BrickDataTracker 

pub struct BrickDataTracker { /* private fields */ }

Track where brick weights/data reside across workers

Used for locality-aware scheduling to minimize data movement.

Implementations

impl BrickDataTracker

pub fn new() -> Self

Create a new data tracker

pub fn track_data(&self, key: &str, worker_id: WorkerId, size_bytes: usize)

Register that a worker has certain data

pub fn track_weights(&self, brick_name: &str, worker_id: WorkerId)

Register that a worker has brick weights

pub fn remove_data(&self, key: &str, worker_id: WorkerId)

Remove data location from a worker

pub fn get_workers_for_data(&self, key: &str) -> Vec<WorkerId>

Get workers that have specific data

pub fn calculate_affinity( &self, dependencies: &[String], ) -> HashMap<WorkerId, f64>

Calculate affinity scores for workers based on data dependencies

pub fn find_best_worker(&self, brick: &dyn Brick) -> Option<WorkerId>

Find the best worker for a brick based on data locality

pub fn find_best_worker_for_distributed<B: Brick>( &self, brick: &DistributedBrick<B>, ) -> Option<WorkerId>

Find best worker for distributed brick with dependencies

pub fn total_data_size(&self) -> usize

Get total data size tracked

Trait Implementations

impl Debug for BrickDataTracker

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for BrickDataTracker

fn default() -> Self

Returns the “default value” for a type.