Rework auth controller

This PR reworks auth controller to use informer framework.
It also fix the problem of tenant and user not created.

Change-Id: I017032f2eb4d83440319729d9f1fb13351f4d72b
Closes-Bug: 1702841
Signed-off-by: Pengfei Ni <feiskyer@gmail.com>
This commit is contained in:
Pengfei Ni 2017-07-07 15:44:11 +08:00
parent f52aae3420
commit 7bb760620f
8 changed files with 335 additions and 603 deletions

View File

@ -26,15 +26,15 @@ var (
"path to stackube config file")
)
func startControllers(cfg tenant.Config) error {
func startControllers(kubeconfig, cloudconfig string) error {
// Creates a new tenant controller
tc, err := tenant.New(cfg)
tc, err := tenant.New(kubeconfig, cloudconfig)
if err != nil {
return err
}
// Creates a new RBAC controller
rm, err := rbacmanager.New(cfg)
rm, err := rbacmanager.New(kubeconfig)
if err != nil {
return err
}
@ -47,9 +47,7 @@ func startControllers(cfg tenant.Config) error {
wg.Go(func() error { return rm.Run(ctx.Done()) })
networkController, err := network.NewNetworkController(
cfg.KubeConfig,
cfg.CloudConfig,
)
kubeconfig, cloudconfig)
if err != nil {
return err
}
@ -106,11 +104,7 @@ func main() {
}
// Start stackube controllers.
cfg := tenant.Config{
KubeConfig: *kubeconfig,
CloudConfig: *cloudconfig,
}
if err := startControllers(cfg); err != nil {
if err := startControllers(*kubeconfig, *cloudconfig); err != nil {
glog.Fatal(err)
}
}

View File

@ -26,6 +26,21 @@ const (
NetworkTerminating = "Terminating"
)
// These are the valid phases of a tenant state.
const (
// TenantInitializing means the tenant is just accepted by system
TenantInitializing = "Initializing"
// TenantActive means the tenant is available for use in the system
TenantActive = "Active"
// TenantPending means the tenant is accepted by system, but it is still
// processing by tenant provider
TenantPending = "Pending"
// TenantFailed means the tenant is not available
TenantFailed = "Failed"
// TenantTerminating means the tenant is undergoing graceful termination
TenantTerminating = "Terminating"
)
// Network describes a Neutron network.
type Network struct {
// TypeMeta defines type of the object and its API schema version.

View File

@ -1,61 +1,29 @@
package auth
import (
"k8s.io/apimachinery/pkg/runtime/schema"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/runtime/serializer"
"k8s.io/client-go/dynamic"
"k8s.io/client-go/pkg/api"
"k8s.io/client-go/rest"
crv1 "git.openstack.org/openstack/stackube/pkg/apis/v1"
)
var (
CRDGroup = crv1.GroupName
CRDVersion = crv1.SchemeGroupVersion.Version
)
func NewClient(cfg *rest.Config) (*rest.RESTClient, *runtime.Scheme, error) {
scheme := runtime.NewScheme()
if err := crv1.AddToScheme(scheme); err != nil {
return nil, nil, err
}
type AuthInterface interface {
RESTClient() rest.Interface
TenantsGetter
//TODO: add networkgetter
}
config := *cfg
config.GroupVersion = &crv1.SchemeGroupVersion
config.APIPath = "/apis"
config.ContentType = runtime.ContentTypeJSON
config.NegotiatedSerializer = serializer.DirectCodecFactory{CodecFactory: serializer.NewCodecFactory(scheme)}
type AuthClient struct {
restClient rest.Interface
dynamicClient *dynamic.Client
}
func (c *AuthClient) Tenants(namespace string) TenantInterface {
return newTenants(c.restClient, c.dynamicClient, namespace)
}
func (c *AuthClient) RESTClient() rest.Interface {
return c.restClient
}
func NewForConfig(c *rest.Config) (*AuthClient, error) {
config := *c
setConfigDefaults(&config)
client, err := rest.RESTClientFor(&config)
if err != nil {
return nil, err
return nil, nil, err
}
dynamicClient, err := dynamic.NewClient(&config)
if err != nil {
return nil, err
}
return &AuthClient{client, dynamicClient}, nil
}
func setConfigDefaults(config *rest.Config) {
config.GroupVersion = &schema.GroupVersion{
Group: CRDGroup,
Version: CRDVersion,
}
config.APIPath = "/apis"
config.NegotiatedSerializer = serializer.DirectCodecFactory{CodecFactory: api.Codecs}
return
return client, scheme, nil
}

View File

@ -1,175 +1,65 @@
package auth
import (
"encoding/json"
"reflect"
"time"
"git.openstack.org/openstack/stackube/pkg/apis/v1"
crv1 "git.openstack.org/openstack/stackube/pkg/apis/v1"
"git.openstack.org/openstack/stackube/pkg/util"
apiextensionsv1beta1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1beta1"
apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/watch"
"k8s.io/client-go/dynamic"
"k8s.io/apimachinery/pkg/util/wait"
apiv1 "k8s.io/client-go/pkg/api/v1"
"k8s.io/client-go/rest"
)
const (
CRDTenantsKind = "Tenant"
CRDTenantName = "tenants"
tenantCRDName = crv1.TenantResourcePlural + "." + crv1.GroupName
)
type TenantsGetter interface {
Tenants(namespace string) TenantInterface
}
type TenantInterface interface {
Create(*v1.Tenant) (*v1.Tenant, error)
Get(name string) (*v1.Tenant, error)
Update(*v1.Tenant) (*v1.Tenant, error)
Delete(name string, options *metav1.DeleteOptions) error
List(opts metav1.ListOptions) (runtime.Object, error)
Watch(opts metav1.ListOptions) (watch.Interface, error)
}
type tenants struct {
restClient rest.Interface
client *dynamic.ResourceClient
ns string
}
func newTenants(r rest.Interface, c *dynamic.Client, namespace string) *tenants {
return &tenants{
r,
c.Resource(
&metav1.APIResource{
Kind: CRDTenantsKind,
Name: CRDTenantName,
Namespaced: true,
func CreateTenantCRD(clientset apiextensionsclient.Interface) (*apiextensionsv1beta1.CustomResourceDefinition, error) {
crd := &apiextensionsv1beta1.CustomResourceDefinition{
ObjectMeta: metav1.ObjectMeta{
Name: tenantCRDName,
},
Spec: apiextensionsv1beta1.CustomResourceDefinitionSpec{
Group: crv1.GroupName,
Version: crv1.SchemeGroupVersion.Version,
Scope: apiextensionsv1beta1.NamespaceScoped,
Names: apiextensionsv1beta1.CustomResourceDefinitionNames{
Plural: crv1.TenantResourcePlural,
Kind: reflect.TypeOf(crv1.Tenant{}).Name(),
},
namespace,
),
namespace,
},
}
}
func (p *tenants) Create(o *v1.Tenant) (*v1.Tenant, error) {
up, err := UnstructuredFromTenant(o)
_, err := clientset.ApiextensionsV1beta1().CustomResourceDefinitions().Create(crd)
if err != nil {
return nil, err
}
up, err = p.client.Create(up)
if err != nil {
// wait for CRD being established
if err = util.WaitForCRDReady(clientset, tenantCRDName); err != nil {
return nil, err
} else {
return crd, nil
}
return TenantFromUnstructured(up)
}
func (p *tenants) Get(name string) (*v1.Tenant, error) {
obj, err := p.client.Get(name)
if err != nil {
return nil, err
}
return TenantFromUnstructured(obj)
}
func (p *tenants) Update(o *v1.Tenant) (*v1.Tenant, error) {
up, err := UnstructuredFromTenant(o)
if err != nil {
return nil, err
}
up, err = p.client.Update(up)
if err != nil {
return nil, err
}
return TenantFromUnstructured(up)
}
func (p *tenants) Delete(name string, options *metav1.DeleteOptions) error {
return p.client.Delete(name, options)
}
func (p *tenants) List(opts metav1.ListOptions) (runtime.Object, error) {
req := p.restClient.Get().
Namespace(p.ns).
Resource("tenants").
// VersionedParams(&options, v1.ParameterCodec)
FieldsSelectorParam(nil)
b, err := req.DoRaw()
if err != nil {
return nil, err
}
var tena v1.TenantList
return &tena, json.Unmarshal(b, &tena)
}
func (p *tenants) Watch(opts metav1.ListOptions) (watch.Interface, error) {
r, err := p.restClient.Get().
Prefix("watch").
Namespace(p.ns).
Resource("tenants").
// VersionedParams(&options, v1.ParameterCodec).
FieldsSelectorParam(nil).
Stream()
if err != nil {
return nil, err
}
return watch.NewStreamWatcher(&tenantDecoder{
dec: json.NewDecoder(r),
close: r.Close,
}), nil
}
// TenantFromUnstructured unmarshals a Tenant object from dynamic client's unstructured
func TenantFromUnstructured(r *unstructured.Unstructured) (*v1.Tenant, error) {
b, err := json.Marshal(r.Object)
if err != nil {
return nil, err
}
var p v1.Tenant
if err := json.Unmarshal(b, &p); err != nil {
return nil, err
}
p.TypeMeta.Kind = CRDTenantsKind
p.TypeMeta.APIVersion = CRDGroup + "/" + CRDVersion
return &p, nil
}
// UnstructuredFromTenant marshals a Tenant object into dynamic client's unstructured
func UnstructuredFromTenant(p *v1.Tenant) (*unstructured.Unstructured, error) {
p.TypeMeta.Kind = CRDTenantsKind
p.TypeMeta.APIVersion = CRDGroup + "/" + CRDVersion
b, err := json.Marshal(p)
if err != nil {
return nil, err
}
var r unstructured.Unstructured
if err := json.Unmarshal(b, &r.Object); err != nil {
return nil, err
}
return &r, nil
}
type tenantDecoder struct {
dec *json.Decoder
close func() error
}
func (d *tenantDecoder) Close() {
d.close()
}
func (d *tenantDecoder) Decode() (action watch.EventType, object runtime.Object, err error) {
var e struct {
Type watch.EventType
Object v1.Tenant
}
if err := d.dec.Decode(&e); err != nil {
return watch.Error, nil, err
}
return e.Type, &e.Object, nil
func WaitForTenantInstanceProcessed(kubeClient *rest.RESTClient, name string) error {
return wait.Poll(100*time.Millisecond, 10*time.Second, func() (bool, error) {
var tenant crv1.Tenant
err := kubeClient.Get().
Resource(crv1.TenantResourcePlural).
Namespace(apiv1.NamespaceDefault).
Name(name).
Do().Into(&tenant)
if err == nil && tenant.Status.State == crv1.TenantActive {
return true, nil
}
return false, err
})
}

View File

@ -5,7 +5,6 @@ import (
"time"
"git.openstack.org/openstack/stackube/pkg/auth-controller/rbacmanager/rbac"
"git.openstack.org/openstack/stackube/pkg/auth-controller/tenant"
"git.openstack.org/openstack/stackube/pkg/util"
"github.com/golang/glog"
@ -29,8 +28,8 @@ type Controller struct {
}
// New creates a new RBAC controller.
func New(conf tenant.Config) (*Controller, error) {
cfg, err := util.NewClusterConfig(conf.KubeConfig)
func New(kubeconfig string) (*Controller, error) {
cfg, err := util.NewClusterConfig(kubeconfig)
if err != nil {
return nil, fmt.Errorf("init cluster config failed: %v", err)
}
@ -62,29 +61,8 @@ func New(conf tenant.Config) (*Controller, error) {
func (c *Controller) Run(stopc <-chan struct{}) error {
defer c.queue.ShutDown()
errChan := make(chan error)
go func() {
v, err := c.kclient.Discovery().ServerVersion()
if err != nil {
errChan <- fmt.Errorf("communicating with server failed: %v", err)
return
}
glog.V(4).Infof("Established connection established, cluster-version: %s", v)
errChan <- nil
}()
select {
case err := <-errChan:
if err != nil {
return err
}
glog.V(4).Info("CRD API endpoints ready")
case <-stopc:
return nil
}
glog.V(4).Info("Starting rbac manager")
go c.worker()
go c.nsInf.Run(stopc)
<-stopc

View File

@ -1,370 +0,0 @@
package tenant
import (
"fmt"
"reflect"
"strings"
"time"
"git.openstack.org/openstack/stackube/pkg/apis/v1"
crv1 "git.openstack.org/openstack/stackube/pkg/apis/v1"
"git.openstack.org/openstack/stackube/pkg/auth-controller/client/auth"
"git.openstack.org/openstack/stackube/pkg/auth-controller/rbacmanager/rbac"
"git.openstack.org/openstack/stackube/pkg/openstack"
"git.openstack.org/openstack/stackube/pkg/util"
"github.com/golang/glog"
apiextensionsv1beta1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1beta1"
apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset"
apierrors "k8s.io/apimachinery/pkg/api/errors"
apismetav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/pkg/api"
apiv1 "k8s.io/client-go/pkg/api/v1"
"k8s.io/client-go/tools/cache"
"k8s.io/client-go/util/workqueue"
)
var (
// NOTE: we should always use crv1.TenantResourcePlural.CRDGroup as CRD name
crdTenant = crv1.TenantResourcePlural + "." + auth.CRDGroup
resyncPeriod = 5 * time.Minute
)
// TenantController manages lify cycle of Tenant.
type TenantController struct {
kclient *kubernetes.Clientset
crdclient *apiextensionsclient.Clientset
tclient *auth.AuthClient
osclient *openstack.Client
tenInf cache.SharedIndexInformer
queue workqueue.RateLimitingInterface
config Config
}
// Config defines configuration parameters for the TenantController.
type Config struct {
KubeConfig string
CloudConfig string
}
// New creates a new tenant controller.
func New(conf Config) (*TenantController, error) {
cfg, err := util.NewClusterConfig(conf.KubeConfig)
if err != nil {
return nil, fmt.Errorf("init cluster config failed: %v", err)
}
client, err := kubernetes.NewForConfig(cfg)
if err != nil {
return nil, fmt.Errorf("init kubernetes client failed: %v", err)
}
tclient, err := auth.NewForConfig(cfg)
if err != nil {
return nil, fmt.Errorf("init restclient for tenant failed: %v", err)
}
crdclient, err := apiextensionsclient.NewForConfig(cfg)
if err != nil {
return nil, fmt.Errorf("init CRD client failed: %v", err)
}
openStackClient, err := openstack.NewClient(conf.CloudConfig)
if err != nil {
return nil, fmt.Errorf("init openstack client failed: %v", err)
}
c := &TenantController{
crdclient: crdclient,
kclient: client,
tclient: tclient,
osclient: openStackClient,
queue: workqueue.NewNamedRateLimitingQueue(workqueue.DefaultControllerRateLimiter(), "tenant"),
config: conf,
}
c.tenInf = cache.NewSharedIndexInformer(
&cache.ListWatch{
ListFunc: tclient.Tenants(api.NamespaceAll).List,
WatchFunc: tclient.Tenants(api.NamespaceAll).Watch,
},
&v1.Tenant{}, resyncPeriod, cache.Indexers{},
)
c.tenInf.AddEventHandler(cache.ResourceEventHandlerFuncs{
AddFunc: c.handleAddTenant,
DeleteFunc: c.handleDeleteTenant,
UpdateFunc: c.handleUpdateTenant,
})
return c, nil
}
// Run the controller.
func (c *TenantController) Run(stopc <-chan struct{}) error {
defer c.queue.ShutDown()
errChan := make(chan error)
go func() {
v, err := c.kclient.Discovery().ServerVersion()
if err != nil {
errChan <- fmt.Errorf("communicating with server failed: %v", err)
return
}
glog.V(4).Infof("Established connection established, cluster-version: %s", v)
// Create CRD
if _, err := c.createTenantCRD(c.crdclient); err != nil {
if err != nil && !apierrors.IsAlreadyExists(err) {
errChan <- fmt.Errorf("creating tenant CRD failed: %v", err)
}
return
}
// Create clusterRole
if err = c.createClusterRoles(); err != nil {
errChan <- fmt.Errorf("creating clusterrole failed: %v", err)
return
}
errChan <- nil
}()
select {
case err := <-errChan:
if err != nil {
return err
}
glog.V(4).Info("CRD API endpoints ready")
case <-stopc:
return nil
}
go c.worker()
go c.tenInf.Run(stopc)
<-stopc
return nil
}
func (c *TenantController) keyFunc(obj interface{}) (string, bool) {
k, err := cache.DeletionHandlingMetaNamespaceKeyFunc(obj)
if err != nil {
glog.V(4).Infof("Failed create key: %v", err)
return k, false
}
return k, true
}
func (c *TenantController) handleAddTenant(obj interface{}) {
key, ok := c.keyFunc(obj)
if !ok {
return
}
glog.V(4).Infof("Added tenant %s", key)
c.enqueue(key)
}
func (c *TenantController) handleDeleteTenant(obj interface{}) {
key, ok := c.keyFunc(obj)
if !ok {
return
}
glog.V(4).Infof("Deleted tenant %s", key)
c.enqueue(key)
}
func (c *TenantController) handleUpdateTenant(old, cur interface{}) {
key, ok := c.keyFunc(cur)
if !ok {
return
}
glog.V(4).Infof("Updated tenant %s", key)
c.enqueue(key)
}
// enqueue adds a key to the queue. If obj is a key already it gets added directly.
// Otherwise, the key is extracted via keyFunc.
func (c *TenantController) enqueue(obj interface{}) {
if obj == nil {
return
}
key, ok := obj.(string)
if !ok {
key, ok = c.keyFunc(obj)
if !ok {
return
}
}
c.queue.Add(key)
}
// worker runs a worker thread that just dequeues items, processes them, and marks them done.
// It enforces that the syncHandler is never invoked concurrently with the same key.
func (c *TenantController) worker() {
for c.processNextWorkItem() {
}
}
func (c *TenantController) processNextWorkItem() bool {
key, quit := c.queue.Get()
if quit {
return false
}
defer c.queue.Done(key)
err := c.sync(key.(string))
if err == nil {
c.queue.Forget(key)
return true
}
utilruntime.HandleError(fmt.Errorf("Sync %q failed: %v", key, err))
c.queue.AddRateLimited(key)
return true
}
func (c *TenantController) sync(key string) error {
obj, exists, err := c.tenInf.GetIndexer().GetByKey(key)
if err != nil {
return err
}
if !exists {
// Delete tenant related resources in k8s
tenant := strings.Split(key, "/")
deleteOptions := &apismetav1.DeleteOptions{
TypeMeta: apismetav1.TypeMeta{
Kind: "ClusterRoleBinding",
APIVersion: "rbac.authorization.k8s.io/v1beta1",
},
}
err = c.kclient.Rbac().ClusterRoleBindings().Delete(tenant[1]+"-namespace-creater", deleteOptions)
if err != nil && !apierrors.IsNotFound(err) {
glog.Errorf("Failed delete ClusterRoleBinding for tenant %s: %v", tenant[1], err)
return err
}
glog.V(4).Infof("Deleted ClusterRoleBinding %s", tenant[1])
//Delete namespace
err = c.deleteNamespace(tenant[1])
if err != nil {
return err
}
glog.V(4).Infof("Deleted namespace %s", tenant[1])
// Delete all users on a tenant
err = c.osclient.DeleteAllUsersOnTenant(tenant[1])
if err != nil {
glog.Errorf("Failed delete all users in the tenant %s: %v", tenant[1], err)
return err
}
// Delete tenant in keystone
err = c.osclient.DeleteTenant(tenant[1])
if err != nil {
glog.Errorf("Failed delete tenant %s: %v", tenant[1], err)
return err
}
return nil
}
t := obj.(*v1.Tenant)
glog.V(4).Infof("Sync tenant %s", key)
err = c.syncTenant(t)
if err != nil {
return err
}
return nil
}
func (c *TenantController) createTenantCRD(clientset apiextensionsclient.Interface) (*apiextensionsv1beta1.CustomResourceDefinition, error) {
crd := &apiextensionsv1beta1.CustomResourceDefinition{
ObjectMeta: apismetav1.ObjectMeta{
Name: crdTenant,
},
Spec: apiextensionsv1beta1.CustomResourceDefinitionSpec{
Group: crv1.GroupName,
Version: crv1.SchemeGroupVersion.Version,
Scope: apiextensionsv1beta1.NamespaceScoped,
Names: apiextensionsv1beta1.CustomResourceDefinitionNames{
Plural: crv1.TenantResourcePlural,
Kind: reflect.TypeOf(crv1.Tenant{}).Name(),
},
},
}
_, err := clientset.ApiextensionsV1beta1().CustomResourceDefinitions().Create(crd)
if err != nil {
return nil, err
}
// wait for CRD being established
if err = util.WaitForCRDReady(clientset, crdTenant); err != nil {
return nil, err
} else {
return crd, nil
}
}
func (c *TenantController) syncTenant(tenant *v1.Tenant) error {
roleBinding := rbac.GenerateClusterRoleBindingByTenant(tenant.Name)
_, err := c.kclient.Rbac().ClusterRoleBindings().Create(roleBinding)
if err != nil && !apierrors.IsAlreadyExists(err) {
glog.Errorf("Failed create ClusterRoleBinding for tenant %s: %v", tenant.Name, err)
return err
}
glog.V(4).Infof("Created ClusterRoleBindings %s-namespace-creater for tenant %s", tenant.Name, tenant.Name)
if tenant.Spec.TenantID != "" {
// Create user with the spec username and password in the given tenant
err = c.osclient.CreateUser(tenant.Spec.UserName, tenant.Spec.Password, tenant.Spec.TenantID)
if err != nil && !openstack.IsAlreadyExists(err) {
glog.Errorf("Failed create user %s: %v", tenant.Spec.UserName, err)
return err
}
} else {
// Create tenant if the tenant not exist in keystone
tenantID, err := c.osclient.CreateTenant(tenant.Name)
if err != nil {
return err
}
// Create user with the spec username and password in the created tenant
err = c.osclient.CreateUser(tenant.Spec.UserName, tenant.Spec.Password, tenantID)
if err != nil {
return err
}
}
// Create namespace which name is the same as the tenant's name
err = c.createNamespace(tenant.Name)
if err != nil {
return err
}
glog.V(4).Infof("Created namespace %s for tenant %s", tenant.Name, tenant.Name)
return nil
}
func (c *TenantController) createClusterRoles() error {
nsCreater := rbac.GenerateClusterRole()
_, err := c.kclient.Rbac().ClusterRoles().Create(nsCreater)
if err != nil && !apierrors.IsAlreadyExists(err) {
glog.Errorf("Failed create ClusterRoles namespace-creater: %v", err)
return err
}
glog.V(4).Info("Created ClusterRoles namespace-creater")
return nil
}
func (c *TenantController) createNamespace(namespace string) error {
_, err := c.kclient.CoreV1().Namespaces().Create(&apiv1.Namespace{
ObjectMeta: apismetav1.ObjectMeta{
Name: namespace,
},
})
if err != nil && !apierrors.IsAlreadyExists(err) {
glog.Errorf("Failed create namespace %s: %v", namespace, err)
return err
}
return nil
}
func (c *TenantController) deleteNamespace(namespace string) error {
err := c.kclient.CoreV1().Namespaces().Delete(namespace, apismetav1.NewDeleteOptions(0))
if err != nil {
glog.Errorf("Failed delete namespace %s: %v", namespace, err)
return err
}
return nil
}

View File

@ -0,0 +1,174 @@
package tenant
import (
"fmt"
crv1 "git.openstack.org/openstack/stackube/pkg/apis/v1"
crdClient "git.openstack.org/openstack/stackube/pkg/auth-controller/client/auth"
"git.openstack.org/openstack/stackube/pkg/openstack"
"github.com/golang/glog"
apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset"
apierrors "k8s.io/apimachinery/pkg/api/errors"
apismetav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/fields"
"k8s.io/apimachinery/pkg/runtime"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
"k8s.io/client-go/kubernetes"
apiv1 "k8s.io/client-go/pkg/api/v1"
"k8s.io/client-go/rest"
"k8s.io/client-go/tools/cache"
"k8s.io/client-go/tools/clientcmd"
)
// TenantController manages lify cycle of Tenant.
type TenantController struct {
k8sClient *kubernetes.Clientset
tenantClient *rest.RESTClient
tenantScheme *runtime.Scheme
openstackClient *openstack.Client
}
// New creates a new tenant controller.
func New(kubeconfig, cloudconfig string) (*TenantController, error) {
// Create OpenStack client from config
openStackClient, err := openstack.NewClient(cloudconfig)
if err != nil {
return nil, fmt.Errorf("init openstack client failed: %v", err)
}
// Create the client config. Use kubeconfig if given, otherwise assume in-cluster.
config, err := buildConfig(kubeconfig)
if err != nil {
return nil, fmt.Errorf("failed to build kubeconfig: %v", err)
}
clientset, err := apiextensionsclient.NewForConfig(config)
if err != nil {
return nil, fmt.Errorf("failed to create kubeclient from config: %v", err)
}
// initialize CRD if it does not exist
_, err = crdClient.CreateTenantCRD(clientset)
if err != nil && !apierrors.IsAlreadyExists(err) {
return nil, fmt.Errorf("failed to create CRD to kube-apiserver: %v", err)
}
k8sClient, err := kubernetes.NewForConfig(config)
if err != nil {
return nil, fmt.Errorf("failed to create kubernetes client: %v", err)
}
// make a new config for our extension's API group, using the first config as a baseline
tenantClient, tenantScheme, err := crdClient.NewClient(config)
if err != nil {
return nil, fmt.Errorf("failed to create client for CRD: %v", err)
}
c := &TenantController{
tenantClient: tenantClient,
tenantScheme: tenantScheme,
k8sClient: k8sClient,
openstackClient: openStackClient,
}
if err = c.createClusterRoles(); err != nil {
return nil, fmt.Errorf("failed to create cluster roles to kube-apiserver: %v", err)
}
return c, nil
}
func buildConfig(kubeconfig string) (*rest.Config, error) {
if kubeconfig != "" {
return clientcmd.BuildConfigFromFlags("", kubeconfig)
}
return rest.InClusterConfig()
}
// Run the controller.
func (c *TenantController) Run(stopCh <-chan struct{}) error {
defer utilruntime.HandleCrash()
source := cache.NewListWatchFromClient(
c.tenantClient,
crv1.TenantResourcePlural,
apiv1.NamespaceAll,
fields.Everything())
_, tenantInformor := cache.NewInformer(
source,
&crv1.Tenant{},
0,
cache.ResourceEventHandlerFuncs{
AddFunc: c.onAdd,
UpdateFunc: c.onUpdate,
DeleteFunc: c.onDelete,
})
go tenantInformor.Run(stopCh)
<-stopCh
return nil
}
func (c *TenantController) onAdd(obj interface{}) {
tenant := obj.(*crv1.Tenant)
glog.V(3).Infof("Tenant controller received new object %q\n", tenant)
copyObj, err := c.tenantScheme.Copy(tenant)
if err != nil {
glog.Errorf("ERROR creating a deep copy of tenant object: %v\n", err)
return
}
newTenant := copyObj.(*crv1.Tenant)
c.syncTenant(newTenant)
}
func (c *TenantController) onUpdate(obj1, obj2 interface{}) {
glog.Warning("tenant updates is not supported yet.")
}
func (c *TenantController) onDelete(obj interface{}) {
tenant, ok := obj.(*crv1.Tenant)
if !ok {
return
}
glog.V(3).Infof("Tenant controller received deleted tenant %q\n", tenant)
deleteOptions := &apismetav1.DeleteOptions{
TypeMeta: apismetav1.TypeMeta{
Kind: "ClusterRoleBinding",
APIVersion: "rbac.authorization.k8s.io/v1beta1",
},
}
tenantName := tenant.Name
err := c.k8sClient.Rbac().ClusterRoleBindings().Delete(tenantName+"-namespace-creater", deleteOptions)
if err != nil && !apierrors.IsNotFound(err) {
glog.Errorf("Failed delete ClusterRoleBinding for tenant %s: %v", tenantName, err)
} else {
glog.V(4).Infof("Deleted ClusterRoleBinding %s", tenantName)
}
//Delete namespace
err = c.deleteNamespace(tenantName)
if err != nil {
glog.Errorf("Delete namespace %s failed: %v", tenantName, err)
} else {
glog.V(4).Infof("Deleted namespace %s", tenantName)
}
// Delete all users on a tenant
err = c.openstackClient.DeleteAllUsersOnTenant(tenantName)
if err != nil {
glog.Errorf("Failed delete all users in the tenant %s: %v", tenantName, err)
}
// Delete tenant in keystone
if tenant.Spec.TenantID == "" {
err = c.openstackClient.DeleteTenant(tenantName)
if err != nil {
glog.Errorf("Failed delete tenant %s: %v", tenantName, err)
}
}
}

View File

@ -0,0 +1,83 @@
package tenant
import (
crv1 "git.openstack.org/openstack/stackube/pkg/apis/v1"
"git.openstack.org/openstack/stackube/pkg/auth-controller/rbacmanager/rbac"
"git.openstack.org/openstack/stackube/pkg/openstack"
"github.com/golang/glog"
apierrors "k8s.io/apimachinery/pkg/api/errors"
apismetav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
apiv1 "k8s.io/client-go/pkg/api/v1"
)
func (c *TenantController) syncTenant(tenant *crv1.Tenant) error {
roleBinding := rbac.GenerateClusterRoleBindingByTenant(tenant.Name)
_, err := c.k8sClient.Rbac().ClusterRoleBindings().Create(roleBinding)
if err != nil && !apierrors.IsAlreadyExists(err) {
glog.Errorf("Failed create ClusterRoleBinding for tenant %s: %v", tenant.Name, err)
return err
}
glog.V(4).Infof("Created ClusterRoleBindings %s-namespace-creater for tenant %s", tenant.Name, tenant.Name)
if tenant.Spec.TenantID != "" {
// Create user with the spec username and password in the given tenant
err = c.openstackClient.CreateUser(tenant.Spec.UserName, tenant.Spec.Password, tenant.Spec.TenantID)
if err != nil && !openstack.IsAlreadyExists(err) {
glog.Errorf("Failed create user %s: %v", tenant.Spec.UserName, err)
return err
}
} else {
// Create tenant if the tenant not exist in keystone
tenantID, err := c.openstackClient.CreateTenant(tenant.Name)
if err != nil {
return err
}
// Create user with the spec username and password in the created tenant
err = c.openstackClient.CreateUser(tenant.Spec.UserName, tenant.Spec.Password, tenantID)
if err != nil {
return err
}
}
// Create namespace which name is the same as the tenant's name
err = c.createNamespace(tenant.Name)
if err != nil {
return err
}
glog.V(4).Infof("Created namespace %s for tenant %s", tenant.Name, tenant.Name)
return nil
}
func (c *TenantController) createClusterRoles() error {
nsCreater := rbac.GenerateClusterRole()
_, err := c.k8sClient.Rbac().ClusterRoles().Create(nsCreater)
if err != nil && !apierrors.IsAlreadyExists(err) {
glog.Errorf("Failed create ClusterRoles namespace-creater: %v", err)
return err
}
glog.V(4).Info("Created ClusterRoles namespace-creater")
return nil
}
func (c *TenantController) createNamespace(namespace string) error {
_, err := c.k8sClient.CoreV1().Namespaces().Create(&apiv1.Namespace{
ObjectMeta: apismetav1.ObjectMeta{
Name: namespace,
},
})
if err != nil && !apierrors.IsAlreadyExists(err) {
glog.Errorf("Failed create namespace %s: %v", namespace, err)
return err
}
return nil
}
func (c *TenantController) deleteNamespace(namespace string) error {
err := c.k8sClient.CoreV1().Namespaces().Delete(namespace, apismetav1.NewDeleteOptions(0))
if err != nil {
glog.Errorf("Failed delete namespace %s: %v", namespace, err)
return err
}
return nil
}