| /* |
| Copyright 2016 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. |
| */ |
| |
| // If you make changes to this file, you should also make the corresponding change in ReplicationController. |
| |
| package storage |
| |
| import ( |
| "context" |
| "fmt" |
| |
| "k8s.io/apimachinery/pkg/api/errors" |
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" |
| "k8s.io/apimachinery/pkg/runtime" |
| "k8s.io/apimachinery/pkg/runtime/schema" |
| "k8s.io/apiserver/pkg/registry/generic" |
| genericregistry "k8s.io/apiserver/pkg/registry/generic/registry" |
| "k8s.io/apiserver/pkg/registry/rest" |
| "k8s.io/kubernetes/pkg/apis/apps" |
| appsv1beta1 "k8s.io/kubernetes/pkg/apis/apps/v1beta1" |
| appsv1beta2 "k8s.io/kubernetes/pkg/apis/apps/v1beta2" |
| "k8s.io/kubernetes/pkg/apis/autoscaling" |
| autoscalingv1 "k8s.io/kubernetes/pkg/apis/autoscaling/v1" |
| autoscalingvalidation "k8s.io/kubernetes/pkg/apis/autoscaling/validation" |
| extensionsv1beta1 "k8s.io/kubernetes/pkg/apis/extensions/v1beta1" |
| "k8s.io/kubernetes/pkg/printers" |
| printersinternal "k8s.io/kubernetes/pkg/printers/internalversion" |
| printerstorage "k8s.io/kubernetes/pkg/printers/storage" |
| "k8s.io/kubernetes/pkg/registry/apps/replicaset" |
| ) |
| |
| // ReplicaSetStorage includes dummy storage for ReplicaSets and for Scale subresource. |
| type ReplicaSetStorage struct { |
| ReplicaSet *REST |
| Status *StatusREST |
| Scale *ScaleREST |
| } |
| |
| func NewStorage(optsGetter generic.RESTOptionsGetter) ReplicaSetStorage { |
| replicaSetRest, replicaSetStatusRest := NewREST(optsGetter) |
| |
| return ReplicaSetStorage{ |
| ReplicaSet: replicaSetRest, |
| Status: replicaSetStatusRest, |
| Scale: &ScaleREST{store: replicaSetRest.Store}, |
| } |
| } |
| |
| type REST struct { |
| *genericregistry.Store |
| categories []string |
| } |
| |
| // NewREST returns a RESTStorage object that will work against ReplicaSet. |
| func NewREST(optsGetter generic.RESTOptionsGetter) (*REST, *StatusREST) { |
| store := &genericregistry.Store{ |
| NewFunc: func() runtime.Object { return &apps.ReplicaSet{} }, |
| NewListFunc: func() runtime.Object { return &apps.ReplicaSetList{} }, |
| PredicateFunc: replicaset.MatchReplicaSet, |
| DefaultQualifiedResource: apps.Resource("replicasets"), |
| |
| CreateStrategy: replicaset.Strategy, |
| UpdateStrategy: replicaset.Strategy, |
| DeleteStrategy: replicaset.Strategy, |
| |
| TableConvertor: printerstorage.TableConvertor{TablePrinter: printers.NewTablePrinter().With(printersinternal.AddHandlers)}, |
| } |
| options := &generic.StoreOptions{RESTOptions: optsGetter, AttrFunc: replicaset.GetAttrs} |
| if err := store.CompleteWithOptions(options); err != nil { |
| panic(err) // TODO: Propagate error up |
| } |
| |
| statusStore := *store |
| statusStore.UpdateStrategy = replicaset.StatusStrategy |
| |
| return &REST{store, []string{"all"}}, &StatusREST{store: &statusStore} |
| } |
| |
| // Implement ShortNamesProvider |
| var _ rest.ShortNamesProvider = &REST{} |
| |
| // ShortNames implements the ShortNamesProvider interface. Returns a list of short names for a resource. |
| func (r *REST) ShortNames() []string { |
| return []string{"rs"} |
| } |
| |
| // Implement CategoriesProvider |
| var _ rest.CategoriesProvider = &REST{} |
| |
| // Categories implements the CategoriesProvider interface. Returns a list of categories a resource is part of. |
| func (r *REST) Categories() []string { |
| return r.categories |
| } |
| |
| func (r *REST) WithCategories(categories []string) *REST { |
| r.categories = categories |
| return r |
| } |
| |
| // StatusREST implements the REST endpoint for changing the status of a ReplicaSet |
| type StatusREST struct { |
| store *genericregistry.Store |
| } |
| |
| func (r *StatusREST) New() runtime.Object { |
| return &apps.ReplicaSet{} |
| } |
| |
| // Get retrieves the object from the storage. It is required to support Patch. |
| func (r *StatusREST) Get(ctx context.Context, name string, options *metav1.GetOptions) (runtime.Object, error) { |
| return r.store.Get(ctx, name, options) |
| } |
| |
| // Update alters the status subset of an object. |
| func (r *StatusREST) Update(ctx context.Context, name string, objInfo rest.UpdatedObjectInfo, createValidation rest.ValidateObjectFunc, updateValidation rest.ValidateObjectUpdateFunc, forceAllowCreate bool, options *metav1.UpdateOptions) (runtime.Object, bool, error) { |
| // We are explicitly setting forceAllowCreate to false in the call to the underlying storage because |
| // subresources should never allow create on update. |
| return r.store.Update(ctx, name, objInfo, createValidation, updateValidation, false, options) |
| } |
| |
| type ScaleREST struct { |
| store *genericregistry.Store |
| } |
| |
| // ScaleREST implements Patcher |
| var _ = rest.Patcher(&ScaleREST{}) |
| var _ = rest.GroupVersionKindProvider(&ScaleREST{}) |
| |
| func (r *ScaleREST) GroupVersionKind(containingGV schema.GroupVersion) schema.GroupVersionKind { |
| switch containingGV { |
| case extensionsv1beta1.SchemeGroupVersion: |
| return extensionsv1beta1.SchemeGroupVersion.WithKind("Scale") |
| case appsv1beta1.SchemeGroupVersion: |
| return appsv1beta1.SchemeGroupVersion.WithKind("Scale") |
| case appsv1beta2.SchemeGroupVersion: |
| return appsv1beta2.SchemeGroupVersion.WithKind("Scale") |
| default: |
| return autoscalingv1.SchemeGroupVersion.WithKind("Scale") |
| } |
| } |
| |
| // New creates a new Scale object |
| func (r *ScaleREST) New() runtime.Object { |
| return &autoscaling.Scale{} |
| } |
| |
| func (r *ScaleREST) Get(ctx context.Context, name string, options *metav1.GetOptions) (runtime.Object, error) { |
| obj, err := r.store.Get(ctx, name, options) |
| if err != nil { |
| return nil, errors.NewNotFound(apps.Resource("replicasets/scale"), name) |
| } |
| rs := obj.(*apps.ReplicaSet) |
| scale, err := scaleFromReplicaSet(rs) |
| if err != nil { |
| return nil, errors.NewBadRequest(fmt.Sprintf("%v", err)) |
| } |
| return scale, err |
| } |
| |
| func (r *ScaleREST) Update(ctx context.Context, name string, objInfo rest.UpdatedObjectInfo, createValidation rest.ValidateObjectFunc, updateValidation rest.ValidateObjectUpdateFunc, forceAllowCreate bool, options *metav1.UpdateOptions) (runtime.Object, bool, error) { |
| obj, err := r.store.Get(ctx, name, &metav1.GetOptions{}) |
| if err != nil { |
| return nil, false, errors.NewNotFound(apps.Resource("replicasets/scale"), name) |
| } |
| rs := obj.(*apps.ReplicaSet) |
| |
| oldScale, err := scaleFromReplicaSet(rs) |
| if err != nil { |
| return nil, false, err |
| } |
| |
| // TODO: should this pass admission? |
| obj, err = objInfo.UpdatedObject(ctx, oldScale) |
| if err != nil { |
| return nil, false, err |
| } |
| if obj == nil { |
| return nil, false, errors.NewBadRequest(fmt.Sprintf("nil update passed to Scale")) |
| } |
| scale, ok := obj.(*autoscaling.Scale) |
| if !ok { |
| return nil, false, errors.NewBadRequest(fmt.Sprintf("wrong object passed to Scale update: %v", obj)) |
| } |
| |
| if errs := autoscalingvalidation.ValidateScale(scale); len(errs) > 0 { |
| return nil, false, errors.NewInvalid(autoscaling.Kind("Scale"), scale.Name, errs) |
| } |
| |
| rs.Spec.Replicas = scale.Spec.Replicas |
| rs.ResourceVersion = scale.ResourceVersion |
| obj, _, err = r.store.Update(ctx, rs.Name, rest.DefaultUpdatedObjectInfo(rs), createValidation, updateValidation, false, options) |
| if err != nil { |
| return nil, false, err |
| } |
| rs = obj.(*apps.ReplicaSet) |
| newScale, err := scaleFromReplicaSet(rs) |
| if err != nil { |
| return nil, false, errors.NewBadRequest(fmt.Sprintf("%v", err)) |
| } |
| return newScale, false, err |
| } |
| |
| // scaleFromReplicaSet returns a scale subresource for a replica set. |
| func scaleFromReplicaSet(rs *apps.ReplicaSet) (*autoscaling.Scale, error) { |
| selector, err := metav1.LabelSelectorAsSelector(rs.Spec.Selector) |
| if err != nil { |
| return nil, err |
| } |
| return &autoscaling.Scale{ |
| // TODO: Create a variant of ObjectMeta type that only contains the fields below. |
| ObjectMeta: metav1.ObjectMeta{ |
| Name: rs.Name, |
| Namespace: rs.Namespace, |
| UID: rs.UID, |
| ResourceVersion: rs.ResourceVersion, |
| CreationTimestamp: rs.CreationTimestamp, |
| }, |
| Spec: autoscaling.ScaleSpec{ |
| Replicas: rs.Spec.Replicas, |
| }, |
| Status: autoscaling.ScaleStatus{ |
| Replicas: rs.Status.Replicas, |
| Selector: selector.String(), |
| }, |
| }, nil |
| } |