Skip to content

Commit 4310732

Browse files
authored
Merge pull request kubernetes#91102 from chendave/capacity_ratio
Move `RequestedToCapacityRatio` Args defaults to versioned packages
2 parents fdbb960 + 838a0a8 commit 4310732

File tree

6 files changed

+49
-11
lines changed

6 files changed

+49
-11
lines changed

pkg/scheduler/apis/config/scheme/scheme_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,8 @@ profiles:
9898
{
9999
Name: "RequestedToCapacityRatio",
100100
Args: &config.RequestedToCapacityRatioArgs{
101-
Shape: []config.UtilizationShapePoint{{Utilization: 1}},
101+
Shape: []config.UtilizationShapePoint{{Utilization: 1}},
102+
Resources: []config.ResourceSpec{{Name: "cpu", Weight: 1}, {Name: "memory", Weight: 1}},
102103
},
103104
},
104105
{

pkg/scheduler/apis/config/v1alpha2/defaults.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,3 +185,10 @@ func SetDefaults_NodeResourcesMostAllocatedArgs(obj *v1alpha2.NodeResourcesMostA
185185
obj.Resources = append(obj.Resources, defaultResourceSpec...)
186186
}
187187
}
188+
189+
func SetDefaults_RequestedToCapacityRatioArgs(obj *v1alpha2.RequestedToCapacityRatioArgs) {
190+
if len(obj.Resources) == 0 {
191+
// If no resources specified, used the default set.
192+
obj.Resources = append(obj.Resources, defaultResourceSpec...)
193+
}
194+
}

pkg/scheduler/apis/config/v1alpha2/defaults_test.go

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -345,6 +345,29 @@ func TestPluginArgsDefaults(t *testing.T) {
345345
},
346346
},
347347
},
348+
{
349+
name: "NodeResourcesMostAllocatedArgs resources empty",
350+
in: &v1alpha2.NodeResourcesMostAllocatedArgs{},
351+
want: &v1alpha2.NodeResourcesMostAllocatedArgs{
352+
Resources: []v1alpha2.ResourceSpec{
353+
{Name: "cpu", Weight: 1},
354+
{Name: "memory", Weight: 1},
355+
},
356+
},
357+
},
358+
{
359+
name: "NodeResourcesMostAllocatedArgs resources with value",
360+
in: &v1alpha2.NodeResourcesMostAllocatedArgs{
361+
Resources: []v1alpha2.ResourceSpec{
362+
{Name: "resource", Weight: 2},
363+
},
364+
},
365+
want: &v1alpha2.NodeResourcesMostAllocatedArgs{
366+
Resources: []v1alpha2.ResourceSpec{
367+
{Name: "resource", Weight: 2},
368+
},
369+
},
370+
},
348371
}
349372
for _, tc := range tests {
350373
scheme := runtime.NewScheme()

pkg/scheduler/apis/config/v1alpha2/zz_generated.defaults.go

Lines changed: 7 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pkg/scheduler/framework/plugins/noderesources/requested_to_capacity_ratio.go

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -76,10 +76,6 @@ func NewRequestedToCapacityRatio(plArgs runtime.Object, handle framework.Framewo
7676
resourceToWeightMap[v1.ResourceName(resource.Name)] = 1
7777
}
7878
}
79-
if len(args.Resources) == 0 {
80-
// If no resources specified, used the default set.
81-
resourceToWeightMap = defaultRequestedRatioResources
82-
}
8379

8480
return &RequestedToCapacityRatio{
8581
handle: handle,

pkg/scheduler/framework/v1alpha1/framework_test.go

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -496,10 +496,12 @@ func TestNewFrameworkPluginDefaults(t *testing.T) {
496496
"InterPodAffinity": &config.InterPodAffinityArgs{
497497
HardPodAffinityWeight: 1,
498498
},
499-
"NodeLabel": &config.NodeLabelArgs{},
500-
"NodeResourcesFit": &config.NodeResourcesFitArgs{},
501-
"RequestedToCapacityRatio": &config.RequestedToCapacityRatioArgs{},
502-
"PodTopologySpread": &config.PodTopologySpreadArgs{},
499+
"NodeLabel": &config.NodeLabelArgs{},
500+
"NodeResourcesFit": &config.NodeResourcesFitArgs{},
501+
"RequestedToCapacityRatio": &config.RequestedToCapacityRatioArgs{
502+
Resources: []config.ResourceSpec{{Name: "cpu", Weight: 1}, {Name: "memory", Weight: 1}},
503+
},
504+
"PodTopologySpread": &config.PodTopologySpreadArgs{},
503505
},
504506
},
505507
{
@@ -526,8 +528,10 @@ func TestNewFrameworkPluginDefaults(t *testing.T) {
526528
"NodeResourcesFit": &config.NodeResourcesFitArgs{
527529
IgnoredResources: []string{"example.com/foo"},
528530
},
529-
"RequestedToCapacityRatio": &config.RequestedToCapacityRatioArgs{},
530-
"PodTopologySpread": &config.PodTopologySpreadArgs{},
531+
"RequestedToCapacityRatio": &config.RequestedToCapacityRatioArgs{
532+
Resources: []config.ResourceSpec{{Name: "cpu", Weight: 1}, {Name: "memory", Weight: 1}},
533+
},
534+
"PodTopologySpread": &config.PodTopologySpreadArgs{},
531535
},
532536
},
533537
}

0 commit comments

Comments
 (0)