scheduler support ephemeral-storage resources

Signed-off-by: wulixuan <[email protected]>
volcano-sh · Sep 30, 2022 · e916867 · e916867
1 parent c1537d5
commit e916867
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/pkg/scheduler/api/resource_info.go b/pkg/scheduler/api/resource_info.go
@@ -76,6 +76,8 @@ func NewResource(rl v1.ResourceList) *Resource {
 			r.Memory += float64(rQuant.Value())
 		case v1.ResourcePods:
 			r.MaxTaskNum += int(rQuant.Value())
+		case v1.ResourceEphemeralStorage:
+			r.AddScalar(rName, float64(rQuant.MilliValue()))
 		default:
 			if IsCountQuota(rName) {
 				continue

diff --git a/pkg/scheduler/api/resource_info_test.go b/pkg/scheduler/api/resource_info_test.go
@@ -40,11 +40,15 @@ func TestNewResource(t *testing.T) {
 				v1.ResourceMemory:                   *resource.NewQuantity(2000, resource.BinarySI),
 				"scalar.test/" + "scalar1":          *resource.NewQuantity(1, resource.DecimalSI),
 				v1.ResourceHugePagesPrefix + "test": *resource.NewQuantity(2, resource.BinarySI),
+				v1.ResourceEphemeralStorage:         *resource.NewQuantity(3, resource.BinarySI),
 			},
 			expected: &Resource{
-				MilliCPU:        4,
-				Memory:          2000,
-				ScalarResources: map[v1.ResourceName]float64{"scalar.test/scalar1": 1000, "hugepages-test": 2000},
+				MilliCPU: 4,
+				Memory:   2000,
+				ScalarResources: map[v1.ResourceName]float64{
+					"scalar.test/scalar1":       1000,
+					"hugepages-test":            2000,
+					v1.ResourceEphemeralStorage: 3000},
 			},
 		},
 	}