mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-12-13 08:30:23 +00:00
98 lines
3.3 KiB
Go
98 lines
3.3 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"
|
||
|
"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) {
|
||
|
gvks, isUnversioned, err := scheme.ObjectKinds(obj)
|
||
|
if err != nil {
|
||
|
return schema.GroupVersionKind{}, err
|
||
|
}
|
||
|
if isUnversioned {
|
||
|
return schema.GroupVersionKind{}, fmt.Errorf("cannot create a new informer for the 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
|
||
|
}
|