support statefulset in kubectl autoscale command

2018-11-16 10:42:39 +08:00
parent 8797db339d
commit c3bb1c4601
3 changed files with 6 additions and 1 deletions
--- a/pkg/kubectl/polymorphichelpers/canbeautoscaled.go
+++ b/pkg/kubectl/polymorphichelpers/canbeautoscaled.go
@@ -31,6 +31,7 @@ func canBeAutoscaled(kind schema.GroupKind) error {
 		corev1.SchemeGroupVersion.WithKind("ReplicationController").GroupKind(),
 		appsv1.SchemeGroupVersion.WithKind("Deployment").GroupKind(),
 		appsv1.SchemeGroupVersion.WithKind("ReplicaSet").GroupKind(),
+		appsv1.SchemeGroupVersion.WithKind("StatefulSet").GroupKind(),
 		extensionsv1beta1.SchemeGroupVersion.WithKind("Deployment").GroupKind(),
 		extensionsv1beta1.SchemeGroupVersion.WithKind("ReplicaSet").GroupKind():
 		// nothing to do here
--- a/pkg/kubectl/polymorphichelpers/canbeautoscaled_test.go
+++ b/pkg/kubectl/polymorphichelpers/canbeautoscaled_test.go
@@ -38,6 +38,10 @@ func TestCanBeAutoscaled(t *testing.T) {
 			kind:      appsv1.SchemeGroupVersion.WithKind("Deployment").GroupKind(),
 			expectErr: false,
 		},
+		{
+			kind:      appsv1.SchemeGroupVersion.WithKind("StatefulSet").GroupKind(),
+			expectErr: false,
+		},
 		{
 			kind:      extensionsv1beta1.SchemeGroupVersion.WithKind("ReplicaSet").GroupKind(),
 			expectErr: false,