renaming transformed logic for pod informer

kubewharf · Mar 19, 2024 · 91a4393 · 91a4393
1 parent 3b8b019
commit 91a4393
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/pkg/controller/lifecycle/agent-healthz/healthz_controller.go b/pkg/controller/lifecycle/agent-healthz/healthz_controller.go
@@ -162,7 +162,7 @@ func NewHealthzController(ctx context.Context,
 			ec.nodeSelector, podIndexer, ec.nodeLister, ec.cnrLister, ec.healthzHelper)
 	}
 
-	native.WithPodTransformer(podTransformerFunc)
+	native.SetPodTransformer(podTransformerFunc)
 	return ec, nil
 }
 

diff --git a/pkg/controller/spd/spd.go b/pkg/controller/spd/spd.go
@@ -199,7 +199,7 @@ func NewSPDController(ctx context.Context, controlCtx *katalystbase.GenericConte
 		return nil, err
 	}
 
-	native.WithPodTransformer(podTransformerFunc)
+	native.SetPodTransformer(podTransformerFunc)
 	return spdController, nil
 }
 

diff --git a/pkg/util/native/informer_test.go b/pkg/util/native/informer_test.go
@@ -29,7 +29,7 @@ func TestPodTransformer(t *testing.T) {
 
 	as := require.New(t)
 
-	WithPodTransformer(func(src, dest *core.Pod) {
+	SetPodTransformer(func(src, dest *core.Pod) {
 		dest.Spec.NodeName = src.Spec.NodeName
 	})
 

diff --git a/pkg/util/native/informer_transformed.go → pkg/util/native/informer_transformed_pod.go b/pkg/util/native/informer_transformed.go → pkg/util/native/informer_transformed_pod.go
@@ -29,12 +29,11 @@ type PodTransformerFunc func(src, dest *corev1.Pod)
 var podTransformers []PodTransformerFunc
 var podTransformerMtx sync.RWMutex
 
-func WithPodTransformer(f PodTransformerFunc) {
+func SetPodTransformer(f PodTransformerFunc) {
 	podTransformerMtx.Lock()
 	defer podTransformerMtx.Unlock()
 	podTransformers = append(podTransformers, f)
 }
-
 func GetPodTransformer() (cache.TransformFunc, bool) {
 	podTransformerMtx.RLock()
 	defer podTransformerMtx.RUnlock()