fix(scheduler): when index is recreated, the cache should be cleared (#…

…2241)
TabbyML · May 26, 2024 · f9c4240 · f9c4240
1 parent 1d1edfe
commit f9c4240
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/crates/tabby-scheduler/src/code/index.rs b/crates/tabby-scheduler/src/code/index.rs
@@ -12,6 +12,9 @@ static AVG_LINE_LENGTH_THRESHOLD: f32 = 150f32;
 
 pub async fn index_repository(cache: &mut CacheStore, repository: &RepositoryConfig) {
     let index = create_code_index();
+    if index.recreated {
+        cache.clear_indexed()
+    }
     add_changed_documents(cache, repository, &index).await;
     index.commit();
 }

diff --git a/crates/tabby-scheduler/src/indexer.rs b/crates/tabby-scheduler/src/indexer.rs
@@ -18,19 +18,21 @@ pub trait IndexAttributeBuilder<T>: Send + Sync {
 pub struct Indexer<T> {
     builder: Box<dyn IndexAttributeBuilder<T>>,
     writer: IndexWriter,
+    pub recreated: bool,
 }
 
 impl<T> Indexer<T> {
     pub fn new(builder: impl IndexAttributeBuilder<T> + 'static) -> Self {
         let doc = IndexSchema::instance();
-        let (_, index) = open_or_create_index(&doc.schema, &path::index_dir());
+        let (recreated, index) = open_or_create_index(&doc.schema, &path::index_dir());
         let writer = index
             .writer(150_000_000)
             .expect("Failed to create index writer");
 
         Self {
             builder: Box::new(builder),
             writer,
+            recreated,
         }
     }