feat: add exec_duration

codebench-dev · Jun 19, 2021 · 2c2e403 · 2c2e403
1 parent 00b043e
commit 2c2e403
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 10 deletions.
diff --git a/job_queue_rabbitmq.go b/job_queue_rabbitmq.go
@@ -16,10 +16,11 @@ type jobQueue struct {
 }
 
 type jobStatus struct {
-	ID     string `json:"id"`
-	Status string `json:"status"`
-	StdErr string `json:"stderr"`
-	StdOut string `json:"stdout"`
+	ID           string `json:"id"`
+	Status       string `json:"status"`
+	StdErr       string `json:"stderr"`
+	StdOut       string `json:"stdout"`
+	ExecDuration int    `json:"exec_duration"`
 }
 
 func newJobQueue(endpoint string) jobQueue {
@@ -137,10 +138,11 @@ func (q jobQueue) setjobFailed(ctx context.Context, job benchJob) error {
 }
 func (q jobQueue) setjobResult(ctx context.Context, job benchJob, res agentExecRes) error {
 	jobStatus := &jobStatus{
-		ID:     job.ID,
-		Status: "done",
-		StdErr: res.StdErr,
-		StdOut: res.StdOut,
+		ID:           job.ID,
+		Status:       "done",
+		StdErr:       res.StdErr,
+		StdOut:       res.StdOut,
+		ExecDuration: res.ExecDuration,
 	}
 	log.WithField("jobStatus", jobStatus).Info("Set job result")
 

diff --git a/main.go b/main.go
@@ -35,8 +35,9 @@ type agentRunReq struct {
 }
 
 type agentExecRes struct {
-	StdErr string `json:"stderr"`
-	StdOut string `json:"stdout"`
+	StdErr       string `json:"stderr"`
+	StdOut       string `json:"stdout"`
+	ExecDuration int    `json:"exec_duration"`
 }
 
 type runningFirecracker struct {