wip

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

common/src/tap/checks/value_check.rs

@@ -58,6 +58,19 @@ impl MinimumValue {
         }
     }
 
+    fn get_expected_value(&self, agora_query: &AgoraQuery) -> anyhow::Result<u128> {
+        // get agora model for the allocation_id
+        let mut cache = self.cost_model_cache.lock().unwrap();
+        // on average, we'll have zero or one model
+        let models = cache.get_mut(&agora_query.deployment_id);
+
+        let expected_value = models
+            .map(|cache| cache.cost(agora_query))
+            .unwrap_or_default();
+
+        Ok(expected_value)
+    }
+
     async fn cost_models_watcher(
         _pgpool: PgPool,
         mut pglistener: PgListener,
@@ -131,20 +144,26 @@ impl MinimumValue {
             }
         }
     }
-}
-
-impl MinimumValue {
-    fn get_expected_value(&self, agora_query: &AgoraQuery) -> anyhow::Result<u128> {
-        // get agora model for the allocation_id
-        let mut cache = self.cost_model_cache.lock().unwrap();
-        // on average, we'll have zero or one model
-        let models = cache.get_mut(&agora_query.deployment_id);
 
-        let expected_value = models
-            .map(|cache| cache.cost(agora_query))
-            .unwrap_or_default();
+    async fn sender_denylist_reload(
+        pgpool: PgPool,
+        cost_model_cache: Arc<Mutex<HashMap<DeploymentId, CostModelCache>>>,
+    ) -> anyhow::Result<()> {
+        // Fetch the denylist from the DB
+        let sender_denylist = sqlx::query!(
+            r#"
+                SELECT sender_address FROM scalar_tap_denylist
+            "#
+        )
+        .fetch_all(&pgpool)
+        .await?
+        .iter()
+        .map(|row| Address::from_str(&row.sender_address))
+        .collect::<Result<HashSet<_>, _>>()?;
+
+        *(denylist_rwlock.write().unwrap()) = sender_denylist;
 
-        Ok(expected_value)
+        Ok(())
     }
 }
 
@@ -279,3 +298,23 @@ impl CostModelCache {
             .unwrap_or_default()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use sqlx::PgPool;
+
+    #[sqlx::test(migrations = "../migrations")]
+    async fn initialize_check(pg_pool: PgPool) {}
+
+    #[sqlx::test(migrations = "../migrations")]
+    async fn should_initialize_check_with_caches(pg_pool: PgPool) {}
+
+    #[sqlx::test(migrations = "../migrations")]
+    async fn should_add_model_to_cache_on_insert(pg_pool: PgPool) {}
+
+    #[sqlx::test(migrations = "../migrations")]
+    async fn should_expire_old_model(pg_pool: PgPool) {}
+
+    #[sqlx::test(migrations = "../migrations")]
+    async fn should_verify_global_model(pg_pool: PgPool) {}
+}