Skip to content

Commit 19fa57b

Browse files
committed
remove dependency to pkg/features on service controller package
Signed-off-by: Yassine TIJANI <[email protected]>
1 parent 18d7f88 commit 19fa57b

File tree

3 files changed

+15
-9
lines changed

3 files changed

+15
-9
lines changed

pkg/controller/service/BUILD

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ go_library(
1717
deps = [
1818
"//pkg/apis/core/v1/helper:go_default_library",
1919
"//pkg/controller:go_default_library",
20-
"//pkg/features:go_default_library",
2120
"//pkg/util/metrics:go_default_library",
2221
"//pkg/util/slice:go_default_library",
2322
"//staging/src/k8s.io/api/core/v1:go_default_library",
@@ -52,7 +51,6 @@ go_test(
5251
deps = [
5352
"//pkg/api/testapi:go_default_library",
5453
"//pkg/controller:go_default_library",
55-
"//pkg/features:go_default_library",
5654
"//staging/src/k8s.io/api/core/v1:go_default_library",
5755
"//staging/src/k8s.io/apimachinery/pkg/api/errors:go_default_library",
5856
"//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",

pkg/controller/service/service_controller.go

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@ import (
4343
"k8s.io/klog"
4444
v1helper "k8s.io/kubernetes/pkg/apis/core/v1/helper"
4545
"k8s.io/kubernetes/pkg/controller"
46-
kubefeatures "k8s.io/kubernetes/pkg/features"
4746
"k8s.io/kubernetes/pkg/util/metrics"
4847
"k8s.io/kubernetes/pkg/util/slice"
4948
)
@@ -67,6 +66,16 @@ const (
6766
// LabelNodeRoleExcludeBalancer specifies that the node should be
6867
// exclude from load balancers created by a cloud provider.
6968
LabelNodeRoleExcludeBalancer = "alpha.service-controller.kubernetes.io/exclude-balancer"
69+
70+
// serviceNodeExclusionFeature is the feature gate name that
71+
// enables nodes to exclude themselves from service load balancers
72+
// originated from: https://github.com/kubernetes/kubernetes/blob/28e800245e/pkg/features/kube_features.go#L178
73+
serviceNodeExclusionFeature = "ServiceNodeExclusion"
74+
75+
// ServiceLoadBalancerFinalizerFeature is the feature gate name that
76+
// enables Finalizer Protection for Service LoadBalancers.
77+
// orginated from: https://github.com/kubernetes/kubernetes/blob/28e800245e/pkg/features/kube_features.go#L433
78+
serviceLoadBalancerFinalizerFeature = "ServiceLoadBalancerFinalizer"
7079
)
7180

7281
type cachedService struct {
@@ -150,7 +159,7 @@ func New(
150159
}
151160
},
152161
DeleteFunc: func(old interface{}) {
153-
if utilfeature.DefaultFeatureGate.Enabled(kubefeatures.ServiceLoadBalancerFinalizer) {
162+
if utilfeature.DefaultFeatureGate.Enabled(serviceLoadBalancerFinalizerFeature) {
154163
// No need to handle deletion event if finalizer feature gate is
155164
// enabled. Because the deletion would be handled by the update
156165
// path when the deletion timestamp is added.
@@ -327,7 +336,7 @@ func (s *ServiceController) syncLoadBalancerIfNeeded(service *v1.Service, key st
327336
op = ensureLoadBalancer
328337
klog.V(2).Infof("Ensuring load balancer for service %s", key)
329338
s.eventRecorder.Event(service, v1.EventTypeNormal, "EnsuringLoadBalancer", "Ensuring load balancer")
330-
if utilfeature.DefaultFeatureGate.Enabled(kubefeatures.ServiceLoadBalancerFinalizer) {
339+
if utilfeature.DefaultFeatureGate.Enabled(serviceLoadBalancerFinalizerFeature) {
331340
// Always try to add finalizer prior to load balancer creation.
332341
// It will be a no-op if finalizer already exists.
333342
// Note this also retrospectively puts on finalizer if the cluster
@@ -623,7 +632,7 @@ func getNodeConditionPredicate() corelisters.NodeConditionPredicate {
623632
return false
624633
}
625634

626-
if utilfeature.DefaultFeatureGate.Enabled(kubefeatures.ServiceNodeExclusion) {
635+
if utilfeature.DefaultFeatureGate.Enabled(serviceNodeExclusionFeature) {
627636
if _, hasExcludeBalancerLabel := node.Labels[LabelNodeRoleExcludeBalancer]; hasExcludeBalancerLabel {
628637
return false
629638
}

pkg/controller/service/service_controller_test.go

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ import (
2525
"testing"
2626
"time"
2727

28-
"k8s.io/api/core/v1"
28+
v1 "k8s.io/api/core/v1"
2929
apierrors "k8s.io/apimachinery/pkg/api/errors"
3030
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3131
"k8s.io/apimachinery/pkg/runtime"
@@ -42,7 +42,6 @@ import (
4242
featuregatetesting "k8s.io/component-base/featuregate/testing"
4343
"k8s.io/kubernetes/pkg/api/testapi"
4444
"k8s.io/kubernetes/pkg/controller"
45-
"k8s.io/kubernetes/pkg/features"
4645
)
4746

4847
const region = "us-central"
@@ -308,7 +307,7 @@ func TestSyncLoadBalancerIfNeeded(t *testing.T) {
308307

309308
for _, tc := range testCases {
310309
t.Run(tc.desc, func(t *testing.T) {
311-
defer featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.ServiceLoadBalancerFinalizer, tc.enableFeatureGate)()
310+
defer featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, serviceLoadBalancerFinalizerFeature, tc.enableFeatureGate)()
312311

313312
controller, cloud, client := newController()
314313
cloud.Exists = tc.lbExists

0 commit comments

Comments
 (0)