mirror of https://github.com/k3s-io/k3s
91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
|
package mapper
|
||
|
|
|
||
|
|
import (
|
||
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
||
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
||
|
|
"k8s.io/client-go/discovery"
|
||
|
|
"k8s.io/client-go/discovery/cached/memory"
|
||
|
|
"k8s.io/client-go/rest"
|
||
|
|
"k8s.io/client-go/restmapper"
|
||
|
|
)
|
||
|
|
|
||
|
|
func New(config *rest.Config) (meta.RESTMapper, error) {
|
||
|
|
d, err := discovery.NewDiscoveryClientForConfig(config)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
cached := memory.NewMemCacheClient(d)
|
||
|
|
return &retryMapper{
|
||
|
|
target: restmapper.NewDeferredDiscoveryRESTMapper(cached),
|
||
|
|
cache: cached,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type retryMapper struct {
|
||
|
|
target meta.RESTMapper
|
||
|
|
cache discovery.CachedDiscoveryInterface
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) KindFor(resource schema.GroupVersionResource) (schema.GroupVersionKind, error) {
|
||
|
|
result, err := r.target.KindFor(resource)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.KindFor(resource)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) KindsFor(resource schema.GroupVersionResource) ([]schema.GroupVersionKind, error) {
|
||
|
|
result, err := r.target.KindsFor(resource)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.KindsFor(resource)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) ResourceFor(input schema.GroupVersionResource) (schema.GroupVersionResource, error) {
|
||
|
|
result, err := r.target.ResourceFor(input)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.ResourceFor(input)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) ResourcesFor(input schema.GroupVersionResource) ([]schema.GroupVersionResource, error) {
|
||
|
|
result, err := r.target.ResourcesFor(input)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.ResourcesFor(input)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) RESTMapping(gk schema.GroupKind, versions ...string) (*meta.RESTMapping, error) {
|
||
|
|
result, err := r.target.RESTMapping(gk, versions...)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.RESTMapping(gk, versions...)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) RESTMappings(gk schema.GroupKind, versions ...string) ([]*meta.RESTMapping, error) {
|
||
|
|
result, err := r.target.RESTMappings(gk, versions...)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.RESTMappings(gk, versions...)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *retryMapper) ResourceSingularizer(resource string) (singular string, err error) {
|
||
|
|
result, err := r.target.ResourceSingularizer(resource)
|
||
|
|
if err != nil {
|
||
|
|
r.cache.Invalidate()
|
||
|
|
return r.target.ResourceSingularizer(resource)
|
||
|
|
}
|
||
|
|
return result, err
|
||
|
|
}
|