Add queued and completed timestamps to successful job status

Author: thinkharderdev

ballista/core/proto/ballista.proto

@@ -904,6 +904,8 @@ message GetJobStatusParams {
 
 message SuccessfulJob {
   repeated PartitionLocation partition_location = 1;
+  uint64 queued_at = 2;
+  uint64 completed_at = 3;
 }
 
 message QueuedJob {}

ballista/scheduler/src/scheduler_server/mod.rs

@@ -498,6 +498,7 @@ mod test {
         match status.status {
             Some(job_status::Status::Successful(SuccessfulJob {
                 partition_location,
+                ..
             })) => {
                 assert_eq!(partition_location.len(), 4);
             }

ballista/scheduler/src/state/execution_graph.rs

@@ -1235,6 +1235,8 @@ impl ExecutionGraph {
         self.status = JobStatus {
             status: Some(job_status::Status::Successful(SuccessfulJob {
                 partition_location,
+                queued_at: self.queued_at,
+                completed_at: timestamp_millis(),
             })),
         };
         self.end_time = SystemTime::now()