diff --git a/rust/theoros/src/storage/updates.rs b/rust/theoros/src/storage/updates.rs index 4dba8c2..a9e35ac 100644 --- a/rust/theoros/src/storage/updates.rs +++ b/rust/theoros/src/storage/updates.rs @@ -8,21 +8,19 @@ use crate::types::hyperlane::DispatchUpdateInfos; /// Contains a mapping between feed ids and their latest dispatch update. #[derive(Debug, Default)] -pub struct LatestUpdatePerFeedStorage { - events: RwLock>, -} +pub struct LatestUpdatePerFeedStorage(RwLock>); impl LatestUpdatePerFeedStorage { /// Insert the latest [`DispatchUpdateInfos`] for a feed id. pub async fn add(&self, feed_id: U256, event: DispatchUpdateInfos) -> Result<()> { - let mut events = self.events.write().await; + let mut events = self.0.write().await; events.insert(feed_id, event); Ok(()) } /// Retrieves the latest [`DispatchUpdateInfos`] for a feed id. pub async fn get(&self, feed_id: &U256) -> Result> { - let events = self.events.read().await; + let events = self.0.read().await; Ok(events.get(feed_id).cloned()) } }