poke_data/collections/
versions.rs1use crate::data::link_context::LinkContext;
2use crate::data_structures::entity_collection::{EntityCollection, HasNameSearchIndex};
3use crate::data_structures::string_search_index::StringSearchIndex;
4use crate::models::version::{Version, VersionId};
5use crate::traits::has_identifier::IdentifierDictionary;
6use crate::traits::has_localized_names::LocalizedNamesDictionary;
7use std::collections::HashMap;
8use std::sync::Arc;
9
10pub struct VersionsCollection {
11 entities: HashMap<VersionId, Arc<Version>>,
12 name_search_index: StringSearchIndex<VersionId>,
13}
14
15impl EntityCollection<VersionId, Version> for VersionsCollection {
16 fn new(context: &LinkContext) -> Self {
17 let entities = context.versions.clone();
18 let mut dictionary = HashMap::new();
19 dictionary.extend(entities.build_identifier_dictionary());
20 dictionary.extend(entities.build_localized_name_dictionary());
21 let name_search_index = StringSearchIndex::new(dictionary);
22 Self {
23 entities,
24 name_search_index,
25 }
26 }
27
28 fn entities(&self) -> &HashMap<VersionId, Arc<Version>> {
29 &self.entities
30 }
31}
32
33impl HasNameSearchIndex<VersionId, Version> for VersionsCollection {
34 fn name_search_index(&self) -> &StringSearchIndex<VersionId> {
35 &self.name_search_index
36 }
37}