Verified Commit fdf4c78c authored by Jean-Gabriel Doyon's avatar Jean-Gabriel Doyon Committed by GitLab
Browse files

feat(indexer): add NATS progress pings to code indexing pipeline

parent 769b22b8

crates/indexer/src/modules/code/indexing_pipeline.rs

+6 −0
Original line number Diff line number Diff line
@@ -71,6 +71,8 @@ impl CodeIndexingPipeline { 
            .repository_fetch_duration

            .record(fetch_start.elapsed().as_secs_f64(), &[]);



        context.progress.notify_in_progress().await;



        let extract_start = Instant::now();

        archive::extract_tar_gz(&archive_bytes, temp_dir.path())

            .map_err(|e| HandlerError::Processing(format!("failed to extract archive: {e}")))
@@ -79,6 +81,8 @@ impl CodeIndexingPipeline { 
            .repository_extract_duration

            .record(extract_start.elapsed().as_secs_f64(), &[]);



        context.progress.notify_in_progress().await;



        let indexed_at = Utc::now();

        self.run_indexing(

            context,
@@ -178,6 +182,8 @@ impl CodeIndexingPipeline { 
            );

        }



        context.progress.notify_in_progress().await;



        let Some(graph_data) = result.graph_data else {

            debug!(project_id, branch = %branch, "indexing produced no graph data, skipping write");

            return Ok(());