114 lines
4.5 KiB
Go
114 lines
4.5 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes 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.
|
|
*/
|
|
|
|
// The external controller manager is responsible for running controller loops that
|
|
// are cloud provider dependent. It uses the API to listen to new events on resources.
|
|
|
|
// This file should be written by each cloud provider.
|
|
// For an minimal working example, please refer to k8s.io/cloud-provider/sample/basic_main.go
|
|
// For more details, please refer to k8s.io/kubernetes/cmd/cloud-controller-manager/main.go
|
|
// The current file demonstrate how other cloud provider should leverage CCM and it uses fake parameters. Please modify for your own use.
|
|
|
|
package main
|
|
|
|
import (
|
|
"math/rand"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/spf13/pflag"
|
|
|
|
"k8s.io/cloud-provider"
|
|
"k8s.io/cloud-provider/app"
|
|
cloudcontrollerconfig "k8s.io/cloud-provider/app/config"
|
|
"k8s.io/cloud-provider/options"
|
|
"k8s.io/component-base/cli/flag"
|
|
"k8s.io/component-base/logs"
|
|
_ "k8s.io/component-base/metrics/prometheus/clientgo" // load all the prometheus client-go plugins
|
|
_ "k8s.io/component-base/metrics/prometheus/version" // for version metric registration
|
|
"k8s.io/klog/v2"
|
|
// For existing cloud providers, the option to import legacy providers is still available.
|
|
// e.g. _"k8s.io/legacy-cloud-providers/<provider>"
|
|
)
|
|
|
|
const (
|
|
// cloudProviderName shows an sample of using hard coded parameter, please edit the value for your case.
|
|
cloudProviderName = "SampleCloudProviderName"
|
|
)
|
|
|
|
func main() {
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
ccmOptions, err := options.NewCloudControllerManagerOptions()
|
|
if err != nil {
|
|
klog.Fatalf("unable to initialize command options: %v", err)
|
|
}
|
|
|
|
controllerInitializers := app.DefaultInitFuncConstructors
|
|
// Here is an example to remove the controller which is not needed.
|
|
// e.g. remove the cloud-node-lifecycle controller which current cloud provider does not need.
|
|
//delete(controllerInitializers, "cloud-node-lifecycle")
|
|
|
|
// Here is an example to add an controller(NodeIpamController) which will be used by cloud provider
|
|
// generate nodeIPAMConfig. Here is an sample code. Please pass the right parameter in your code.
|
|
// If you do not need additional controller, please ignore.
|
|
controllerInitializers["nodeipam"] = startNodeIpamControllerWrapper
|
|
|
|
command := app.NewCloudControllerManagerCommand(ccmOptions, cloudInitializer, controllerInitializers)
|
|
|
|
// TODO: once we switch everything over to Cobra commands, we can go back to calling
|
|
// utilflag.InitFlags() (by removing its pflag.Parse() call). For now, we have to set the
|
|
// normalize func and add the go flag set by hand.
|
|
// Here is an sample
|
|
pflag.CommandLine.SetNormalizeFunc(flag.WordSepNormalizeFunc)
|
|
// utilflag.InitFlags()
|
|
logs.InitLogs()
|
|
defer logs.FlushLogs()
|
|
|
|
if err := command.Execute(); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func cloudInitializer(config *cloudcontrollerconfig.CompletedConfig) cloudprovider.Interface {
|
|
cloudConfigFile := config.ComponentConfig.KubeCloudShared.CloudProvider.CloudConfigFile
|
|
// initialize cloud provider with the cloud provider name and config file provided
|
|
cloud, err := cloudprovider.InitCloudProvider(cloudProviderName, cloudConfigFile)
|
|
if err != nil {
|
|
klog.Fatalf("Cloud provider could not be initialized: %v", err)
|
|
}
|
|
if cloud == nil {
|
|
klog.Fatalf("Cloud provider is nil")
|
|
}
|
|
|
|
if !cloud.HasClusterID() {
|
|
if config.ComponentConfig.KubeCloudShared.AllowUntaggedCloud {
|
|
klog.Warning("detected a cluster without a ClusterID. A ClusterID will be required in the future. Please tag your cluster to avoid any future issues")
|
|
} else {
|
|
klog.Fatalf("no ClusterID found. A ClusterID is required for the cloud provider to function properly. This check can be bypassed by setting the allow-untagged-cloud option")
|
|
}
|
|
}
|
|
|
|
// Initialize the cloud provider with a reference to the clientBuilder
|
|
cloud.Initialize(config.ClientBuilder, make(chan struct{}))
|
|
// Set the informer on the user cloud object
|
|
if informerUserCloud, ok := cloud.(cloudprovider.InformerUser); ok {
|
|
informerUserCloud.SetInformers(config.SharedInformers)
|
|
}
|
|
|
|
return cloud
|
|
}
|