chore: clippy

AppFlowy-IO · Dec 24, 2024 · 2677aa0 · 2677aa0
1 parent 452a6ab
commit 2677aa0
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 0 deletions.
diff --git a/.sqlx/query-d0e5f5097b35a15f19e9e7faf2c62336d5f130e939331e84c7d834f6028ea673.json b/.sqlx/query-d0e5f5097b35a15f19e9e7faf2c62336d5f130e939331e84c7d834f6028ea673.json
diff --git a/.sqlx/query-f8c909517885cb30e3f7d573edf47138f90ea9c5fa73eb927cc5487c3d9ad0be.json b/.sqlx/query-f8c909517885cb30e3f7d573edf47138f90ea9c5fa73eb927cc5487c3d9ad0be.json
diff --git a/services/appflowy-worker/src/application.rs b/services/appflowy-worker/src/application.rs
@@ -141,6 +141,9 @@ pub async fn create_app(listener: TcpListener, config: Config) -> Result<(), Err
     state.metrics.embedder_metrics.clone(),
     threads.clone(),
     BackgroundIndexerConfig {
+      enable: appflowy_collaborate::config::get_env_var("APPFLOWY_INDEXER_ENABLED", "true")
+        .parse::<bool>()
+        .unwrap_or(true),
       open_api_key: appflowy_collaborate::config::get_env_var("APPFLOWY_AI_OPENAI_API_KEY", ""),
       tick_interval_secs: 10,
     },

diff --git a/services/appflowy-worker/src/indexer_worker/worker.rs b/services/appflowy-worker/src/indexer_worker/worker.rs
@@ -1,4 +1,5 @@
 use app_error::AppError;
+use appflowy_collaborate::config::get_env_var;
 use collab_entity::CollabType;
 use database::index::get_collabs_indexed_at;
 use indexer::collab_indexer::{Indexer, IndexerProvider};
@@ -25,6 +26,7 @@ use tracing::{error, info, trace};
 
 #[derive(Debug)]
 pub struct BackgroundIndexerConfig {
+  pub enable: bool,
   pub open_api_key: String,
   pub tick_interval_secs: u64,
 }
@@ -36,6 +38,11 @@ pub async fn run_background_indexer(
   threads: Arc<ThreadPoolNoAbort>,
   config: BackgroundIndexerConfig,
 ) {
+  if !config.enable {
+    info!("Background indexer is disabled. Stop background indexer");
+    return;
+  }
+
   if config.open_api_key.is_empty() {
     error!("OpenAI API key is not set. Stop background indexer");
     return;