chore: disable batch index (#1098)
This commit is contained in:
parent
5ea714b004
commit
381b02a4d0
|
|
@ -33,6 +33,7 @@ use uuid::Uuid;
|
||||||
pub struct IndexerScheduler {
|
pub struct IndexerScheduler {
|
||||||
indexer_provider: Arc<IndexerProvider>,
|
indexer_provider: Arc<IndexerProvider>,
|
||||||
pg_pool: PgPool,
|
pg_pool: PgPool,
|
||||||
|
#[allow(dead_code)]
|
||||||
storage: Arc<dyn CollabStorage>,
|
storage: Arc<dyn CollabStorage>,
|
||||||
threads: Arc<ThreadPoolNoAbort>,
|
threads: Arc<ThreadPoolNoAbort>,
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
|
|
@ -107,7 +108,7 @@ impl IndexerScheduler {
|
||||||
this.pg_pool.clone(),
|
this.pg_pool.clone(),
|
||||||
this.metrics.clone(),
|
this.metrics.clone(),
|
||||||
));
|
));
|
||||||
tokio::spawn(handle_unindexed_collabs(this.clone()));
|
// tokio::spawn(handle_unindexed_collabs(this.clone()));
|
||||||
}
|
}
|
||||||
|
|
||||||
this
|
this
|
||||||
|
|
@ -396,6 +397,7 @@ fn should_embed(
|
||||||
should_embed
|
should_embed
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
async fn handle_unindexed_collabs(scheduler: Arc<IndexerScheduler>) {
|
async fn handle_unindexed_collabs(scheduler: Arc<IndexerScheduler>) {
|
||||||
// wait for 30 seconds before starting indexing
|
// wait for 30 seconds before starting indexing
|
||||||
tokio::time::sleep(tokio::time::Duration::from_secs(30)).await;
|
tokio::time::sleep(tokio::time::Duration::from_secs(30)).await;
|
||||||
|
|
@ -479,6 +481,7 @@ pub async fn get_unindexed_collabs(
|
||||||
.boxed()
|
.boxed()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
async fn index_unindexd_collab(
|
async fn index_unindexd_collab(
|
||||||
embedder: Embedder,
|
embedder: Embedder,
|
||||||
indexer_provider: &Arc<IndexerProvider>,
|
indexer_provider: &Arc<IndexerProvider>,
|
||||||
|
|
|
||||||
|
|
@ -837,23 +837,23 @@ async fn batch_create_collab_handler(
|
||||||
start.elapsed()
|
start.elapsed()
|
||||||
);
|
);
|
||||||
|
|
||||||
if state
|
// if state
|
||||||
.indexer_scheduler
|
// .indexer_scheduler
|
||||||
.can_index_workspace(&workspace_id)
|
// .can_index_workspace(&workspace_id)
|
||||||
.await?
|
// .await?
|
||||||
{
|
// {
|
||||||
let indexed_collabs: Vec<_> = collab_params_list
|
// let indexed_collabs: Vec<_> = collab_params_list
|
||||||
.iter()
|
// .iter()
|
||||||
.filter(|p| state.indexer_scheduler.is_indexing_enabled(&p.collab_type))
|
// .filter(|p| state.indexer_scheduler.is_indexing_enabled(&p.collab_type))
|
||||||
.map(IndexedCollab::from)
|
// .map(IndexedCollab::from)
|
||||||
.collect();
|
// .collect();
|
||||||
|
//
|
||||||
if !indexed_collabs.is_empty() {
|
// if !indexed_collabs.is_empty() {
|
||||||
state
|
// state
|
||||||
.indexer_scheduler
|
// .indexer_scheduler
|
||||||
.index_encoded_collabs(&workspace_id, indexed_collabs)?;
|
// .index_encoded_collabs(&workspace_id, indexed_collabs)?;
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
|
|
||||||
let start = Instant::now();
|
let start = Instant::now();
|
||||||
state
|
state
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue