feat: add atomic operation for setting resource aggregations (#6301)

Author: rangoo94

pkg/cloud/data/testworkflow/execution.go

@@ -206,3 +206,7 @@ func (r *CloudRepository) GetUnassigned(ctx context.Context) (result []testkube.
 func (r *CloudRepository) AbortIfQueued(ctx context.Context, id string) (bool, error) {
 	return false, errors.New("not supported")
 }
+
+func (r *CloudRepository) UpdateResourceAggregations(ctx context.Context, id string, resourceAggregations *testkube.TestWorkflowExecutionResourceAggregationsReport) error {
+	return errors.New("not supported")
+}

pkg/repository/testworkflow/interface.go

@@ -95,6 +95,8 @@ type Repository interface {
 	UpdateReport(ctx context.Context, id string, report *testkube.TestWorkflowReport) (err error)
 	// UpdateOutput updates list of output references in the execution result
 	UpdateOutput(ctx context.Context, id string, output []testkube.TestWorkflowOutput) (err error)
+	// UpdateResourceAggregations updates the summary of resource metrics
+	UpdateResourceAggregations(ctx context.Context, id string, resourceAggregations *testkube.TestWorkflowExecutionResourceAggregationsReport) (err error)
 	// DeleteByTestWorkflow deletes execution results by workflow
 	DeleteByTestWorkflow(ctx context.Context, workflowName string) error
 	// DeleteAll deletes all execution results

pkg/repository/testworkflow/mock_repository.go

@@ -392,6 +392,11 @@ func (r *MongoRepository) UpdateOutput(ctx context.Context, id string, refs []te
 	return
 }
 
+func (r *MongoRepository) UpdateResourceAggregations(ctx context.Context, id string, resourceAggregations *testkube.TestWorkflowExecutionResourceAggregationsReport) (err error) {
+	_, err = r.Coll.UpdateOne(ctx, bson.M{"id": id}, bson.M{"$set": bson.M{"resourceaggregations": resourceAggregations}})
+	return
+}
+
 func composeQueryAndOpts(filter Filter) (bson.M, *options.FindOptions) {
 	query := bson.M{}
 	opts := options.Find()