Struct ModelDownloadTracker 

pub struct ModelDownloadTracker { /* private fields */ }

Tracks the status of model downloads through the distributed registry backend.

Implementations

impl ModelDownloadTracker

pub fn new(registry: Arc<RegistryManager>) -> Self

pub async fn get_status(&self, model_name: &str) -> Option<ModelStatus>

Gets the status of a model from the registry, bumping last_used_at on hit. Returns None on lookup failure (error logged) or unknown model.

pub async fn set_status_and_notify( &self, model_name: String, status: ModelStatus, provider: ModelProvider, message: Option<String>, )

Sets the status of a model and notifies all waiting channels on this replica.

pub async fn set_status( &self, model_name: String, status: ModelStatus, provider: ModelProvider, )

Sets the status of a model (no message), notifying waiters.

pub fn add_waiting_channel( &self, model_name: &str, tx: Sender<Result<ModelStatusUpdate, Status>>, )

Adds a channel that wants updates on a specific model (server-replica-local).

pub async fn delete_status(&self, model_name: &str)

Deletes a model record from the registry and clears local waiters.

pub async fn ensure_model_downloaded( &self, model_name: &str, provider: ModelProvider, tx: &Sender<Result<ModelStatusUpdate, Status>>, ignore_weights: bool, ) -> ModelStatus

Initiates a download for a model and streams status updates.

Trait Implementations

impl Clone for ModelDownloadTracker

fn clone(&self) -> ModelDownloadTracker

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.