@@ -48,7 +48,7 @@ import (
48
48
"k8s.io/kubernetes/pkg/kubelet/lifecycle"
49
49
"k8s.io/kubernetes/pkg/kubelet/metrics"
50
50
"k8s.io/kubernetes/pkg/kubelet/pluginmanager/cache"
51
- schedulernodeinfo "k8s.io/kubernetes/pkg/scheduler/nodeinfo "
51
+ schedulerframework "k8s.io/kubernetes/pkg/scheduler/framework/v1alpha1 "
52
52
"k8s.io/kubernetes/pkg/util/selinux"
53
53
)
54
54
@@ -391,7 +391,7 @@ func (m *ManagerImpl) Allocate(pod *v1.Pod, container *v1.Container) error {
391
391
}
392
392
393
393
// UpdatePluginResources updates node resources based on devices already allocated to pods.
394
- func (m * ManagerImpl ) UpdatePluginResources (node * schedulernodeinfo .NodeInfo , attrs * lifecycle.PodAdmitAttributes ) error {
394
+ func (m * ManagerImpl ) UpdatePluginResources (node * schedulerframework .NodeInfo , attrs * lifecycle.PodAdmitAttributes ) error {
395
395
pod := attrs .Pod
396
396
397
397
m .mutex .Lock ()
@@ -924,9 +924,9 @@ func (m *ManagerImpl) callPreStartContainerIfNeeded(podUID, contName, resource s
924
924
// and if necessary, updates allocatableResource in nodeInfo to at least equal to
925
925
// the allocated capacity. This allows pods that have already been scheduled on
926
926
// the node to pass GeneralPredicates admission checking even upon device plugin failure.
927
- func (m * ManagerImpl ) sanitizeNodeAllocatable (node * schedulernodeinfo .NodeInfo ) {
928
- var newAllocatableResource * schedulernodeinfo .Resource
929
- allocatableResource := node .AllocatableResource ()
927
+ func (m * ManagerImpl ) sanitizeNodeAllocatable (node * schedulerframework .NodeInfo ) {
928
+ var newAllocatableResource * schedulerframework .Resource
929
+ allocatableResource := node .Allocatable
930
930
if allocatableResource .ScalarResources == nil {
931
931
allocatableResource .ScalarResources = make (map [v1.ResourceName ]int64 )
932
932
}
@@ -944,7 +944,7 @@ func (m *ManagerImpl) sanitizeNodeAllocatable(node *schedulernodeinfo.NodeInfo)
944
944
newAllocatableResource .ScalarResources [v1 .ResourceName (resource )] = int64 (needed )
945
945
}
946
946
if newAllocatableResource != nil {
947
- node .SetAllocatableResource ( newAllocatableResource )
947
+ node .Allocatable = newAllocatableResource
948
948
}
949
949
}
950
950
0 commit comments