use std::collections::HashMap;
use serde::{Deserialize, Serialize};
use crate::key::KeyDescription;
use crate::schema::view::ViewUpdatePolicy;
use crate::schema::{CollectionName, SchemaName, Schematic, ViewName};
#[derive(Debug, Eq, PartialEq, Serialize, Deserialize, Clone)]
pub struct SchemaSummary {
pub name: SchemaName,
collections: HashMap<CollectionName, CollectionSummary>,
}
impl SchemaSummary {
#[must_use]
pub fn collection(&self, name: &CollectionName) -> Option<&CollectionSummary> {
self.collections.get(name)
}
pub fn collections(&self) -> impl Iterator<Item = &CollectionSummary> {
self.collections.values()
}
}
impl<'a> From<&'a Schematic> for SchemaSummary {
fn from(schematic: &'a Schematic) -> Self {
let mut summary = Self {
name: schematic.name.clone(),
collections: HashMap::new(),
};
for collection_name in schematic.collections() {
let collection = summary
.collections
.entry(collection_name.clone())
.or_insert_with(|| CollectionSummary {
name: collection_name.clone(),
primary_key: schematic
.collection_primary_key_description(collection_name)
.expect("invalid schematic")
.clone(),
views: HashMap::new(),
});
for view in schematic.views_in_collection(collection_name) {
let name = view.view_name();
collection.views.insert(
name.clone(),
ViewSummary {
name,
key: view.key_description(),
policy: view.update_policy(),
version: view.version(),
},
);
}
}
summary
}
}
#[derive(Debug, Eq, PartialEq, Serialize, Deserialize, Clone)]
pub struct CollectionSummary {
pub name: CollectionName,
pub primary_key: KeyDescription,
views: HashMap<ViewName, ViewSummary>,
}
impl CollectionSummary {
#[must_use]
pub fn view(&self, name: &ViewName) -> Option<&ViewSummary> {
self.views.get(name)
}
pub fn views(&self) -> impl Iterator<Item = &ViewSummary> {
self.views.values()
}
}
#[derive(Debug, Eq, PartialEq, Serialize, Deserialize, Clone)]
pub struct ViewSummary {
pub name: ViewName,
pub key: KeyDescription,
pub policy: ViewUpdatePolicy,
pub version: u64,
}