use std::collections::HashSet;
use icu::experimental::measure::provider::UnitIdsV1;
use icu_provider::prelude::*;
use icu_provider::DataError;
use icu_provider::DataMarkerAttributes;
use icu_provider::DataProvider;
use icu_provider::DataRequest;
use icu_provider::DataResponse;
use crate::cldr_serde;
use crate::SourceDataProvider;
impl DataProvider<UnitIdsV1> for SourceDataProvider {
fn load(&self, req: DataRequest) -> Result<DataResponse<UnitIdsV1>, DataError> {
self.check_req::<UnitIdsV1>(req)?;
let units_data: &cldr_serde::units::info::Resource = self
.cldr()?
.core()
.read_and_parse("supplemental/units.json")?;
let unit = req.id.marker_attributes.as_str();
Ok(DataResponse {
payload: DataPayload::from_owned(units_data.unit_id(unit)?),
metadata: Default::default(),
})
}
}
impl crate::IterableDataProviderCached<UnitIdsV1> for SourceDataProvider {
fn iter_ids_cached(&self) -> Result<HashSet<DataIdentifierCow<'static>>, DataError> {
let units_data: &cldr_serde::units::info::Resource = self
.cldr()?
.core()
.read_and_parse("supplemental/units.json")?;
let ids_set = units_data
.unit_ids_map()?
.keys()
.map(|unit_name| {
DataIdentifierCow::from_marker_attributes_owned(
DataMarkerAttributes::try_from_string(unit_name.clone()).unwrap(),
)
})
.collect();
Ok(ids_set)
}
}