Merge pull request #2398 from Poor12/add-cluster-check

add cluster resource model check
This commit is contained in:
karmada-bot 2022-08-23 16:26:51 +08:00 committed by GitHub
commit 4657282293
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 414 additions and 5 deletions

View File

@ -2,6 +2,7 @@ package mutation
import (
"math"
"sort"
corev1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/resource"
@ -30,11 +31,29 @@ func MutateClusterTaints(taints []corev1.Taint) {
}
}
// SetDefaultClusterResourceModels set default cluster resource models for cluster.
func SetDefaultClusterResourceModels(cluster *clusterapis.Cluster) {
if cluster.Spec.ResourceModels != nil {
// StandardizeClusterResourceModels set cluster resource models for given order and boundary value.
func StandardizeClusterResourceModels(models []clusterapis.ResourceModel) {
if len(models) == 0 {
return
}
sort.Slice(models, func(i, j int) bool {
return models[i].Grade < models[j].Grade
})
// The Min value of the first grade(usually 0) always acts as zero.
for index := range models[0].Ranges {
models[0].Ranges[index].Min.Set(0)
}
// The Max value of the last grade always acts as MaxInt64.
for index := range models[len(models)-1].Ranges {
models[len(models)-1].Ranges[index].Max.Set(math.MaxInt64)
}
}
// SetDefaultClusterResourceModels set default cluster resource models for cluster.
func SetDefaultClusterResourceModels(cluster *clusterapis.Cluster) {
cluster.Spec.ResourceModels = []clusterapis.ResourceModel{
{
Grade: 0,

View File

@ -0,0 +1,125 @@
package mutation
import (
"math"
"reflect"
"testing"
"k8s.io/apimachinery/pkg/api/resource"
clusterapis "github.com/karmada-io/karmada/pkg/apis/cluster"
)
func TestStandardizeClusterResourceModels(t *testing.T) {
testCases := map[string]struct {
models []clusterapis.ResourceModel
expectedModels []clusterapis.ResourceModel
}{
"sort models": {
models: []clusterapis.ResourceModel{
{
Grade: 2,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
{
Grade: 1,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
},
expectedModels: []clusterapis.ResourceModel{
{
Grade: 1,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
{
Grade: 2,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
"start with 0": {
models: []clusterapis.ResourceModel{
{
Grade: 1,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(1, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
expectedModels: []clusterapis.ResourceModel{
{
Grade: 1,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
"end with MaxInt64": {
models: []clusterapis.ResourceModel{
{
Grade: 1,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
},
expectedModels: []clusterapis.ResourceModel{
{
Grade: 1,
Ranges: []clusterapis.ResourceModelRange{
{
Name: clusterapis.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
}
for name, testCase := range testCases {
StandardizeClusterResourceModels(testCase.models)
if !reflect.DeepEqual(testCase.models, testCase.expectedModels) {
t.Errorf("expected sorted resource models for %q, but it did not work", name)
return
}
}
}

View File

@ -2,6 +2,7 @@ package validation
import (
"fmt"
"math"
"net/url"
apimachineryvalidation "k8s.io/apimachinery/pkg/api/validation"
@ -34,7 +35,10 @@ func ValidateClusterName(name string) []string {
return kubevalidation.IsDNS1123Label(name)
}
var supportedSyncModes = sets.NewString(string(api.Pull), string(api.Push))
var (
supportedSyncModes = sets.NewString(string(api.Pull), string(api.Push))
supportedResourceNames = sets.NewString(string(api.ResourceCPU), string(api.ResourceMemory), string(api.ResourceStorage), string(api.ResourceEphemeralStorage))
)
// ValidateCluster tests if required fields in the Cluster are set.
func ValidateCluster(cluster *api.Cluster) field.ErrorList {
@ -90,6 +94,12 @@ func ValidateClusterSpec(spec *api.ClusterSpec, fldPath *field.Path) field.Error
allErrs = append(allErrs, lifted.ValidateClusterTaints(spec.Taints, fldPath.Child("taints"))...)
}
if len(spec.ResourceModels) > 0 {
if err := ValidateClusterResourceModels(fldPath.Child("resourceModels"), spec.ResourceModels); err != nil {
allErrs = append(allErrs, err)
}
}
return allErrs
}
@ -133,3 +143,41 @@ func ValidateClusterProxyURL(fldPath *field.Path, proxyURL string) field.ErrorLi
}
return allErrs
}
// ValidateClusterResourceModels validates cluster's resource models.
func ValidateClusterResourceModels(fldPath *field.Path, models []api.ResourceModel) *field.Error {
for i, resourceModel := range models {
if i != 0 && resourceModel.Grade == models[i-1].Grade {
return field.Invalid(fldPath, models, "The grade of each models should not be the same")
}
if i != 0 && len(models[i-1].Ranges) != len(resourceModel.Ranges) {
return field.Invalid(fldPath, models, "The number of resource types should be the same")
}
for j, resourceModelRange := range resourceModel.Ranges {
if !supportedResourceNames.Has(string(resourceModelRange.Name)) {
return field.NotSupported(fldPath.Child("ranges").Child("name"), resourceModelRange.Name, supportedResourceNames.List())
}
if resourceModelRange.Max.Cmp(resourceModelRange.Min) <= 0 {
return field.Invalid(fldPath, models, "The max value of each resource must be greater than the min value")
}
if i == 0 {
if !resourceModelRange.Min.IsZero() {
return field.Invalid(fldPath, models, "The min value of each resource in the first model should be 0")
}
} else if models[i-1].Ranges[j].Name != resourceModelRange.Name {
return field.Invalid(fldPath, models, "The resource types of each models should be the same")
} else if models[i-1].Ranges[j].Max != resourceModelRange.Min {
return field.Invalid(fldPath, models, "Model intervals for resources must be contiguous and non-overlapping")
}
if i == len(models)-1 {
if resourceModelRange.Max.Value() != math.MaxInt64 {
return field.Invalid(fldPath, models, "The max value of each resource in the last model should be MaxInt64")
}
}
}
}
return nil
}

View File

@ -1,10 +1,12 @@
package validation
import (
"math"
"strings"
"testing"
corev1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/resource"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
api "github.com/karmada-io/karmada/pkg/apis/cluster"
@ -67,6 +69,211 @@ func TestValidateCluster(t *testing.T) {
},
expectError: true,
},
"invalid cluster resource models with the same grade": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
{
Grade: 2,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models with different numbers of resource types": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
{
Grade: 2,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
{
Name: api.ResourceMemory,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models with unreasonable ranges": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(0, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models which the min value of each resource in the first model is not 0": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(1, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models which the max value of each resource in the last model is not MaxInt64": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models which the resource types of each models are different": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
{
Grade: 2,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceMemory,
Min: *resource.NewQuantity(2, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models with contiguous and non-overlapping": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(2, resource.DecimalSI),
},
},
},
{
Grade: 2,
Ranges: []api.ResourceModelRange{
{
Name: api.ResourceCPU,
Min: *resource.NewQuantity(1, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
"invalid cluster resource models with invalid resource name": {
cluster: api.Cluster{
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
Spec: api.ClusterSpec{
ResourceModels: []api.ResourceModel{
{
Grade: 1,
Ranges: []api.ResourceModelRange{
{
Name: "test",
Min: *resource.NewQuantity(0, resource.DecimalSI),
Max: *resource.NewQuantity(math.MaxInt64, resource.DecimalSI),
},
},
},
},
},
},
expectError: true,
},
}
for name, testCase := range testCases {

View File

@ -74,12 +74,22 @@ func (Strategy) GetResetFields() map[fieldpath.APIVersion]*fieldpath.Set {
func (Strategy) PrepareForCreate(ctx context.Context, obj runtime.Object) {
cluster := obj.(*clusterapis.Cluster)
if utilfeature.DefaultMutableFeatureGate.Enabled(features.CustomizedClusterResourceModeling) {
if len(cluster.Spec.ResourceModels) == 0 {
mutation.SetDefaultClusterResourceModels(cluster)
} else {
mutation.StandardizeClusterResourceModels(cluster.Spec.ResourceModels)
}
}
}
// PrepareForUpdate is invoked on update before validation to normalize the object.
func (Strategy) PrepareForUpdate(ctx context.Context, obj, old runtime.Object) {
cluster := obj.(*clusterapis.Cluster)
if utilfeature.DefaultMutableFeatureGate.Enabled(features.CustomizedClusterResourceModeling) {
if len(cluster.Spec.ResourceModels) != 0 {
mutation.StandardizeClusterResourceModels(cluster.Spec.ResourceModels)
}
}
}
// Validate returns an ErrorList with validation errors or nil.