From 733e8dfd4a95badadc6bee792b52d906d6b49981 Mon Sep 17 00:00:00 2001 From: akhercha Date: Fri, 1 Nov 2024 15:23:48 +0100 Subject: [PATCH] dev(better_theoros): Simpler struct --- rust/theoros/src/storage/updates.rs | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) 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()) } }