|
- /*
- Copyright 2021.
-
- 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 main
-
- import (
- "flag"
- "os"
-
- // Import all Kubernetes client auth plugins (e.g. Azure, GCP, OIDC, etc.)
- // to ensure that exec-entrypoint and run can make use of them.
- _ "k8s.io/client-go/plugin/pkg/client/auth"
-
- "k8s.io/apimachinery/pkg/runtime"
- utilruntime "k8s.io/apimachinery/pkg/util/runtime"
- clientgoscheme "k8s.io/client-go/kubernetes/scheme"
- ctrl "sigs.k8s.io/controller-runtime"
- "sigs.k8s.io/controller-runtime/pkg/healthz"
- "sigs.k8s.io/controller-runtime/pkg/log/zap"
-
- agentv1 "nodeagent/apis/agent/v1"
- agentcontrollers "nodeagent/controllers/agent"
- agentsvc "nodeagent/controllers/agent/service"
- agentconf "nodeagent/controllers/config"
- //+kubebuilder:scaffold:imports
- )
-
- var (
- scheme = runtime.NewScheme()
- setupLog = ctrl.Log.WithName("setup")
- )
-
- func init() {
- utilruntime.Must(clientgoscheme.AddToScheme(scheme))
-
- utilruntime.Must(agentv1.AddToScheme(scheme))
- //+kubebuilder:scaffold:scheme
- }
-
- const (
- NODE_AGENT_NODE_NAME = "NODE_AGENT_NODE_NAME"
- )
-
- func main() {
- var metricsAddr string
- var enableLeaderElection bool = false // force
- var probeAddr string
- // load agent config
- var servConf string
- flag.StringVar(&servConf, "service-conf", "config.yaml", "config path, eg: -conf config.yaml")
- flag.StringVar(&metricsAddr, "metrics-bind-address", ":8080", "The address the metric endpoint binds to.")
- flag.StringVar(&probeAddr, "health-probe-bind-address", ":8081", "The address the probe endpoint binds to.")
- //flag.BoolVar(&enableLeaderElection, "leader-elect", false,
- // "Enable leader election for controller manager. "+
- // "Enabling this will ensure there is only one active controller manager.")
- opts := zap.Options{
- Development: true,
- }
-
- opts.BindFlags(flag.CommandLine)
- flag.Parse()
-
- ctrl.SetLogger(zap.New(zap.UseFlagOptions(&opts)))
-
- agentConfig := agentconf.FromFile(servConf)
- mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{
- Scheme: scheme,
- MetricsBindAddress: metricsAddr,
- Port: 9443,
- HealthProbeBindAddress: probeAddr,
- LeaderElection: enableLeaderElection,
- LeaderElectionID: "f72bd760.octopus.openi.org.cn",
- })
- if err != nil {
- setupLog.Error(err, "unable to start manager")
- os.Exit(1)
- }
-
- nodeName := os.Getenv(NODE_AGENT_NODE_NAME)
- setupLog.Info("Node Agent Controller: ", NODE_AGENT_NODE_NAME, nodeName)
- if err = (&agentcontrollers.NodeActionReconciler{
- NodeName: nodeName,
- Svc: agentsvc.NewActionServiceManager(agentConfig),
-
- Client: mgr.GetClient(),
- Scheme: mgr.GetScheme(),
- }).SetupWithManager(mgr); err != nil {
- setupLog.Error(err, "unable to create controller", "controller", "NodeAction")
- os.Exit(1)
- }
- //+kubebuilder:scaffold:builder
-
- if err := mgr.AddHealthzCheck("healthz", healthz.Ping); err != nil {
- setupLog.Error(err, "unable to set up health check")
- os.Exit(1)
- }
- if err := mgr.AddReadyzCheck("readyz", healthz.Ping); err != nil {
- setupLog.Error(err, "unable to set up ready check")
- os.Exit(1)
- }
-
- setupLog.Info("starting manager")
- if err := mgr.Start(ctrl.SetupSignalHandler()); err != nil {
- setupLog.Error(err, "problem running manager")
- os.Exit(1)
- }
- }
|