mirror of https://github.com/k3s-io/k3s
113 lines
4.1 KiB
Go
113 lines
4.1 KiB
Go
/*
|
|
Copyright 2015 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 cache
|
|
|
|
import (
|
|
"context"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/fields"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/watch"
|
|
restclient "k8s.io/client-go/rest"
|
|
)
|
|
|
|
// Lister is any object that knows how to perform an initial list.
|
|
type Lister interface {
|
|
// List should return a list type object; the Items field will be extracted, and the
|
|
// ResourceVersion field will be used to start the watch in the right place.
|
|
List(options metav1.ListOptions) (runtime.Object, error)
|
|
}
|
|
|
|
// Watcher is any object that knows how to start a watch on a resource.
|
|
type Watcher interface {
|
|
// Watch should begin a watch at the specified version.
|
|
Watch(options metav1.ListOptions) (watch.Interface, error)
|
|
}
|
|
|
|
// ListerWatcher is any object that knows how to perform an initial list and start a watch on a resource.
|
|
type ListerWatcher interface {
|
|
Lister
|
|
Watcher
|
|
}
|
|
|
|
// ListFunc knows how to list resources
|
|
type ListFunc func(options metav1.ListOptions) (runtime.Object, error)
|
|
|
|
// WatchFunc knows how to watch resources
|
|
type WatchFunc func(options metav1.ListOptions) (watch.Interface, error)
|
|
|
|
// ListWatch knows how to list and watch a set of apiserver resources. It satisfies the ListerWatcher interface.
|
|
// It is a convenience function for users of NewReflector, etc.
|
|
// ListFunc and WatchFunc must not be nil
|
|
type ListWatch struct {
|
|
ListFunc ListFunc
|
|
WatchFunc WatchFunc
|
|
// DisableChunking requests no chunking for this list watcher.
|
|
DisableChunking bool
|
|
}
|
|
|
|
// Getter interface knows how to access Get method from RESTClient.
|
|
type Getter interface {
|
|
Get() *restclient.Request
|
|
}
|
|
|
|
// NewListWatchFromClient creates a new ListWatch from the specified client, resource, namespace and field selector.
|
|
func NewListWatchFromClient(c Getter, resource string, namespace string, fieldSelector fields.Selector) *ListWatch {
|
|
optionsModifier := func(options *metav1.ListOptions) {
|
|
options.FieldSelector = fieldSelector.String()
|
|
}
|
|
return NewFilteredListWatchFromClient(c, resource, namespace, optionsModifier)
|
|
}
|
|
|
|
// NewFilteredListWatchFromClient creates a new ListWatch from the specified client, resource, namespace, and option modifier.
|
|
// Option modifier is a function takes a ListOptions and modifies the consumed ListOptions. Provide customized modifier function
|
|
// to apply modification to ListOptions with a field selector, a label selector, or any other desired options.
|
|
func NewFilteredListWatchFromClient(c Getter, resource string, namespace string, optionsModifier func(options *metav1.ListOptions)) *ListWatch {
|
|
listFunc := func(options metav1.ListOptions) (runtime.Object, error) {
|
|
optionsModifier(&options)
|
|
return c.Get().
|
|
Namespace(namespace).
|
|
Resource(resource).
|
|
VersionedParams(&options, metav1.ParameterCodec).
|
|
Do(context.TODO()).
|
|
Get()
|
|
}
|
|
watchFunc := func(options metav1.ListOptions) (watch.Interface, error) {
|
|
options.Watch = true
|
|
optionsModifier(&options)
|
|
return c.Get().
|
|
Namespace(namespace).
|
|
Resource(resource).
|
|
VersionedParams(&options, metav1.ParameterCodec).
|
|
Watch(context.TODO())
|
|
}
|
|
return &ListWatch{ListFunc: listFunc, WatchFunc: watchFunc}
|
|
}
|
|
|
|
// List a set of apiserver resources
|
|
func (lw *ListWatch) List(options metav1.ListOptions) (runtime.Object, error) {
|
|
// ListWatch is used in Reflector, which already supports pagination.
|
|
// Don't paginate here to avoid duplication.
|
|
return lw.ListFunc(options)
|
|
}
|
|
|
|
// Watch a set of apiserver resources
|
|
func (lw *ListWatch) Watch(options metav1.ListOptions) (watch.Interface, error) {
|
|
return lw.WatchFunc(options)
|
|
}
|