fix: use index for value_for_key query

ceramicnetwork · Jan 19, 2024 · c8b8902 · c8b8902
1 parent 212504f
commit c8b8902
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/one/src/lib.rs b/one/src/lib.rs
@@ -5,7 +5,6 @@ mod events;
 mod http;
 mod metrics;
 mod network;
-mod sql;
 
 use std::{env, num::NonZeroUsize, path::PathBuf, time::Duration};
 

diff --git a/recon/src/recon/sqlitestore.rs b/recon/src/recon/sqlitestore.rs
@@ -366,7 +366,7 @@ where
             .bind(value)
             .bind(&self.sort_key)
             .bind(key.as_bytes())
-            .fetch_all(&self.pool)
+            .fetch_all(self.pool.writer())
             .await;
         match resp {
             Ok(_) => Ok(true),
@@ -376,9 +376,13 @@ where
 
     #[instrument(skip(self))]
     async fn value_for_key(&mut self, key: &Self::Key) -> Result<Option<Vec<u8>>> {
-        let query = sqlx::query("SELECT value FROM recon WHERE key=?;");
-        let rows = query.bind(key.as_bytes()).fetch_all(&self.pool).await?;
-        if let Some(row) = rows.first() {
+        let query = sqlx::query("SELECT value FROM recon WHERE sort_key=? AND key=?;");
+        let row = query
+            .bind(&self.sort_key)
+            .bind(key.as_bytes())
+            .fetch_optional(self.pool.reader())
+            .await?;
+        if let Some(row) = row {
             let value: Vec<u8> = row.get(0);
             Ok(Some(value))
         } else {