120 lines
4.4 KiB
Go
120 lines
4.4 KiB
Go
/*
|
|
Copyright 2018 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 apiutil contains utilities for working with raw Kubernetes
|
|
// API machinery, such as creating RESTMappers and raw REST clients,
|
|
// and extracting the GVK of an object.
|
|
package apiutil
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/apimachinery/pkg/runtime/serializer"
|
|
"k8s.io/client-go/discovery"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/restmapper"
|
|
)
|
|
|
|
// NewDiscoveryRESTMapper constructs a new RESTMapper based on discovery
|
|
// information fetched by a new client with the given config.
|
|
func NewDiscoveryRESTMapper(c *rest.Config) (meta.RESTMapper, error) {
|
|
// Get a mapper
|
|
dc, err := discovery.NewDiscoveryClientForConfig(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
gr, err := restmapper.GetAPIGroupResources(dc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return restmapper.NewDiscoveryRESTMapper(gr), nil
|
|
}
|
|
|
|
// GVKForObject finds the GroupVersionKind associated with the given object, if there is only a single such GVK.
|
|
func GVKForObject(obj runtime.Object, scheme *runtime.Scheme) (schema.GroupVersionKind, error) {
|
|
// TODO(directxman12): do we want to generalize this to arbitrary container types?
|
|
// I think we'd need a generalized form of scheme or something. It's a
|
|
// shame there's not a reliable "GetGVK" interface that works by default
|
|
// for unpopulated static types and populated "dynamic" types
|
|
// (unstructured, partial, etc)
|
|
|
|
// check for PartialObjectMetadata, which is analogous to unstructured, but isn't handled by ObjectKinds
|
|
_, isPartial := obj.(*metav1.PartialObjectMetadata)
|
|
_, isPartialList := obj.(*metav1.PartialObjectMetadataList)
|
|
if isPartial || isPartialList {
|
|
// we require that the GVK be populated in order to recognize the object
|
|
gvk := obj.GetObjectKind().GroupVersionKind()
|
|
if len(gvk.Kind) == 0 {
|
|
return schema.GroupVersionKind{}, runtime.NewMissingKindErr("unstructured object has no kind")
|
|
}
|
|
if len(gvk.Version) == 0 {
|
|
return schema.GroupVersionKind{}, runtime.NewMissingVersionErr("unstructured object has no version")
|
|
}
|
|
return gvk, nil
|
|
}
|
|
|
|
gvks, isUnversioned, err := scheme.ObjectKinds(obj)
|
|
if err != nil {
|
|
return schema.GroupVersionKind{}, err
|
|
}
|
|
if isUnversioned {
|
|
return schema.GroupVersionKind{}, fmt.Errorf("cannot create group-version-kind for unversioned type %T", obj)
|
|
}
|
|
|
|
if len(gvks) < 1 {
|
|
return schema.GroupVersionKind{}, fmt.Errorf("no group-version-kinds associated with type %T", obj)
|
|
}
|
|
if len(gvks) > 1 {
|
|
// this should only trigger for things like metav1.XYZ --
|
|
// normal versioned types should be fine
|
|
return schema.GroupVersionKind{}, fmt.Errorf(
|
|
"multiple group-version-kinds associated with type %T, refusing to guess at one", obj)
|
|
}
|
|
return gvks[0], nil
|
|
}
|
|
|
|
// RESTClientForGVK constructs a new rest.Interface capable of accessing the resource associated
|
|
// with the given GroupVersionKind. The REST client will be configured to use the negotiated serializer from
|
|
// baseConfig, if set, otherwise a default serializer will be set.
|
|
func RESTClientForGVK(gvk schema.GroupVersionKind, baseConfig *rest.Config, codecs serializer.CodecFactory) (rest.Interface, error) {
|
|
cfg := createRestConfig(gvk, baseConfig)
|
|
if cfg.NegotiatedSerializer == nil {
|
|
cfg.NegotiatedSerializer = serializer.WithoutConversionCodecFactory{CodecFactory: codecs}
|
|
}
|
|
return rest.RESTClientFor(cfg)
|
|
}
|
|
|
|
//createRestConfig copies the base config and updates needed fields for a new rest config
|
|
func createRestConfig(gvk schema.GroupVersionKind, baseConfig *rest.Config) *rest.Config {
|
|
gv := gvk.GroupVersion()
|
|
|
|
cfg := rest.CopyConfig(baseConfig)
|
|
cfg.GroupVersion = &gv
|
|
if gvk.Group == "" {
|
|
cfg.APIPath = "/api"
|
|
} else {
|
|
cfg.APIPath = "/apis"
|
|
}
|
|
if cfg.UserAgent == "" {
|
|
cfg.UserAgent = rest.DefaultKubernetesUserAgent()
|
|
}
|
|
return cfg
|
|
}
|