mirror of https://github.com/k3s-io/k3s
added algorithm provider cli option to the kube-scheduler command and renamed default provider to follow naming convention
parent
e12a78cb1d
commit
e3da5e15bc
|
@ -37,9 +37,10 @@ import (
|
|||
)
|
||||
|
||||
var (
|
||||
port = flag.Int("port", ports.SchedulerPort, "The port that the scheduler's http service runs on")
|
||||
address = util.IP(net.ParseIP("127.0.0.1"))
|
||||
clientConfig = &client.Config{}
|
||||
port = flag.Int("port", ports.SchedulerPort, "The port that the scheduler's http service runs on")
|
||||
address = util.IP(net.ParseIP("127.0.0.1"))
|
||||
clientConfig = &client.Config{}
|
||||
algorithmProvider = flag.String("algorithm_provider", factory.DefaultProvider, "The scheduling algorithm provider to use")
|
||||
)
|
||||
|
||||
func init() {
|
||||
|
@ -64,7 +65,7 @@ func main() {
|
|||
go http.ListenAndServe(net.JoinHostPort(address.String(), strconv.Itoa(*port)), nil)
|
||||
|
||||
configFactory := factory.NewConfigFactory(kubeClient)
|
||||
config, err := configFactory.Create()
|
||||
config, err := createConfig(configFactory)
|
||||
if err != nil {
|
||||
glog.Fatalf("Failed to create scheduler configuration: %v", err)
|
||||
}
|
||||
|
@ -73,3 +74,12 @@ func main() {
|
|||
|
||||
select {}
|
||||
}
|
||||
|
||||
func createConfig(configFactory *factory.ConfigFactory) (*scheduler.Config, error) {
|
||||
// check of algorithm provider is registered and fail fast
|
||||
_, err := factory.GetAlgorithmProvider(*algorithmProvider)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return configFactory.CreateFromProvider(*algorithmProvider)
|
||||
}
|
||||
|
|
|
@ -36,7 +36,7 @@ var (
|
|||
)
|
||||
|
||||
const (
|
||||
DefaultProvider = "default"
|
||||
DefaultProvider = "DefaultProvider"
|
||||
)
|
||||
|
||||
type AlgorithmProviderConfig struct {
|
||||
|
|
Loading…
Reference in New Issue