adding context

Author: iamayushm

common-lib/utils/k8s/K8sUtil.go

@@ -148,9 +148,9 @@ type K8sService interface {
 	CreateNsWithLabels(namespace string, labels map[string]string, client *v12.CoreV1Client) (ns *v1.Namespace, err error)
 	CreateNs(namespace string, client *v12.CoreV1Client) (ns *v1.Namespace, err error)
 	GetGVRForCRD(config *rest.Config, CRDName string) (schema.GroupVersionResource, error)
-	GetResourceByGVR(config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string) (*unstructured.Unstructured, error)
-	PatchResourceByGVR(config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, patchType types.PatchType, patchData []byte) (*unstructured.Unstructured, error)
-	DeleteResourceByGVR(config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, forceDelete bool) error
+	GetResourceByGVR(ctx context.Context, config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string) (*unstructured.Unstructured, error)
+	PatchResourceByGVR(ctx context.Context, config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, patchType types.PatchType, patchData []byte) (*unstructured.Unstructured, error)
+	DeleteResourceByGVR(ctx context.Context, config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, forceDelete bool) error
 }
 
 func NewK8sUtil(logger *zap.SugaredLogger, runTimeConfig *RuntimeConfig) (*K8sServiceImpl, error) {
@@ -1936,35 +1936,35 @@ func (impl *K8sServiceImpl) GetGVRForCRD(config *rest.Config, CRDName string) (s
 	}, nil
 }
 
-func (impl *K8sServiceImpl) GetResourceByGVR(config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string) (*unstructured.Unstructured, error) {
+func (impl *K8sServiceImpl) GetResourceByGVR(ctx context.Context, config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string) (*unstructured.Unstructured, error) {
 	dynClient, err := dynamic.NewForConfig(config)
 	if err != nil {
 		impl.logger.Errorw("failed to create dynamic client", "err", err)
 		return nil, err
 	}
-	resource, err := dynClient.Resource(GVR).Namespace(namespace).Get(context.TODO(), resourceName, metav1.GetOptions{})
+	resource, err := dynClient.Resource(GVR).Namespace(namespace).Get(ctx, resourceName, metav1.GetOptions{})
 	if err != nil {
 		impl.logger.Errorw("failed to get resource", "resourceName", resourceName, "namespace", namespace, "err", err)
 		return nil, err
 	}
 	return resource, nil
 }
 
-func (impl *K8sServiceImpl) PatchResourceByGVR(config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, patchType types.PatchType, patchData []byte) (*unstructured.Unstructured, error) {
+func (impl *K8sServiceImpl) PatchResourceByGVR(ctx context.Context, config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, patchType types.PatchType, patchData []byte) (*unstructured.Unstructured, error) {
 	dynClient, err := dynamic.NewForConfig(config)
 	if err != nil {
 		impl.logger.Errorw("failed to create dynamic client", "err", err)
 		return nil, err
 	}
-	resource, err := dynClient.Resource(GVR).Namespace(namespace).Patch(context.TODO(), resourceName, patchType, patchData, metav1.PatchOptions{FieldManager: "patch"})
+	resource, err := dynClient.Resource(GVR).Namespace(namespace).Patch(ctx, resourceName, patchType, patchData, metav1.PatchOptions{FieldManager: "patch"})
 	if err != nil {
 		impl.logger.Errorw("failed to get resource", "resourceName", resourceName, "namespace", namespace, "err", err)
 		return nil, err
 	}
 	return resource, nil
 }
 
-func (impl *K8sServiceImpl) DeleteResourceByGVR(config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, forceDelete bool) error {
+func (impl *K8sServiceImpl) DeleteResourceByGVR(ctx context.Context, config *rest.Config, GVR schema.GroupVersionResource, resourceName, namespace string, forceDelete bool) error {
 	dynClient, err := dynamic.NewForConfig(config)
 	if err != nil {
 		impl.logger.Errorw("failed to create dynamic client", "err", err)
@@ -1974,7 +1974,7 @@ func (impl *K8sServiceImpl) DeleteResourceByGVR(config *rest.Config, GVR schema.
 	if forceDelete {
 		deleteOptions.GracePeriodSeconds = pointer.Int64Ptr(0)
 	}
-	err = dynClient.Resource(GVR).Namespace(namespace).Delete(context.TODO(), resourceName, deleteOptions)
+	err = dynClient.Resource(GVR).Namespace(namespace).Delete(ctx, resourceName, deleteOptions)
 	if err != nil {
 		impl.logger.Errorw("failed to get resource", "resourceName", resourceName, "namespace", namespace, "err", err)
 		return err