KR-bluejay commented on code in PR #1314: URL: https://github.com/apache/datafusion-ballista/pull/1314#discussion_r2338070309
########## ballista/executor/src/execution_loop.rs: ########## @@ -88,8 +90,29 @@ pub async fn poll_loop<T: 'static + AsLogicalPlan, U: 'static + AsExecutionPlan> match poll_work_result { Ok(result) => { - let tasks = result.into_inner().tasks; + let PollWorkResult { + tasks, + jobs_to_clean, + } = result.into_inner(); active_job = !tasks.is_empty(); + let work_dir = PathBuf::from(&executor.work_dir); + + // Clean up any state related to the listed jobs Review Comment: Cleanup logic is tied to both scheduler and executor, so I wasn’t sure how to add a focused test here. I did verify that job removal works through `integration-test.sh` (end-to-end). Do you think that is enough for now, or should I also add an additional unit/integration test? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: github-unsubscr...@datafusion.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: github-unsubscr...@datafusion.apache.org For additional commands, e-mail: github-h...@datafusion.apache.org