chore: use context in more workflows

Author: PeterSchafer

pkg/local_workflows/ignore_workflow/ignore_workflow.go

@@ -348,7 +348,7 @@ func sendCreateIgnore(invocationCtx workflow.InvocationContext, input policyApi.
 		return nil, err
 	}
 
-	ctx, cancelFunc := context.WithTimeout(context.Background(), policyApiTimeout)
+	ctx, cancelFunc := context.WithTimeout(invocationCtx.Context(), policyApiTimeout)
 	defer cancelFunc()
 
 	resp, err := client.CreateOrgPolicyWithApplicationVndAPIPlusJSONBody(ctx, orgUuid, &params, input)

pkg/local_workflows/output_workflow/findings_model_tools.go

@@ -1,7 +1,6 @@
 package output_workflow
 
 import (
-	"context"
 	"encoding/json"
 	"fmt"
 	"strings"
@@ -117,7 +116,7 @@ func HandleContentTypeFindingsModel(input []workflow.Data, invocation workflow.I
 	writerMap := applyTemplatesToWriters(supportedMimeTypes, writers)
 
 	// iterate over all writers and render for each of them
-	ctx := context.Background()
+	ctx := invocation.Context()
 	availableThreads := semaphore.NewWeighted(threadCount)
 	for k, v := range writerMap {
 		err = availableThreads.Acquire(ctx, 1)

pkg/local_workflows/output_workflow/ufm_tools.go

@@ -1,7 +1,6 @@
 package output_workflow
 
 import (
-	"context"
 	"errors"
 	"sync"
 
@@ -111,7 +110,7 @@ func HandleContentTypeUnifiedModel(input []workflow.Data, invocation workflow.In
 	writerMap := applyTemplatesToWriters(supportedMimeTypes, writers)
 
 	// iterate over all writers and render for each of them
-	ctx := context.Background()
+	ctx := invocation.Context()
 	availableThreads := semaphore.NewWeighted(threadCount)
 	var errMu sync.Mutex
 	var errs []error