feat: add MemUsage support

codebench-dev · Jul 21, 2021 · fd7afd4 · fd7afd4
1 parent 4dcbb44
commit fd7afd4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/job_queue_rabbitmq.go b/job_queue_rabbitmq.go
@@ -23,6 +23,7 @@ type jobStatus struct {
 	StdErr       string `json:"stderr"`
 	StdOut       string `json:"stdout"`
 	ExecDuration int    `json:"exec_duration"`
+	MemUsage     int64  `json:"mem_usage"`
 }
 
 func newJobQueue(endpoint string) jobQueue {
@@ -149,6 +150,7 @@ func (q jobQueue) setjobResult(ctx context.Context, job benchJob, res agentExecR
 		StdErr:       res.StdErr,
 		StdOut:       res.StdOut,
 		ExecDuration: res.ExecDuration,
+		MemUsage:     res.MemUsage,
 	}
 	log.WithField("jobStatus", jobStatus).Info("Set job result")
 

diff --git a/main.go b/main.go
@@ -41,6 +41,7 @@ type agentExecRes struct {
 	StdErr       string `json:"stderr"`
 	StdOut       string `json:"stdout"`
 	ExecDuration int    `json:"exec_duration"`
+	MemUsage     int64  `json:"mem_usage"`
 }
 
 type runningFirecracker struct {