Files
kubevela/pkg/controller/setup.go
wyike 4eb33e9239 split out oam-spec v0.2 charts (#1842)
* WIP disable oam related controller and webhok

change velue

fix spell error

change namespace

add e2e for oam-runtime

fix charts hack

fix e2e definition namespace

fix diasble contains list

fix e2e-test

* disable components handler

* fix flaky ac test

add every definition in oam-runtime-system namespace

* upload ac e2e-test

replace files

fix upload reports

* more wait time
2021-07-01 16:32:01 +08:00

67 lines
2.3 KiB
Go

/*
Copyright 2019 The Crossplane 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 controller
import (
ctrl "sigs.k8s.io/controller-runtime"
controller "github.com/oam-dev/kubevela/pkg/controller/core.oam.dev"
"github.com/oam-dev/kubevela/pkg/controller/core.oam.dev/v1alpha2/core/scopes/healthscope"
"github.com/oam-dev/kubevela/pkg/controller/core.oam.dev/v1alpha2/core/traits/manualscalertrait"
"github.com/oam-dev/kubevela/pkg/controller/core.oam.dev/v1alpha2/core/workloads/containerizedworkload"
"github.com/oam-dev/kubevela/pkg/controller/common"
"github.com/oam-dev/kubevela/pkg/controller/standard.oam.dev/v1alpha1/podspecworkload"
"github.com/oam-dev/kubevela/pkg/controller/utils"
)
// Setup workload controllers.
func Setup(mgr ctrl.Manager, disableCaps string, args controller.Args) error {
var functions []func(ctrl.Manager, controller.Args) error
switch disableCaps {
case common.DisableNoneCaps:
functions = []func(ctrl.Manager, controller.Args) error{
podspecworkload.Setup,
manualscalertrait.Setup,
containerizedworkload.Setup,
healthscope.Setup,
}
case common.DisableAllCaps:
default:
disableCapsSet := utils.StoreInSet(disableCaps)
if !disableCapsSet.Contains(common.PodspecWorkloadControllerName) {
functions = append(functions, podspecworkload.Setup)
}
if !disableCapsSet.Contains(common.ManualScalerTraitControllerName) {
functions = append(functions, manualscalertrait.Setup)
}
if !disableCapsSet.Contains(common.ContainerizedWorkloadControllerName) {
functions = append(functions, containerizedworkload.Setup)
}
if !disableCapsSet.Contains(common.HealthScopeControllerName) {
functions = append(functions, healthscope.Setup)
}
}
for _, setup := range functions {
if err := setup(mgr, args); err != nil {
return err
}
}
return nil
}