crd/overlayextensionconfig/client.go (64 lines of code) (raw):
package overlayextensionconfig
import (
"context"
"reflect"
"github.com/Azure/azure-container-networking/crd"
"github.com/Azure/azure-container-networking/crd/overlayextensionconfig/api/v1alpha1"
"github.com/pkg/errors"
v1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
typedv1 "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset/typed/apiextensions/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/client-go/kubernetes/scheme"
"k8s.io/client-go/rest"
)
// Scheme is a runtime scheme containing the client-go scheme and the OverlayExtensionConfig scheme.
var Scheme = runtime.NewScheme()
func init() {
_ = scheme.AddToScheme(Scheme)
_ = v1alpha1.AddToScheme(Scheme)
}
// Installer provides methods to manage the lifecycle of the OverlayExtensionConfig resource definition.
type Installer struct {
cli typedv1.CustomResourceDefinitionInterface
}
func NewInstaller(c *rest.Config) (*Installer, error) {
cli, err := crd.NewCRDClientFromConfig(c)
if err != nil {
return nil, errors.Wrap(err, "failed to init crd client")
}
return &Installer{
cli: cli,
}, nil
}
func (i *Installer) create(ctx context.Context, res *v1.CustomResourceDefinition) (*v1.CustomResourceDefinition, error) {
res, err := i.cli.Create(ctx, res, metav1.CreateOptions{})
if err != nil {
return nil, errors.Wrap(err, "failed to create oec crd")
}
return res, nil
}
// InstallOrUpdate installs the embedded OverlayExtensionConfig CRD definition in the cluster or updates it if present.
func (i *Installer) InstallOrUpdate(ctx context.Context) (*v1.CustomResourceDefinition, error) {
oec, err := GetOverlayExtensionConfigs()
if err != nil {
return nil, errors.Wrap(err, "failed to get embedded oec crd")
}
current, err := i.create(ctx, oec)
if !apierrors.IsAlreadyExists(err) {
return current, err
}
if current == nil {
current, err = i.cli.Get(ctx, oec.Name, metav1.GetOptions{})
if err != nil {
return nil, errors.Wrap(err, "failed to get existing oec crd")
}
}
if !reflect.DeepEqual(oec.Spec.Versions, current.Spec.Versions) {
oec.SetResourceVersion(current.GetResourceVersion())
previous := *current
current, err = i.cli.Update(ctx, oec, metav1.UpdateOptions{})
if err != nil {
return &previous, errors.Wrap(err, "failed to update existing oec crd")
}
}
return current, nil
}