8456f51d7f
Move tenant to default system ns Delete auto network when tenant is deleted Implements: blueprint cni-controller Change-Id: I886d9ee4892297461c7a1d2eb02570c8c14bd581
133 lines
3.2 KiB
Go
133 lines
3.2 KiB
Go
package kubecrd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/serializer"
|
|
"k8s.io/client-go/rest"
|
|
|
|
crv1 "git.openstack.org/openstack/stackube/pkg/apis/v1"
|
|
"git.openstack.org/openstack/stackube/pkg/util"
|
|
|
|
"github.com/golang/glog"
|
|
)
|
|
|
|
type CRDClient struct {
|
|
Client *rest.RESTClient
|
|
Scheme *runtime.Scheme
|
|
}
|
|
|
|
func NewCRDClient(cfg *rest.Config) (*CRDClient, error) {
|
|
scheme := runtime.NewScheme()
|
|
if err := crv1.AddToScheme(scheme); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
config := *cfg
|
|
config.GroupVersion = &crv1.SchemeGroupVersion
|
|
config.APIPath = "/apis"
|
|
config.ContentType = runtime.ContentTypeJSON
|
|
config.NegotiatedSerializer = serializer.DirectCodecFactory{CodecFactory: serializer.NewCodecFactory(scheme)}
|
|
|
|
client, err := rest.RESTClientFor(&config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &CRDClient{
|
|
Client: client,
|
|
Scheme: scheme,
|
|
}, nil
|
|
}
|
|
|
|
// UpdateNetwork updates Network CRD object by given object
|
|
func (c *CRDClient) UpdateNetwork(network *crv1.Network) {
|
|
err := c.Client.Put().
|
|
Name(network.Name).
|
|
Namespace(network.Namespace).
|
|
Resource(crv1.NetworkResourcePlural).
|
|
Body(network).
|
|
Do().
|
|
Error()
|
|
|
|
if err != nil {
|
|
glog.Errorf("ERROR updating network: %v\n", err)
|
|
} else {
|
|
glog.V(3).Infof("UPDATED network: %#v\n", network)
|
|
}
|
|
}
|
|
|
|
// UpdateTenant updates Network CRD object by given object
|
|
func (c *CRDClient) UpdateTenant(tenant *crv1.Tenant) {
|
|
err := c.Client.Put().
|
|
Name(tenant.Name).
|
|
Namespace(util.SystemTenant).
|
|
Resource(crv1.TenantResourcePlural).
|
|
Body(tenant).
|
|
Do().
|
|
Error()
|
|
|
|
if err != nil {
|
|
glog.Errorf("ERROR updating tenant: %v\n", err)
|
|
} else {
|
|
glog.V(3).Infof("UPDATED tenant: %#v\n", tenant)
|
|
}
|
|
}
|
|
|
|
// GetTenant returns tenant from CRD
|
|
// NOTE: all tenant are stored under system namespace
|
|
func (c *CRDClient) GetTenant(tenantName string) (*crv1.Tenant, error) {
|
|
tenant := crv1.Tenant{}
|
|
// tenant always has same name and namespace
|
|
err := c.Client.Get().
|
|
Resource(crv1.TenantResourcePlural).
|
|
Namespace(util.SystemTenant).
|
|
Name(tenantName).
|
|
Do().Into(&tenant)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &tenant, nil
|
|
}
|
|
|
|
// AddTenant adds tenant to CRD
|
|
// NOTE: all tenant are added to system namespace
|
|
func (c *CRDClient) AddTenant(tenant *crv1.Tenant) error {
|
|
err := c.Client.Post().
|
|
Namespace(util.SystemTenant).
|
|
Resource(crv1.TenantResourcePlural).
|
|
Body(tenant).
|
|
Do().Error()
|
|
if err != nil && !apierrors.IsAlreadyExists(err) {
|
|
return fmt.Errorf("failed to create Tenant: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *CRDClient) AddNetwork(network *crv1.Network) error {
|
|
err := c.Client.Post().
|
|
Resource(crv1.NetworkResourcePlural).
|
|
Namespace(network.GetNamespace()).
|
|
Body(network).
|
|
Do().Error()
|
|
if err != nil && !apierrors.IsAlreadyExists(err) {
|
|
return fmt.Errorf("failed to create Network: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *CRDClient) DeleteNetork(namespace string) error {
|
|
// NOTE: the automatically created network for tenant use namespace as name
|
|
err := c.Client.Delete().
|
|
Resource(crv1.NetworkResourcePlural).
|
|
Namespace(namespace).
|
|
Name(namespace).
|
|
Do().Error()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to delete Network: %v", err)
|
|
}
|
|
return nil
|
|
}
|