72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
"k8s.io/apimachinery/pkg/util/errors"
|
|
"k8s.io/apimachinery/pkg/util/runtime"
|
|
cliflag "k8s.io/component-base/cli/flag"
|
|
"k8s.io/component-base/term"
|
|
"k8s.io/klog/v2"
|
|
"sigs.k8s.io/controller-runtime/pkg/manager/signals"
|
|
|
|
"code.gitea.io/gitea/modules/k8s/cmd/controller-manager/app/options"
|
|
"code.gitea.io/gitea/modules/k8s/controller"
|
|
"code.gitea.io/gitea/modules/k8s/controller/application"
|
|
"code.gitea.io/gitea/modules/k8s/controller/devcontainer"
|
|
)
|
|
|
|
func init() {
|
|
// 在初始化时注册所有控制器
|
|
runtime.Must(controller.Register(&devcontainer.Controller{}))
|
|
runtime.Must(controller.Register(&application.Controller{}))
|
|
}
|
|
|
|
// NewControllerManagerCommand 创建一个启动 controller manager 的命令
|
|
func NewControllerManagerCommand() *cobra.Command {
|
|
s := options.NewControllerManagerOptions()
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "controller-manager",
|
|
Long: `DevStar controller manager is a daemon that embeds the control loops shipped with DevStar.`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if errs := s.Validate(); len(errs) != 0 {
|
|
return errors.NewAggregate(errs)
|
|
}
|
|
|
|
return Run(signals.SetupSignalHandler(), s)
|
|
},
|
|
}
|
|
|
|
fs := cmd.Flags()
|
|
namedFlagSets := s.Flags()
|
|
for _, f := range namedFlagSets.FlagSets {
|
|
fs.AddFlagSet(f)
|
|
}
|
|
|
|
usageFmt := "Usage:\n %s\n"
|
|
cols, _, _ := term.TerminalSize(cmd.OutOrStdout())
|
|
cmd.SetUsageFunc(func(cmd *cobra.Command) error {
|
|
_, _ = fmt.Fprintf(cmd.OutOrStderr(), usageFmt, cmd.UseLine())
|
|
cliflag.PrintSections(cmd.OutOrStderr(), namedFlagSets, cols)
|
|
return nil
|
|
})
|
|
|
|
return cmd
|
|
}
|
|
|
|
// Run 启动控制器管理器
|
|
func Run(ctx context.Context, s *options.ControllerManagerOptions) error {
|
|
klog.InfoS("Starting DevStar controller manager")
|
|
|
|
cm, err := s.NewControllerManager()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 启动控制器管理器
|
|
return cm.Start(ctx)
|
|
}
|