add context to DataGatherer.Fetch

Signed-off-by: Ashley Davis <ashley.davis@cyberark.com>

Author: SgtCoDFish

pkg/datagatherer/datagatherer.go

@@ -14,7 +14,7 @@ type DataGatherer interface {
 	// Fetch retrieves data.
 	// count is the number of items that were discovered. A negative count means the number
 	// of items was indeterminate.
-	Fetch() (data any, count int, err error)
+	Fetch(ctx context.Context) (data any, count int, err error)
 	// Run starts the data gatherer's informers for resource collection.
 	// Returns error if the data gatherer informer wasn't initialized
 	Run(ctx context.Context) error

pkg/datagatherer/k8sdiscovery/discovery.go

@@ -76,7 +76,7 @@ func (g *DataGathererDiscovery) WaitForCacheSync(ctx context.Context) error {
 }
 
 // Fetch will fetch discovery data from the apiserver, or return an error
-func (g *DataGathererDiscovery) Fetch() (any, int, error) {
+func (g *DataGathererDiscovery) Fetch(ctx context.Context) (any, int, error) {
 	data, err := g.cl.ServerVersion()
 	if err != nil {
 		return nil, -1, fmt.Errorf("failed to get server version: %v", err)

pkg/datagatherer/k8sdynamic/dynamic.go

@@ -385,7 +385,7 @@ func (g *DataGathererDynamic) WaitForCacheSync(ctx context.Context) error {
 
 // Fetch will fetch the requested data from the apiserver, or return an error
 // if fetching the data fails.
-func (g *DataGathererDynamic) Fetch() (any, int, error) {
+func (g *DataGathererDynamic) Fetch(ctx context.Context) (any, int, error) {
 	if g.groupVersionResource.String() == "" {
 		return nil, -1, fmt.Errorf("resource type must be specified")
 	}

pkg/datagatherer/k8sdynamic/dynamic_test.go

@@ -117,6 +117,7 @@ func sortGatheredResources(list []*api.GatheredResource) {
 
 func TestNewDataGathererWithClientAndDynamicInformer(t *testing.T) {
 	ctx := t.Context()
+
 	config := ConfigDynamic{
 		ExcludeNamespaces:    []string{"kube-system"},
 		GroupVersionResource: schema.GroupVersionResource{Group: "foobar", Version: "v1", Resource: "foos"},
@@ -748,7 +749,7 @@ func TestDynamicGatherer_Fetch(t *testing.T) {
 			if waitTimeout(&wg, 30*time.Second) {
 				t.Fatalf("unexpected timeout")
 			}
-			res, expectCount, err := dynamiDg.Fetch()
+			res, expectCount, err := dynamiDg.Fetch(ctx)
 			if err != nil && !tc.err {
 				t.Errorf("expected no error but got: %v", err)
 			}
@@ -1061,7 +1062,7 @@ func TestDynamicGathererNativeResources_Fetch(t *testing.T) {
 			if waitTimeout(&wg, 5*time.Second) {
 				t.Fatalf("unexpected timeout")
 			}
-			rawRes, count, err := dynamiDg.Fetch()
+			rawRes, count, err := dynamiDg.Fetch(ctx)
 			if tc.err {
 				require.Error(t, err)
 			} else {

pkg/datagatherer/local/local.go

@@ -49,7 +49,7 @@ func (g *DataGatherer) WaitForCacheSync(ctx context.Context) error {
 }
 
 // Fetch loads and returns the data from the LocalDatagatherer's dataPath
-func (g *DataGatherer) Fetch() (any, int, error) {
+func (g *DataGatherer) Fetch(ctx context.Context) (any, int, error) {
 	dataBytes, err := os.ReadFile(g.dataPath)
 	if err != nil {
 		return nil, -1, err

pkg/datagatherer/oidc/oidc.go

@@ -66,9 +66,7 @@ func (g *DataGathererOIDC) WaitForCacheSync(ctx context.Context) error {
 }
 
 // Fetch will fetch the OIDC discovery document and JWKS from the cluster API server.
-func (g *DataGathererOIDC) Fetch() (any, int, error) {
-	ctx := context.Background()
-
+func (g *DataGathererOIDC) Fetch(ctx context.Context) (any, int, error) {
 	oidcResponse, oidcErr := g.fetchOIDCConfig(ctx)
 	jwksResponse, jwksErr := g.fetchJWKS(ctx)
 

pkg/datagatherer/oidc/oidc_test.go

@@ -51,7 +51,7 @@ func TestFetch_Success(t *testing.T) {
 	rc := makeRESTClient(t, ts)
 	g := &DataGathererOIDC{cl: rc}
 
-	anyRes, count, err := g.Fetch()
+	anyRes, count, err := g.Fetch(t.Context())
 	require.NoError(t, err)
 	require.Equal(t, 1, count)
 
@@ -197,7 +197,7 @@ func TestFetch_Errors(t *testing.T) {
 			rc := makeRESTClient(t, ts)
 			g := &DataGathererOIDC{cl: rc}
 
-			anyRes, count, err := g.Fetch()
+			anyRes, count, err := g.Fetch(t.Context())
 			require.NoError(t, err)
 			require.Equal(t, 1, count)