feat: kubesphere 4.0 (#6115)
* feat: kubesphere 4.0 Signed-off-by: ci-bot <ci-bot@kubesphere.io> * feat: kubesphere 4.0 Signed-off-by: ci-bot <ci-bot@kubesphere.io> --------- Signed-off-by: ci-bot <ci-bot@kubesphere.io> Co-authored-by: ks-ci-bot <ks-ci-bot@example.com> Co-authored-by: joyceliu <joyceliu@yunify.com>
This commit is contained in:
committed by
GitHub
parent
b5015ec7b9
commit
447a51f08b
104
pkg/controller/role/role_controller.go
Normal file
104
pkg/controller/role/role_controller.go
Normal file
@@ -0,0 +1,104 @@
|
||||
/*
|
||||
* Please refer to the LICENSE file in the root directory of the project.
|
||||
* https://github.com/kubesphere/kubesphere/blob/master/LICENSE
|
||||
*/
|
||||
|
||||
package role
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
|
||||
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
|
||||
rbacutils "kubesphere.io/kubesphere/pkg/utils/rbac"
|
||||
|
||||
rbacv1 "k8s.io/api/rbac/v1"
|
||||
|
||||
kscontroller "kubesphere.io/kubesphere/pkg/controller"
|
||||
|
||||
"github.com/go-logr/logr"
|
||||
"k8s.io/client-go/tools/record"
|
||||
iamv1beta1 "kubesphere.io/api/iam/v1beta1"
|
||||
ctrl "sigs.k8s.io/controller-runtime"
|
||||
"sigs.k8s.io/controller-runtime/pkg/client"
|
||||
|
||||
rbachelper "kubesphere.io/kubesphere/pkg/componenthelper/auth/rbac"
|
||||
)
|
||||
|
||||
const (
|
||||
controllerName = "role"
|
||||
roleRef = "iam.kubesphere.io/role-ref"
|
||||
)
|
||||
|
||||
var _ kscontroller.Controller = &Reconciler{}
|
||||
|
||||
type Reconciler struct {
|
||||
client.Client
|
||||
logger logr.Logger
|
||||
recorder record.EventRecorder
|
||||
helper *rbachelper.Helper
|
||||
}
|
||||
|
||||
func (r *Reconciler) Name() string {
|
||||
return controllerName
|
||||
}
|
||||
|
||||
func (r *Reconciler) SetupWithManager(mgr *kscontroller.Manager) error {
|
||||
r.Client = mgr.GetClient()
|
||||
r.logger = ctrl.Log.WithName("controllers").WithName(controllerName)
|
||||
r.recorder = mgr.GetEventRecorderFor(controllerName)
|
||||
r.helper = rbachelper.NewHelper(r.Client)
|
||||
return ctrl.NewControllerManagedBy(mgr).
|
||||
Named(controllerName).
|
||||
For(&iamv1beta1.Role{}).
|
||||
Complete(r)
|
||||
}
|
||||
|
||||
func (r *Reconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
|
||||
log := r.logger.WithValues("Role", req.String())
|
||||
role := &iamv1beta1.Role{}
|
||||
if err := r.Get(ctx, req.NamespacedName, role); err != nil {
|
||||
return ctrl.Result{}, client.IgnoreNotFound(err)
|
||||
}
|
||||
|
||||
if role.AggregationRoleTemplates != nil {
|
||||
if err := r.helper.AggregationRole(ctx, rbachelper.RoleRuleOwner{Role: role}, r.recorder); err != nil {
|
||||
return ctrl.Result{}, err
|
||||
}
|
||||
}
|
||||
|
||||
if err := r.syncToKubernetes(ctx, role); err != nil {
|
||||
log.Error(err, "sync role failed")
|
||||
return ctrl.Result{}, err
|
||||
}
|
||||
|
||||
return ctrl.Result{}, nil
|
||||
}
|
||||
|
||||
func (r *Reconciler) syncToKubernetes(ctx context.Context, role *iamv1beta1.Role) error {
|
||||
k8sRole := &rbacv1.Role{
|
||||
ObjectMeta: metav1.ObjectMeta{Namespace: role.Namespace, Name: rbacutils.RelatedK8sResourceName(role.Name)},
|
||||
}
|
||||
|
||||
op, err := controllerutil.CreateOrUpdate(ctx, r.Client, k8sRole, func() error {
|
||||
if k8sRole.Labels == nil {
|
||||
k8sRole.Labels = make(map[string]string)
|
||||
}
|
||||
k8sRole.Labels[roleRef] = role.Name
|
||||
k8sRole.Rules = role.Rules
|
||||
if err := controllerutil.SetOwnerReference(role, k8sRole, r.Scheme()); err != nil {
|
||||
return fmt.Errorf("failed to set owner reference: %s", err)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
r.logger.Error(err, "sync role failed", "role", role.Name)
|
||||
}
|
||||
|
||||
r.logger.V(4).Info("sync role to K8s", "role", role.Name, "op", op)
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user