Add index for cache

Author: linyinfeng

src/repo/cache.rs

@@ -6,17 +6,22 @@ use crate::error::Error;
 
 pub fn initialize(cache: &Connection) -> Result<(), Error> {
     cache.execute(
-        "CREATE TABLE commits_cache (
-               branch        TEXT    NOT NULL,
-               commit_hash   TEXT    NOT NULL
-             )",
+        "CREATE TABLE IF NOT EXISTS commits_cache (
+            branch        TEXT    NOT NULL,
+            commit_hash   TEXT    NOT NULL
+        )",
         [],
     )?;
     cache.execute(
-        "CREATE TABLE branches (
-               branch           TEXT    NOT NULL PRIMARY KEY,
-               current_commit   TEXT    NOT NULL
-             )",
+        "CREATE TABLE IF NOT EXISTS branches (
+            branch           TEXT    NOT NULL PRIMARY KEY,
+            current_commit   TEXT    NOT NULL
+        )",
+        [],
+    )?;
+    cache.execute(
+        "CREATE INDEX IF NOT EXISTS idx_commit_branches
+         ON commits_cache (commit_hash)",
         [],
     )?;
 

src/repo/resources.rs

@@ -32,16 +32,13 @@ impl Resource<String> for RepoResources {
         // load repo
         let repo = Mutex::new(Repository::open(&paths.repo)?);
         // load cache
-        let cache_exists = paths.cache.is_file();
         let cache_cfg = deadpool_sqlite::Config::new(&paths.cache);
         let cache = cache_cfg.create_pool(deadpool_sqlite::Runtime::Tokio1)?;
-        if !cache_exists {
-            log::debug!("initializing cache for {name}...");
-            let conn = cache.get().await?;
-            conn.interact(|c| cache::initialize(c))
-                .await
-                .map_err(|e| Error::DBInteract(Mutex::new(e)))??;
-        }
+        log::debug!("initializing cache for {name}...");
+        let conn = cache.get().await?;
+        conn.interact(|c| cache::initialize(c))
+            .await
+            .map_err(|e| Error::DBInteract(Mutex::new(e)))??;
         // load settings
         let settings = RwLock::new(read_json(&paths.settings)?);