commiting uncommited changes on 2025-11-09 15:45:53 -0500

Author: gnufied

cluster-autoscaler/simulator/dynamicresources/snapshot/snapshot.go

@@ -91,6 +91,10 @@ func (s *Snapshot) ResourceClaims() schedulerframework.ResourceClaimTracker {
 	return snapshotClaimTracker{snapshot: s}
 }
 
+func (s *Snapshot) DeviceClassResolver() schedulerframework.DeviceClassResolver {
+	return snapshotClassResolver{snapshot: s}
+}
+
 // ResourceSlices exposes the Snapshot as schedulerframework.ResourceSliceLister, in order to interact with
 // the scheduler framework.
 func (s *Snapshot) ResourceSlices() schedulerframework.ResourceSliceLister {

cluster-autoscaler/simulator/dynamicresources/snapshot/snapshot_class_resolver.go

@@ -0,0 +1,14 @@
+package snapshot
+
+import (
+	v1 "k8s.io/api/core/v1"
+	resourceapi "k8s.io/api/resource/v1"
+)
+
+type snapshotClassResolver struct {
+	snapshot *Snapshot
+}
+
+func (s snapshotClassResolver) GetDeviceClass(resourceName v1.ResourceName) *resourceapi.DeviceClass {
+	return s.snapshot.getDeviceClass(resourceName)
+}

cluster-autoscaler/simulator/framework/delegating_shared_lister.go

@@ -19,6 +19,7 @@ package framework
 import (
 	"fmt"
 
+	v1 "k8s.io/api/core/v1"
 	resourceapi "k8s.io/api/resource/v1"
 	storagev1 "k8s.io/api/storage/v1"
 	"k8s.io/apimachinery/pkg/types"
@@ -83,6 +84,10 @@ func (lister *DelegatingSchedulerSharedLister) UpdateDelegate(delegate SharedLis
 	lister.delegate = delegate
 }
 
+func (lister *DelegatingSchedulerSharedLister) DeviceClassResolver() fwk.DeviceClassResolver {
+	return lister.delegate.DeviceClassResolver()
+}
+
 // ResetDelegate resets delegate to
 func (lister *DelegatingSchedulerSharedLister) ResetDelegate() {
 	lister.delegate = unsetSharedListerSingleton
@@ -95,6 +100,7 @@ type unsetResourceClaimTracker unsetSharedLister
 type unsetResourceSliceLister unsetSharedLister
 type unsetDeviceClassLister unsetSharedLister
 type unsetCSINodeLister unsetSharedLister
+type unsetDeviceClassResolver unsetSharedLister
 
 // List always returns an error
 func (lister *unsetNodeInfoLister) List() ([]fwk.NodeInfo, error) {
@@ -178,6 +184,10 @@ func (u *unsetDeviceClassLister) Get(className string) (*resourceapi.DeviceClass
 	return nil, fmt.Errorf("lister not set in delegate")
 }
 
+func (u *unsetDeviceClassResolver) GetDeviceClass(resourceName v1.ResourceName) *resourceapi.DeviceClass {
+	return nil
+}
+
 // NodeInfos returns a fake NodeInfoLister which always returns an error
 func (lister *unsetSharedLister) NodeInfos() fwk.NodeInfoLister {
 	return (*unsetNodeInfoLister)(lister)
@@ -204,4 +214,8 @@ func (lister *unsetSharedLister) CSINodes() fwk.CSINodeLister {
 	return (*unsetCSINodeLister)(lister)
 }
 
+func (lister *unsetSharedLister) DeviceClassResolver() fwk.DeviceClassResolver {
+	return (*unsetDeviceClassResolver)(lister)
+}
+
 var unsetSharedListerSingleton *unsetSharedLister