Remove typecast for ROOT_TASK_* constants

risingwavelabs · May 17, 2024 · f54bb6c · f54bb6c
1 parent 5a0c927
commit f54bb6c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/frontend/src/scheduler/distributed/query.rs b/src/frontend/src/scheduler/distributed/query.rs
@@ -392,12 +392,12 @@ impl QueryRunner {
             let root_task_id_prost = TaskIdPb {
                 query_id: self.query.query_id.clone().id,
                 stage_id: self.query.root_stage_id(),
-                task_id: ROOT_TASK_ID as u64,
+                task_id: ROOT_TASK_ID,
             };
 
             TaskOutputIdPb {
                 task_id: Some(root_task_id_prost),
-                output_id: ROOT_TASK_OUTPUT_ID as u64,
+                output_id: ROOT_TASK_OUTPUT_ID,
             }
         };
 

diff --git a/src/frontend/src/scheduler/distributed/stage.rs b/src/frontend/src/scheduler/distributed/stage.rs
@@ -584,7 +584,7 @@ impl StageRunner {
         let root_stage_id = self.stage.id;
         // Currently, the dml or table scan should never be root fragment, so the partition is None.
         // And root fragment only contain one task.
-        let plan_fragment = self.create_plan_fragment(ROOT_TASK_ID as u64, None);
+        let plan_fragment = self.create_plan_fragment(ROOT_TASK_ID, None);
         let plan_node = plan_fragment.root.unwrap();
         let task_id = TaskIdBatch {
             query_id: self.stage.query_id.id.clone(),