Files
kubevela/pkg/controller/setup.go
Zheng Xi Zhou 06585fea92 Support disable installation of builtin capabilities (#687)
* Support disable installation of builtin capabilities

Added vela core bootstrap flag `--enable-caps` to support no
installation for builtin capabiliteis

Feature #658

* revert flag to "disable-cpas"

Co-authored-by: Jianbo Sun <wonderflow.sun@gmail.com>

* Address comments

Co-authored-by: Jianbo Sun <wonderflow.sun@gmail.com>
2020-12-04 10:03:25 +08:00

62 lines
2.0 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"
"github.com/oam-dev/kubevela/pkg/controller/common"
"github.com/oam-dev/kubevela/pkg/controller/standard.oam.dev/v1alpha1/autoscaler"
"github.com/oam-dev/kubevela/pkg/controller/standard.oam.dev/v1alpha1/metrics"
"github.com/oam-dev/kubevela/pkg/controller/standard.oam.dev/v1alpha1/podspecworkload"
"github.com/oam-dev/kubevela/pkg/controller/standard.oam.dev/v1alpha1/routes"
"github.com/oam-dev/kubevela/pkg/controller/utils"
)
// Setup workload controllers.
func Setup(mgr ctrl.Manager, disableCaps string) error {
var functions []func(ctrl.Manager) error
switch disableCaps {
case common.DisableNoneCaps:
functions = []func(ctrl.Manager) error{
metrics.Setup, podspecworkload.Setup, routes.Setup, autoscaler.Setup,
}
case common.DisableAllCaps:
default:
disableCapsSet := utils.StoreInSet(disableCaps)
if !disableCapsSet.Contains(common.MetricsControllerName) {
functions = append(functions, metrics.Setup)
}
if !disableCapsSet.Contains(common.PodspecWorkloadControllerName) {
functions = append(functions, podspecworkload.Setup)
}
if !disableCapsSet.Contains(common.RouteControllerName) {
functions = append(functions, routes.Setup)
}
if !disableCapsSet.Contains(common.AutoscaleControllerName) {
functions = append(functions, autoscaler.Setup)
}
}
for _, setup := range functions {
if err := setup(mgr); err != nil {
return err
}
}
return nil
}