119 lines
3.0 KiB
Go
119 lines
3.0 KiB
Go
//go:generate go run pkg/codegen/cleanup/cleanup.go
|
|
//go:generate rm -rf pkg/generated pkg/crds/yaml/generated
|
|
//go:generate go run pkg/codegen/codegen.go
|
|
//go:generate controller-gen crd:generateEmbeddedObjectMeta=true paths=./pkg/apis/... output:crd:dir=./pkg/crds/yaml/generated
|
|
//go:generate crd-ref-docs --config=crd-ref-docs.yaml --renderer=markdown --output-path=doc/plan.md
|
|
|
|
package main
|
|
|
|
// Copyright 2019 Rancher Labs, Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/rancher/system-upgrade-controller/pkg/upgrade"
|
|
"github.com/rancher/system-upgrade-controller/pkg/version"
|
|
"github.com/rancher/wrangler/v3/pkg/signals"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
)
|
|
|
|
var (
|
|
debug, leaderElect bool
|
|
kubeConfig, masterURL, nodeName string
|
|
namespace, name, serviceAccountName string
|
|
threads int
|
|
)
|
|
|
|
func main() {
|
|
app := cli.NewApp()
|
|
app.Name = version.Program
|
|
app.Usage = "in ur system controllin ur upgradez"
|
|
app.Version = fmt.Sprintf("%s (%s)", version.Version, version.GitCommit)
|
|
app.Flags = []cli.Flag{
|
|
cli.BoolFlag{
|
|
Name: "debug",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_DEBUG",
|
|
Destination: &debug,
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "leader-elect",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_LEADER_ELECT",
|
|
Destination: &leaderElect,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "kubeconfig",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_KUBE_CONFIG",
|
|
//Value: "${HOME}/.kube/config",
|
|
Destination: &kubeConfig,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "master",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_MASTER_URL",
|
|
Destination: &masterURL,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "name",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_NAME",
|
|
Required: true,
|
|
Destination: &name,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "node-name",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_NODE_NAME",
|
|
Required: false,
|
|
Destination: &nodeName,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "namespace",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_NAMESPACE",
|
|
Required: true,
|
|
Destination: &namespace,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "service-account",
|
|
Hidden: true,
|
|
Destination: &serviceAccountName,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "threads",
|
|
EnvVar: "SYSTEM_UPGRADE_CONTROLLER_THREADS",
|
|
Value: 2,
|
|
Destination: &threads,
|
|
},
|
|
}
|
|
app.Action = Run
|
|
|
|
if serviceAccountName != "" {
|
|
logrus.Warn("deprecated flag `service-account` is ignored")
|
|
}
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func Run(_ *cli.Context) {
|
|
if debug {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
logrus.SetReportCaller(true)
|
|
}
|
|
cfg, err := clientcmd.BuildConfigFromFlags(masterURL, kubeConfig)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
ctl, err := upgrade.NewController(cfg, namespace, name, nodeName, leaderElect, 2*time.Hour)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
ctx := signals.SetupSignalContext()
|
|
if err := ctl.Start(ctx, threads); err != nil {
|
|
logrus.Fatalf("Error starting: %v", err)
|
|
}
|
|
<-ctx.Done()
|
|
}
|