functions on Entry instead of match

cowprotocol · Dec 20, 2024 · ade2360 · ade2360
1 parent e7530af
commit ade2360
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/crates/driver/src/domain/competition/bad_tokens/cache.rs b/crates/driver/src/domain/competition/bad_tokens/cache.rs
@@ -40,9 +40,10 @@ impl Cache {
 
     /// Updates whether or not a token should be considered supported.
     pub fn update_quality(&self, token: eth::TokenAddress, quality: Quality, now: Instant) {
-        match self.0.cache.entry(token) {
-            Entry::Occupied(mut o) => {
-                let value = o.get_mut();
+        self.0
+            .cache
+            .entry(token)
+            .and_modify(|value| {
                 if quality == Quality::Unsupported
                     || now.duration_since(value.timestamp) > self.0.max_age
                 {
@@ -52,14 +53,11 @@ impl Cache {
                     value.quality = quality;
                 }
                 value.timestamp = now;
-            }
-            Entry::Vacant(v) => {
-                v.insert(CacheEntry {
-                    quality,
-                    timestamp: now,
-                });
-            }
-        }
+            })
+            .or_insert_with(|| CacheEntry {
+                quality,
+                timestamp: now,
+            });
     }
 
     pub fn evict_outdated_entries(&self) {