Updated job_ids to ObjectId; refactored loop and get_fragment_mappings.

risingwavelabs · Jul 19, 2024 · 1f23de9 · 1f23de9
1 parent 48111f5
commit 1f23de9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/meta/src/controller/fragment.rs b/src/meta/src/controller/fragment.rs
@@ -68,7 +68,7 @@ impl CatalogControllerInner {
     ) -> MetaResult<impl Iterator<Item = FragmentWorkerSlotMapping> + '_> {
         let txn = self.db.begin().await?;
 
-        let job_ids: Vec<i32> = StreamingJob::find()
+        let job_ids: Vec<ObjectId> = StreamingJob::find()
             .select_only()
             .column(streaming_job::Column::JobId)
             .filter(streaming_job::Column::JobStatus.eq(JobStatus::Created))
@@ -78,7 +78,7 @@ impl CatalogControllerInner {
 
         let mut result = vec![];
         for job_id in job_ids {
-            let mappings = get_fragment_mappings(&txn, job_id as ObjectId).await?;
+            let mappings = get_fragment_mappings(&txn, job_id).await?;
 
             result.extend(mappings.into_iter());
         }