-
Notifications
You must be signed in to change notification settings - Fork 327
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: Joseph <joseph.t.lee@outlook.com>
- Loading branch information
Showing
12 changed files
with
1,196 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,4 @@ | ||
FROM gcr.io/distroless/static:latest | ||
WORKDIR / | ||
COPY koord-descheduler . | ||
ENTRYPOINT ["/koord-descheduler"] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,71 @@ | ||
/* | ||
Copyright 2022 The Koordinator Authors. | ||
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 config | ||
|
||
import ( | ||
apiserver "k8s.io/apiserver/pkg/server" | ||
"k8s.io/client-go/dynamic/dynamicinformer" | ||
"k8s.io/client-go/informers" | ||
clientset "k8s.io/client-go/kubernetes" | ||
restclient "k8s.io/client-go/rest" | ||
"k8s.io/client-go/tools/events" | ||
"k8s.io/client-go/tools/leaderelection" | ||
ctrl "sigs.k8s.io/controller-runtime" | ||
|
||
deschedulerconfig "github.com/koordinator-sh/koordinator/pkg/descheduler/apis/config" | ||
) | ||
|
||
// Config has all the context to run a Scheduler | ||
type Config struct { | ||
// ComponentConfig is the scheduler server's configuration object. | ||
ComponentConfig deschedulerconfig.DeschedulerConfiguration | ||
|
||
// LoopbackClientConfig is a config for a privileged loopback connection | ||
LoopbackClientConfig *restclient.Config | ||
|
||
InsecureServing *apiserver.DeprecatedInsecureServingInfo // nil will disable serving on an insecure port | ||
InsecureMetricsServing *apiserver.DeprecatedInsecureServingInfo // non-nil if metrics should be served independently | ||
SecureServing *apiserver.SecureServingInfo | ||
|
||
Manager ctrl.Manager | ||
Client clientset.Interface | ||
KubeConfig *restclient.Config | ||
InformerFactory informers.SharedInformerFactory | ||
DynInformerFactory dynamicinformer.DynamicSharedInformerFactory | ||
|
||
// nolint:staticcheck // SA1019 this deprecated field still needs to be used for now. It will be removed once the migration is done. | ||
EventBroadcaster events.EventBroadcasterAdapter | ||
|
||
// LeaderElection is optional. | ||
LeaderElection *leaderelection.LeaderElectionConfig | ||
} | ||
|
||
type completedConfig struct { | ||
*Config | ||
} | ||
|
||
// CompletedConfig same as Config, just to swap private object. | ||
type CompletedConfig struct { | ||
// Embed a private pointer that cannot be instantiated outside of this package. | ||
*completedConfig | ||
} | ||
|
||
// Complete fills in any fields not set that are required to have valid data. It's mutating the receiver. | ||
func (c *Config) Complete() CompletedConfig { | ||
cc := completedConfig{c} | ||
return CompletedConfig{&cc} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,109 @@ | ||
/* | ||
Copyright 2022 The Koordinator Authors. | ||
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 options | ||
|
||
import ( | ||
"bytes" | ||
"fmt" | ||
"io" | ||
"os" | ||
|
||
"k8s.io/apimachinery/pkg/runtime" | ||
"k8s.io/klog/v2" | ||
|
||
deschedulerconfig "github.com/koordinator-sh/koordinator/pkg/descheduler/apis/config" | ||
deschedulerconfigscheme "github.com/koordinator-sh/koordinator/pkg/descheduler/apis/config/scheme" | ||
"github.com/koordinator-sh/koordinator/pkg/descheduler/apis/config/v1alpha2" | ||
) | ||
|
||
func loadConfigFromFile(file string) (*deschedulerconfig.DeschedulerConfiguration, error) { | ||
data, err := os.ReadFile(file) | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
return loadConfig(data) | ||
} | ||
|
||
func loadConfig(data []byte) (*deschedulerconfig.DeschedulerConfiguration, error) { | ||
// The UniversalDecoder runs defaulting and returns the internal type by default. | ||
obj, gvk, err := deschedulerconfigscheme.Codecs.UniversalDecoder().Decode(data, nil, nil) | ||
if err != nil { | ||
return nil, err | ||
} | ||
if cfgObj, ok := obj.(*deschedulerconfig.DeschedulerConfiguration); ok { | ||
// the field will be cleared later by API machinery during | ||
// conversion. See DeschedulerConfiguration internal type definition for | ||
// more details. | ||
cfgObj.TypeMeta.APIVersion = gvk.GroupVersion().String() | ||
return cfgObj, nil | ||
} | ||
return nil, fmt.Errorf("couldn't decode as DeschedulerConfiguration, got %s: ", gvk) | ||
} | ||
|
||
func encodeConfig(cfg *deschedulerconfig.DeschedulerConfiguration) (*bytes.Buffer, error) { | ||
buf := new(bytes.Buffer) | ||
const mediaType = runtime.ContentTypeYAML | ||
info, ok := runtime.SerializerInfoForMediaType(deschedulerconfigscheme.Codecs.SupportedMediaTypes(), mediaType) | ||
if !ok { | ||
return buf, fmt.Errorf("unable to locate encoder -- %q is not a supported media type", mediaType) | ||
} | ||
|
||
var encoder runtime.Encoder | ||
switch cfg.TypeMeta.APIVersion { | ||
case v1alpha2.SchemeGroupVersion.String(): | ||
encoder = deschedulerconfigscheme.Codecs.EncoderForVersion(info.Serializer, v1alpha2.SchemeGroupVersion) | ||
default: | ||
encoder = deschedulerconfigscheme.Codecs.EncoderForVersion(info.Serializer, v1alpha2.SchemeGroupVersion) | ||
} | ||
if err := encoder.Encode(cfg, buf); err != nil { | ||
return buf, err | ||
} | ||
return buf, nil | ||
} | ||
|
||
// LogOrWriteConfig logs the completed component config and writes it into the given file name as YAML, if either is enabled | ||
func LogOrWriteConfig(fileName string, cfg *deschedulerconfig.DeschedulerConfiguration, completedProfiles []deschedulerconfig.DeschedulerProfile) error { | ||
klogV := klog.V(2) | ||
if !klogV.Enabled() && len(fileName) == 0 { | ||
return nil | ||
} | ||
cfg.Profiles = completedProfiles | ||
|
||
buf, err := encodeConfig(cfg) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
if klogV.Enabled() { | ||
klogV.InfoS("Using component config", "config", buf.String()) | ||
} | ||
|
||
if len(fileName) > 0 { | ||
configFile, err := os.Create(fileName) | ||
if err != nil { | ||
return err | ||
} | ||
defer configFile.Close() | ||
if _, err := io.Copy(configFile, buf); err != nil { | ||
return err | ||
} | ||
klog.InfoS("Wrote configuration", "file", fileName) | ||
os.Exit(0) | ||
} | ||
return nil | ||
} |
Oops, something went wrong.