48 lines
1.8 KiB
Go
Executable File
48 lines
1.8 KiB
Go
Executable File
package apply
|
|
|
|
// Author: Weisen Pan
|
|
// Date: 2023-10-24
|
|
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
applypkg "github.com/hkust-adsl/kubernetes-scheduler-simulator/knets_pkg/apply"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// options stores the command line options.
|
|
var options = applypkg.Options{}
|
|
|
|
// ApplyCmd represents the 'apply' command.
|
|
var ApplyCmd = &cobra.Command{
|
|
Use: "apply",
|
|
Short: "Make a reasonable cluster capacity planning based on application resource requirements",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// Create a new Applier instance with the specified options.
|
|
applier := applypkg.NewApplier(options)
|
|
|
|
// Run the Applier to perform cluster capacity planning.
|
|
if err := applier.Run(); err != nil {
|
|
fmt.Printf("apply error: %s", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
// Define command-line flags for the 'apply' command.
|
|
ApplyCmd.Flags().StringVarP(&options.SimonConfig, "simon-config", "f", options.SimonConfig, "path to the cluster kube-config file used to connect cluster, one of both kube-config and cluster-config must exist.")
|
|
ApplyCmd.Flags().StringVarP(&options.DefaultSchedulerConfigFile, "default-scheduler-config", "s", options.DefaultSchedulerConfigFile, "path to JSON or YAML file containing scheduler configuration.")
|
|
ApplyCmd.Flags().BoolVar(&options.UseGreed, "use-greed", false, "use greedy algorithm when queue pods")
|
|
ApplyCmd.Flags().BoolVarP(&options.Interactive, "interactive", "i", false, "interactive mode")
|
|
ApplyCmd.Flags().StringSliceVarP(&options.ExtendedResources, "extended-resources", "e", nil, "show extended resources when reporting, e.g. open-local, gpu")
|
|
|
|
// Mark the 'simon-config' flag as required.
|
|
if err := ApplyCmd.MarkFlagRequired("simon-config"); err != nil {
|
|
log.Fatal("failed to init ApplyCmd on simon-config flag")
|
|
}
|
|
}
|