mirror of
https://github.com/kubernetes-sigs/descheduler.git
synced 2026-01-26 21:31:18 +01:00
67 lines
2.3 KiB
Go
67 lines
2.3 KiB
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/client-go/informers"
|
|
"k8s.io/client-go/tools/events"
|
|
|
|
clientset "k8s.io/client-go/kubernetes"
|
|
"sigs.k8s.io/descheduler/pkg/descheduler/evictions"
|
|
podutil "sigs.k8s.io/descheduler/pkg/descheduler/pod"
|
|
frameworkfake "sigs.k8s.io/descheduler/pkg/framework/fake"
|
|
"sigs.k8s.io/descheduler/pkg/framework/plugins/defaultevictor"
|
|
frameworktypes "sigs.k8s.io/descheduler/pkg/framework/types"
|
|
)
|
|
|
|
func InitFrameworkHandle(
|
|
ctx context.Context,
|
|
client clientset.Interface,
|
|
evictionOptions *evictions.Options,
|
|
defaultEvictorArgs defaultevictor.DefaultEvictorArgs,
|
|
getPodsAssignedToNodeSorter func([]*v1.Pod),
|
|
) (*frameworkfake.HandleImpl, *evictions.PodEvictor, error) {
|
|
sharedInformerFactory := informers.NewSharedInformerFactory(client, 0)
|
|
podInformer := sharedInformerFactory.Core().V1().Pods().Informer()
|
|
podsAssignedToNode, err := podutil.BuildGetPodsAssignedToNodeFunc(podInformer)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("Build get pods assigned to node function error: %v", err)
|
|
}
|
|
|
|
var getPodsAssignedToNode func(s string, filterFunc podutil.FilterFunc) ([]*v1.Pod, error)
|
|
if getPodsAssignedToNodeSorter != nil {
|
|
getPodsAssignedToNode = func(s string, filterFunc podutil.FilterFunc) ([]*v1.Pod, error) {
|
|
pods, err := podsAssignedToNode(s, filterFunc)
|
|
getPodsAssignedToNodeSorter(pods)
|
|
return pods, err
|
|
}
|
|
} else {
|
|
getPodsAssignedToNode = podsAssignedToNode
|
|
}
|
|
|
|
sharedInformerFactory.Start(ctx.Done())
|
|
sharedInformerFactory.WaitForCacheSync(ctx.Done())
|
|
eventRecorder := &events.FakeRecorder{}
|
|
podEvictor := evictions.NewPodEvictor(client, eventRecorder, evictionOptions)
|
|
evictorFilter, err := defaultevictor.New(
|
|
&defaultEvictorArgs,
|
|
&frameworkfake.HandleImpl{
|
|
ClientsetImpl: client,
|
|
GetPodsAssignedToNodeFuncImpl: getPodsAssignedToNode,
|
|
SharedInformerFactoryImpl: sharedInformerFactory,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("Unable to initialize the plugin: %v", err)
|
|
}
|
|
return &frameworkfake.HandleImpl{
|
|
ClientsetImpl: client,
|
|
GetPodsAssignedToNodeFuncImpl: getPodsAssignedToNode,
|
|
PodEvictorImpl: podEvictor,
|
|
EvictorFilterImpl: evictorFilter.(frameworktypes.EvictorPlugin),
|
|
SharedInformerFactoryImpl: sharedInformerFactory,
|
|
}, podEvictor, nil
|
|
}
|