chore: fix duplicated add

TabbyML · Nov 25, 2024 · 835b5de · 835b5de
1 parent b447bf0
commit 835b5de
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/ee/tabby-webserver/src/service/background_job/third_party_integration.rs b/ee/tabby-webserver/src/service/background_job/third_party_integration.rs
@@ -143,13 +143,14 @@ impl SchedulerGithubGitlabJob {
             let mut num_deleted = 0;
             for await (state, doc) in issue_stream.chain(pull_stream) {
                 let id = &doc.id().to_owned();
-                if index.add(state.updated_at, doc).await {
+                if !state.should_clean && index.add(state.updated_at, doc).await {
                     num_updated += 1
                 }
                 if state.should_clean && index.delete(id).await {
                     num_deleted += 1;
                 }
 
+println!("doc {}", id);
                 count += 1;
                 if count % 100 == 0 {
                     logkit::info!("{} docs seen, {} docs updated, {} docs deleted", count, num_updated, num_deleted);

diff --git a/ee/tabby-webserver/src/service/background_job/third_party_integration/issues.rs b/ee/tabby-webserver/src/service/background_job/third_party_integration/issues.rs
@@ -65,7 +65,7 @@ pub async fn list_github_issues(
                 };
                 yield (FetchState {
                     updated_at: issue.updated_at,
-                     should_clean: false,
+                    should_clean: false,
                 }, doc);
             }
 

diff --git a/ee/tabby-webserver/src/service/background_job/third_party_integration/pulls.rs b/ee/tabby-webserver/src/service/background_job/third_party_integration/pulls.rs
@@ -65,6 +65,7 @@ pub async fn list_github_pulls(
                             updated_at: pull.updated_at.unwrap(),
                             should_clean: true,
                         }, doc);
+                        continue;
                     }
                 }