andygrove commented on code in PR #59:
URL: https://github.com/apache/arrow-ballista/pull/59#discussion_r889731148
##########
ballista/rust/core/proto/ballista.proto:
##########
@@ -622,6 +622,37 @@ enum JoinSide{
///////////////////////////////////////////////////////////////////////////////////////////////////
// Ballista Scheduling
///////////////////////////////////////////////////////////////////////////////////////////////////
+message TaskInputPartitions {
+ uint32 partition = 1;
+ repeated PartitionLocation partition_location = 2;
+}
+
+message GraphStageInput {
+ uint32 stage_id = 1;
+ repeated TaskInputPartitions partition_locations = 2;
+ bool complete = 3;
+}
+
+
+message ExecutionGraphStage {
+ uint64 stage_id = 1;
+ uint32 partitions = 2;
+ PhysicalHashRepartition output_partitioning = 3;
+ repeated GraphStageInput inputs = 4;
+ bytes plan = 5;
+ repeated TaskStatus task_statuses = 6;
+ uint32 output_link = 7;
+ bool resolved = 8;
+}
+
+message ExecutionGraph {
+ string job_id = 1;
+ string session_id = 2;
+ JobStatus status = 3;
+ repeated ExecutionGraphStage stages = 4;
Review Comment:
Got it. Yes, some comments in the structs here would be great,
--
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...@arrow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
