Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Thread-safe in-memory model registry with versioning and tagging.

Per measurement+name key, maintains an ordered list of versions. At most one version can be tagged Champion per key.

All methods take &self and acquire internal locks as needed, so the registry can be shared across threads via Arc<ModelRegistry>.

Iteration Order

The inner BTreeMap guarantees deterministic (sorted) iteration order, which is important for snapshot reproducibility and consistent API responses.

Implementations

impl ModelRegistry

pub fn new() -> Self

Create a new empty registry.

pub fn register( &self, measurement: &str, model_name: &str, algorithm: &str, hyperparameters: HashMap<String, String>, training_range: (i64, i64), model_bytes: Vec<u8>, ) -> ModelVersion

Register a new model version, auto-incrementing the version number.

Returns the registered ModelVersion (cloned, since the registry holds an internal lock).

pub fn get_latest( &self, measurement: &str, model_name: &str, ) -> Option<ModelVersion>

Get the latest version for a measurement+name.

Verifies model integrity on retrieval via SHA-256 hash. Uses borrowed key lookup to avoid String allocation.

pub fn get_version( &self, measurement: &str, model_name: &str, version: u64, ) -> Option<ModelVersion>

Get a specific version.

pub fn get_champion( &self, measurement: &str, model_name: &str, ) -> Option<ModelVersion>

Get the champion version for a measurement+name.

pub fn list_versions( &self, measurement: &str, model_name: &str, ) -> Vec<ModelVersion>

List all versions for a measurement+name.

pub fn set_tag( &self, measurement: &str, model_name: &str, version: u64, tag: ModelTag, ) -> bool

Tag a specific version. Ensures only one champion per key.

Returns false if the key or version doesn’t exist, without modifying any existing tags.

pub fn update_metrics( &self, measurement: &str, model_name: &str, version: u64, metrics: AccuracyMetrics, ) -> bool

Update accuracy metrics for a specific version.

pub fn rollback_champion( &self, measurement: &str, model_name: &str, version: u64, ) -> bool

Rollback: promote a previous version to champion, retiring the current one.

Convenience method for the common rollback workflow. Equivalent to set_tag(measurement, model_name, version, ModelTag::Champion) but logs the rollback event for auditability.

Returns false if the key or version doesn’t exist.

pub fn delete_version( &self, measurement: &str, model_name: &str, version: u64, ) -> bool

Delete a specific model version.

Allows removing bad versions from the registry. Refuses to delete the current champion — call set_tag or rollback_champion first to promote a different version.

Returns true if the version was found and deleted.

pub fn total_versions(&self) -> usize

Total number of registered model versions across all keys.

Trait Implementations

impl Debug for ModelRegistry

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

Formats the value using the given formatter.

impl Default for ModelRegistry

fn default() -> Self

Returns the “default value” for a type.