fix(torii-sqlite): erc721 tokens check both cache + db for write

dojoengine · Feb 3, 2025 · b7ac26d · b7ac26d
1 parent e823d42
commit b7ac26d
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/crates/torii/sqlite/src/executor/mod.rs b/crates/torii/sqlite/src/executor/mod.rs
@@ -620,8 +620,19 @@ impl<'c, P: Provider + Sync + Send + 'static> Executor<'c, P> {
             QueryType::RegisterErc721Token(register_erc721_token) => {
                 let semaphore = self.semaphore.clone();
                 let provider = self.provider.clone();
+
+                let token_exists =
+                    sqlx::query_scalar::<_, i64>("SELECT COUNT(*) FROM tokens WHERE id = ?")
+                        .bind(register_erc721_token.id.to_string())
+                        .fetch_one(&mut **tx)
+                        .await?;
+
+                if token_exists > 0 {
+                    return Ok(());
+                }
+
                 let res = sqlx::query_as::<_, (String, String)>(&format!(
-                    "SELECT name, symbol FROM {TOKENS_TABLE} WHERE contract_address = ? LIMIT 1"
+                    "SELECT name, symbol FROM {TOKENS_TABLE} WHERE id = ? LIMIT 1"
                 ))
                 .bind(felt_to_sql_string(&register_erc721_token.contract_address))
                 .fetch_one(&mut **tx)