diff --git a/compactor/src/components/commit.rs b/compactor/src/components/commit.rs index 6460306264..fbbbd20a24 100644 --- a/compactor/src/components/commit.rs +++ b/compactor/src/components/commit.rs @@ -4,7 +4,7 @@ use compactor_scheduler::{ CommitUpdate, CompactionJob, CompactionJobStatus, CompactionJobStatusResponse, CompactionJobStatusVariant, Scheduler, }; -use data_types::{CompactionLevel, ParquetFile, ParquetFileId, ParquetFileParams, PartitionId}; +use data_types::{CompactionLevel, ParquetFile, ParquetFileId, ParquetFileParams}; #[derive(Debug)] pub struct CommitToScheduler { @@ -18,7 +18,7 @@ impl CommitToScheduler { pub async fn commit( &self, - partition_id: PartitionId, + job: CompactionJob, delete: &[ParquetFile], upgrade: &[ParquetFile], create: &[ParquetFileParams], @@ -27,9 +27,9 @@ impl CommitToScheduler { match self .scheduler .update_job_status(CompactionJobStatus { - job: CompactionJob::new(partition_id), + job: job.clone(), status: CompactionJobStatusVariant::Update(CommitUpdate::new( - partition_id, + job.partition_id, delete.into(), upgrade.into(), create.into(), diff --git a/compactor/src/driver.rs b/compactor/src/driver.rs index 5c73969226..27e1f7a7c7 100644 --- a/compactor/src/driver.rs +++ b/compactor/src/driver.rs @@ -651,7 +651,7 @@ async fn update_catalog( let created_ids = components .commit .commit( - partition_id, + job, &files_to_delete, &files_to_upgrade, &file_params_to_create,