mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-09-20 23:49:53 +00:00
116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes 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 kubectl
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
autoscalingv1 "k8s.io/api/autoscaling/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
)
|
|
|
|
type HorizontalPodAutoscalerV1 struct{}
|
|
|
|
func (HorizontalPodAutoscalerV1) ParamNames() []GeneratorParam {
|
|
return []GeneratorParam{
|
|
{"default-name", true},
|
|
{"name", false},
|
|
{"scaleRef-kind", false},
|
|
{"scaleRef-name", false},
|
|
{"scaleRef-apiVersion", false},
|
|
{"min", false},
|
|
{"max", true},
|
|
{"cpu-percent", false},
|
|
}
|
|
}
|
|
|
|
func (HorizontalPodAutoscalerV1) Generate(genericParams map[string]interface{}) (runtime.Object, error) {
|
|
return generateHPA(genericParams)
|
|
}
|
|
|
|
func generateHPA(genericParams map[string]interface{}) (runtime.Object, error) {
|
|
params := map[string]string{}
|
|
for key, value := range genericParams {
|
|
strVal, isString := value.(string)
|
|
if !isString {
|
|
return nil, fmt.Errorf("expected string, saw %v for '%s'", value, key)
|
|
}
|
|
params[key] = strVal
|
|
}
|
|
|
|
name, found := params["name"]
|
|
if !found || len(name) == 0 {
|
|
name, found = params["default-name"]
|
|
if !found || len(name) == 0 {
|
|
return nil, fmt.Errorf("'name' is a required parameter.")
|
|
}
|
|
}
|
|
minString, found := params["min"]
|
|
min := -1
|
|
var err error
|
|
if found {
|
|
if min, err = strconv.Atoi(minString); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
maxString, found := params["max"]
|
|
if !found {
|
|
return nil, fmt.Errorf("'max' is a required parameter.")
|
|
}
|
|
max, err := strconv.Atoi(maxString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if min > max {
|
|
return nil, fmt.Errorf("'max' must be greater than or equal to 'min'.")
|
|
}
|
|
|
|
cpuString, found := params["cpu-percent"]
|
|
cpu := -1
|
|
if found {
|
|
if cpu, err = strconv.Atoi(cpuString); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
scaler := autoscalingv1.HorizontalPodAutoscaler{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: name,
|
|
},
|
|
Spec: autoscalingv1.HorizontalPodAutoscalerSpec{
|
|
ScaleTargetRef: autoscalingv1.CrossVersionObjectReference{
|
|
Kind: params["scaleRef-kind"],
|
|
Name: params["scaleRef-name"],
|
|
APIVersion: params["scaleRef-apiVersion"],
|
|
},
|
|
MaxReplicas: int32(max),
|
|
},
|
|
}
|
|
if min > 0 {
|
|
v := int32(min)
|
|
scaler.Spec.MinReplicas = &v
|
|
}
|
|
if cpu >= 0 {
|
|
c := int32(cpu)
|
|
scaler.Spec.TargetCPUUtilizationPercentage = &c
|
|
}
|
|
return &scaler, nil
|
|
}
|