Implement DeviceClassResolver interface

Author: gnufied

cluster-autoscaler/simulator/clustersnapshot/store/basic.go

@@ -360,6 +360,10 @@ func (snapshot *BasicSnapshotStore) DeviceClasses() fwk.DeviceClassLister {
 	return snapshot.DraSnapshot().DeviceClasses()
 }
 
+func (snapshot *BasicSnapshotStore) DeviceClassResolver() fwk.DeviceClassResolver {
+	return snapshot.DraSnapshot().DeviceClassResolver()
+}
+
 // List returns the list of nodes in the snapshot.
 func (snapshot *basicSnapshotStoreNodeLister) List() ([]fwk.NodeInfo, error) {
 	return (*BasicSnapshotStore)(snapshot).getInternalData().listNodeInfos(), nil

cluster-autoscaler/simulator/clustersnapshot/store/delta.go

@@ -443,6 +443,10 @@ func (snapshot *DeltaSnapshotStore) CSINodes() fwk.CSINodeLister {
 	return snapshot.csiSnapshot.CSINodes()
 }
 
+func (snapshot *DeltaSnapshotStore) DeviceClassResolver() fwk.DeviceClassResolver {
+	return snapshot.DraSnapshot().DeviceClassResolver()
+}
+
 // AddSchedulerNodeInfo adds a NodeInfo.
 func (snapshot *DeltaSnapshotStore) AddSchedulerNodeInfo(nodeInfo fwk.NodeInfo) error {
 	if _, err := snapshot.data.addNode(nodeInfo.Node()); err != nil {

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 newSnapshotDeviceClassResolver(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,39 @@
+package snapshot
+
+import (
+	v1 "k8s.io/api/core/v1"
+	resourceapi "k8s.io/api/resource/v1"
+	schedulerframework "k8s.io/kube-scheduler/framework"
+)
+
+type snapshotDeviceClassResolver struct {
+	// resourceName2class maps extended resource name to device class
+	resourceName2class map[v1.ResourceName]*resourceapi.DeviceClass
+}
+
+var _ schedulerframework.DeviceClassResolver = &snapshotDeviceClassResolver{}
+
+func newSnapshotDeviceClassResolver(snapshot *Snapshot) schedulerframework.DeviceClassResolver {
+	resourceName2class := make(map[v1.ResourceName]*resourceapi.DeviceClass)
+	for _, class := range snapshot.listDeviceClasses() {
+		classResourceName := class.Name
+		extendedResourceName := class.Spec.ExtendedResourceName
+		if extendedResourceName != nil {
+			resourceName2class[v1.ResourceName(*extendedResourceName)] = class
+		}
+
+		// Also add the default extended resource name
+		defaultResourceName := v1.ResourceName(resourceapi.ResourceDeviceClassPrefix + classResourceName)
+		resourceName2class[defaultResourceName] = class
+	}
+	return snapshotDeviceClassResolver{resourceName2class: resourceName2class}
+}
+
+// GetDeviceClass returns the device class for the given extended resource name
+func (s snapshotDeviceClassResolver) GetDeviceClass(resourceName v1.ResourceName) *resourceapi.DeviceClass {
+	class, found := s.resourceName2class[resourceName]
+	if !found {
+		return nil
+	}
+	return class
+}

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