From 84946970274408ed87035eb309c0cd29fc1bd90e Mon Sep 17 00:00:00 2001 From: Jordan Liggitt Date: Sun, 27 Aug 2017 22:19:28 -0400 Subject: [PATCH] Feature gate initializers field Kubernetes-commit: 658956f0631c6bd3ecf9f756db4a0710efea98ea --- pkg/features/kube_features.go | 8 ++++ pkg/registry/generic/registry/BUILD | 3 ++ pkg/registry/generic/registry/store_test.go | 9 +++++ pkg/registry/rest/BUILD | 2 + pkg/registry/rest/create.go | 7 ++++ pkg/registry/rest/update.go | 8 ++++ pkg/util/feature/BUILD | 5 ++- pkg/util/feature/testing/BUILD | 22 ++++++++++ .../feature/testing/feature_gate_testing.go | 40 +++++++++++++++++++ 9 files changed, 103 insertions(+), 1 deletion(-) create mode 100644 pkg/util/feature/testing/BUILD create mode 100644 pkg/util/feature/testing/feature_gate_testing.go diff --git a/pkg/features/kube_features.go b/pkg/features/kube_features.go index eab9a28d8..8e2478aab 100644 --- a/pkg/features/kube_features.go +++ b/pkg/features/kube_features.go @@ -47,6 +47,13 @@ const ( // // Enables compression of REST responses (GET and LIST only) APIResponseCompression utilfeature.Feature = "APIResponseCompression" + + // owner: @smarterclayton + // alpha: v1.7 + // + // Allow asynchronous coordination of object creation. + // Auto-enabled by the Initializers admission plugin. + Initializers utilfeature.Feature = "Initializers" ) func init() { @@ -60,4 +67,5 @@ var defaultKubernetesFeatureGates = map[utilfeature.Feature]utilfeature.FeatureS StreamingProxyRedirects: {Default: true, PreRelease: utilfeature.Beta}, AdvancedAuditing: {Default: false, PreRelease: utilfeature.Alpha}, APIResponseCompression: {Default: false, PreRelease: utilfeature.Alpha}, + Initializers: {Default: false, PreRelease: utilfeature.Alpha}, } diff --git a/pkg/registry/generic/registry/BUILD b/pkg/registry/generic/registry/BUILD index 82da56630..2a397147f 100644 --- a/pkg/registry/generic/registry/BUILD +++ b/pkg/registry/generic/registry/BUILD @@ -31,6 +31,7 @@ go_test( "//vendor/k8s.io/apiserver/pkg/apis/example:go_default_library", "//vendor/k8s.io/apiserver/pkg/apis/example/v1:go_default_library", "//vendor/k8s.io/apiserver/pkg/endpoints/request:go_default_library", + "//vendor/k8s.io/apiserver/pkg/features:go_default_library", "//vendor/k8s.io/apiserver/pkg/registry/generic:go_default_library", "//vendor/k8s.io/apiserver/pkg/registry/rest:go_default_library", "//vendor/k8s.io/apiserver/pkg/storage:go_default_library", @@ -39,6 +40,8 @@ go_test( "//vendor/k8s.io/apiserver/pkg/storage/names:go_default_library", "//vendor/k8s.io/apiserver/pkg/storage/storagebackend/factory:go_default_library", "//vendor/k8s.io/apiserver/pkg/storage/testing:go_default_library", + "//vendor/k8s.io/apiserver/pkg/util/feature:go_default_library", + "//vendor/k8s.io/apiserver/pkg/util/feature/testing:go_default_library", ], ) diff --git a/pkg/registry/generic/registry/store_test.go b/pkg/registry/generic/registry/store_test.go index cbe510eed..de03c9f63 100644 --- a/pkg/registry/generic/registry/store_test.go +++ b/pkg/registry/generic/registry/store_test.go @@ -44,6 +44,7 @@ import ( examplev1 "k8s.io/apiserver/pkg/apis/example/v1" "k8s.io/apiserver/pkg/endpoints/request" genericapirequest "k8s.io/apiserver/pkg/endpoints/request" + "k8s.io/apiserver/pkg/features" "k8s.io/apiserver/pkg/registry/generic" "k8s.io/apiserver/pkg/registry/rest" "k8s.io/apiserver/pkg/storage" @@ -52,6 +53,8 @@ import ( "k8s.io/apiserver/pkg/storage/names" "k8s.io/apiserver/pkg/storage/storagebackend/factory" storagetesting "k8s.io/apiserver/pkg/storage/testing" + utilfeature "k8s.io/apiserver/pkg/util/feature" + utilfeaturetesting "k8s.io/apiserver/pkg/util/feature/testing" ) var scheme = runtime.NewScheme() @@ -390,6 +393,8 @@ func isQualifiedResource(err error, kind, group string) bool { } func TestStoreCreateInitialized(t *testing.T) { + defer utilfeaturetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.Initializers, true)() + podA := &example.Pod{ ObjectMeta: metav1.ObjectMeta{ Name: "foo", Namespace: "test", @@ -493,6 +498,8 @@ func TestStoreCreateInitialized(t *testing.T) { } func TestStoreCreateInitializedFailed(t *testing.T) { + defer utilfeaturetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.Initializers, true)() + podA := &example.Pod{ ObjectMeta: metav1.ObjectMeta{ Name: "foo", Namespace: "test", @@ -991,6 +998,8 @@ func TestGracefulStoreHandleFinalizers(t *testing.T) { } func TestFailedInitializationStoreUpdate(t *testing.T) { + defer utilfeaturetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.Initializers, true)() + initialGeneration := int64(1) podInitializing := &example.Pod{ ObjectMeta: metav1.ObjectMeta{Name: "foo", Initializers: &metav1.Initializers{Pending: []metav1.Initializer{{Name: "Test"}}}, Generation: initialGeneration}, diff --git a/pkg/registry/rest/BUILD b/pkg/registry/rest/BUILD index 3e5b4d578..860516b3b 100644 --- a/pkg/registry/rest/BUILD +++ b/pkg/registry/rest/BUILD @@ -48,7 +48,9 @@ go_library( "//vendor/k8s.io/apimachinery/pkg/util/validation/field:go_default_library", "//vendor/k8s.io/apimachinery/pkg/watch:go_default_library", "//vendor/k8s.io/apiserver/pkg/endpoints/request:go_default_library", + "//vendor/k8s.io/apiserver/pkg/features:go_default_library", "//vendor/k8s.io/apiserver/pkg/storage/names:go_default_library", + "//vendor/k8s.io/apiserver/pkg/util/feature:go_default_library", ], ) diff --git a/pkg/registry/rest/create.go b/pkg/registry/rest/create.go index 25beba65f..55b628f03 100644 --- a/pkg/registry/rest/create.go +++ b/pkg/registry/rest/create.go @@ -26,7 +26,9 @@ import ( "k8s.io/apimachinery/pkg/runtime/schema" "k8s.io/apimachinery/pkg/util/validation/field" genericapirequest "k8s.io/apiserver/pkg/endpoints/request" + "k8s.io/apiserver/pkg/features" "k8s.io/apiserver/pkg/storage/names" + utilfeature "k8s.io/apiserver/pkg/util/feature" ) // RESTCreateStrategy defines the minimum validation, accepted input, and @@ -88,6 +90,11 @@ func BeforeCreate(strategy RESTCreateStrategy, ctx genericapirequest.Context, ob objectMeta.SetName(strategy.GenerateName(objectMeta.GetGenerateName())) } + // Ensure Initializers are not set unless the feature is enabled + if !utilfeature.DefaultFeatureGate.Enabled(features.Initializers) { + objectMeta.SetInitializers(nil) + } + // ClusterName is ignored and should not be saved objectMeta.SetClusterName("") diff --git a/pkg/registry/rest/update.go b/pkg/registry/rest/update.go index c50c3eac1..cc80db195 100644 --- a/pkg/registry/rest/update.go +++ b/pkg/registry/rest/update.go @@ -27,6 +27,8 @@ import ( "k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/util/validation/field" genericapirequest "k8s.io/apiserver/pkg/endpoints/request" + "k8s.io/apiserver/pkg/features" + utilfeature "k8s.io/apiserver/pkg/util/feature" ) // RESTUpdateStrategy defines the minimum validation, accepted input, and @@ -99,6 +101,12 @@ func BeforeUpdate(strategy RESTUpdateStrategy, ctx genericapirequest.Context, ob } objectMeta.SetGeneration(oldMeta.GetGeneration()) + // Ensure Initializers are not set unless the feature is enabled + if !utilfeature.DefaultFeatureGate.Enabled(features.Initializers) { + oldMeta.SetInitializers(nil) + objectMeta.SetInitializers(nil) + } + strategy.PrepareForUpdate(ctx, obj, old) // ClusterName is ignored and should not be saved diff --git a/pkg/util/feature/BUILD b/pkg/util/feature/BUILD index 582bf86bd..7788cf0af 100644 --- a/pkg/util/feature/BUILD +++ b/pkg/util/feature/BUILD @@ -31,6 +31,9 @@ filegroup( filegroup( name = "all-srcs", - srcs = [":package-srcs"], + srcs = [ + ":package-srcs", + "//staging/src/k8s.io/apiserver/pkg/util/feature/testing:all-srcs", + ], tags = ["automanaged"], ) diff --git a/pkg/util/feature/testing/BUILD b/pkg/util/feature/testing/BUILD new file mode 100644 index 000000000..e7c3b71dd --- /dev/null +++ b/pkg/util/feature/testing/BUILD @@ -0,0 +1,22 @@ +load("@io_bazel_rules_go//go:def.bzl", "go_library") + +go_library( + name = "go_default_library", + srcs = ["feature_gate_testing.go"], + visibility = ["//visibility:public"], + deps = ["//vendor/k8s.io/apiserver/pkg/util/feature:go_default_library"], +) + +filegroup( + name = "package-srcs", + srcs = glob(["**"]), + tags = ["automanaged"], + visibility = ["//visibility:private"], +) + +filegroup( + name = "all-srcs", + srcs = [":package-srcs"], + tags = ["automanaged"], + visibility = ["//visibility:public"], +) diff --git a/pkg/util/feature/testing/feature_gate_testing.go b/pkg/util/feature/testing/feature_gate_testing.go new file mode 100644 index 000000000..fdbd7735a --- /dev/null +++ b/pkg/util/feature/testing/feature_gate_testing.go @@ -0,0 +1,40 @@ +/* +Copyright 2017 The Kubernetes Authors. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +*/ + +package testing + +import ( + "fmt" + "testing" + + "k8s.io/apiserver/pkg/util/feature" +) + +// SetFeatureGateDuringTest sets the specified gate to the specified value, and returns a function that restores the original value. +// Failures to set or restore cause the test to fail. +func SetFeatureGateDuringTest(t *testing.T, gate feature.FeatureGate, feature feature.Feature, value bool) func() { + originalValue := gate.Enabled(feature) + + if err := gate.Set(fmt.Sprintf("%s=%v", feature, value)); err != nil { + t.Errorf("error setting %s=%v: %v", feature, value, err) + } + + return func() { + if err := gate.Set(fmt.Sprintf("%s=%v", feature, originalValue)); err != nil { + t.Errorf("error restoring %s=%v: %v", feature, originalValue, err) + } + } +}