mirror of
https://github.com/kubesphere/kubekey.git
synced 2025-12-30 15:12:54 +00:00
113 lines
3.8 KiB
Go
113 lines
3.8 KiB
Go
/*
|
|
Copyright 2024 The KubeSphere 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 task
|
|
|
|
import (
|
|
"context"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
apigeneric "k8s.io/apiserver/pkg/registry/generic"
|
|
apiregistry "k8s.io/apiserver/pkg/registry/generic/registry"
|
|
apirest "k8s.io/apiserver/pkg/registry/rest"
|
|
apistorage "k8s.io/apiserver/pkg/storage"
|
|
"sigs.k8s.io/structured-merge-diff/v4/fieldpath"
|
|
|
|
kubekeyv1alpha1 "github.com/kubesphere/kubekey/v4/pkg/apis/kubekey/v1alpha1"
|
|
)
|
|
|
|
type TaskStorage struct {
|
|
Task *REST
|
|
TaskStatus *StatusREST
|
|
}
|
|
|
|
type REST struct {
|
|
*apiregistry.Store
|
|
}
|
|
|
|
type StatusREST struct {
|
|
store *apiregistry.Store
|
|
}
|
|
|
|
func (r *StatusREST) NamespaceScoped() bool {
|
|
return true
|
|
}
|
|
|
|
// New creates a new Node object.
|
|
func (r *StatusREST) New() runtime.Object {
|
|
return &kubekeyv1alpha1.Task{}
|
|
}
|
|
|
|
// Destroy cleans up resources on shutdown.
|
|
func (r *StatusREST) Destroy() {
|
|
// Given that underlying store is shared with REST,
|
|
// we don't destroy it here explicitly.
|
|
}
|
|
|
|
// 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 apirest.UpdatedObjectInfo, createValidation apirest.ValidateObjectFunc, updateValidation apirest.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)
|
|
}
|
|
|
|
// GetResetFields implements rest.ResetFieldsStrategy
|
|
func (r *StatusREST) GetResetFields() map[fieldpath.APIVersion]*fieldpath.Set {
|
|
return r.store.GetResetFields()
|
|
}
|
|
|
|
func (r *StatusREST) ConvertToTable(ctx context.Context, object runtime.Object, tableOptions runtime.Object) (*metav1.Table, error) {
|
|
return r.store.ConvertToTable(ctx, object, tableOptions)
|
|
}
|
|
|
|
func NewStorage(optsGetter apigeneric.RESTOptionsGetter) (TaskStorage, error) {
|
|
store := &apiregistry.Store{
|
|
NewFunc: func() runtime.Object { return &kubekeyv1alpha1.Task{} },
|
|
NewListFunc: func() runtime.Object { return &kubekeyv1alpha1.TaskList{} },
|
|
PredicateFunc: MatchTask,
|
|
DefaultQualifiedResource: kubekeyv1alpha1.SchemeGroupVersion.WithResource("tasks").GroupResource(),
|
|
SingularQualifiedResource: kubekeyv1alpha1.SchemeGroupVersion.WithResource("task").GroupResource(),
|
|
|
|
CreateStrategy: Strategy,
|
|
UpdateStrategy: Strategy,
|
|
DeleteStrategy: Strategy,
|
|
//ResetFieldsStrategy: Strategy,
|
|
ReturnDeletedObject: true,
|
|
|
|
TableConvertor: apirest.NewDefaultTableConvertor(kubekeyv1alpha1.SchemeGroupVersion.WithResource("tasks").GroupResource()),
|
|
}
|
|
options := &apigeneric.StoreOptions{
|
|
RESTOptions: optsGetter,
|
|
AttrFunc: GetAttrs,
|
|
TriggerFunc: map[string]apistorage.IndexerFunc{"ownerReferences:pipeline": OwnerPipelineTriggerFunc},
|
|
Indexers: Indexers(),
|
|
}
|
|
if err := store.CompleteWithOptions(options); err != nil {
|
|
return TaskStorage{}, err
|
|
}
|
|
|
|
return TaskStorage{
|
|
Task: &REST{store},
|
|
TaskStatus: &StatusREST{store},
|
|
}, nil
|
|
}
|