mirror of
https://github.com/nclabteam/THPA.git
synced 2025-02-12 10:31:54 +00:00
21 lines
632 B
Go
21 lines
632 B
Go
package main
|
|
|
|
import (
|
|
"k8s.io/api/core/v1"
|
|
schedulerapi "k8s.io/kubernetes/pkg/scheduler/apis/extender/v1"
|
|
)
|
|
|
|
type Preemption struct {
|
|
Func func(
|
|
pod v1.Pod,
|
|
nodeNameToVictims map[string]*schedulerapi.Victims,
|
|
nodeNameToMetaVictims map[string]*schedulerapi.MetaVictims,
|
|
) map[string]*schedulerapi.MetaVictims
|
|
}
|
|
|
|
func (b Preemption) Handler(args schedulerapi.ExtenderPreemptionArgs,) *schedulerapi.ExtenderPreemptionResult {
|
|
nodeNameToMetaVictims := b.Func(*args.Pod, args.NodeNameToVictims, args.NodeNameToMetaVictims)
|
|
return &schedulerapi.ExtenderPreemptionResult{
|
|
NodeNameToMetaVictims: nodeNameToMetaVictims,
|
|
}
|
|
}
|