Plugin 系统

type Plugin interface {
    Name() string                    // Plugin 唯一名称
    OnSessionOpen(ssn *Session)      // Session 打开时调用，注册扩展点
    OnSessionClose(ssn *Session)     // Session 关闭时调用，清理资源
}

// 可选接口：支持绑定上下文扩展
type BindContextHandler interface {
    SetupBindContextExtension(state *k8sframework.CycleState, bindCtx *cache.BindContext)
}

type PluginBuilder = func(Arguments) Plugin

var pluginBuilders = map[string]PluginBuilder{}

// 注册 Plugin
func RegisterPluginBuilder(name string, pc PluginBuilder) {
    pluginBuilders[name] = pc
}

// 查找 Plugin
func GetPluginBuilder(name string) (PluginBuilder, bool) {
    pb, found := pluginBuilders[name]
    return pb, found
}

func init() {
    framework.RegisterPluginBuilder("drf", drf.New)
    framework.RegisterPluginBuilder("gang", gang.New)
    framework.RegisterPluginBuilder("predicates", predicates.New)
    framework.RegisterPluginBuilder("nodeorder", nodeorder.New)
    framework.RegisterPluginBuilder("proportion", proportion.New)
    framework.RegisterPluginBuilder("binpack", binpack.New)
    // ... 更多 Plugin
}

func LoadCustomPlugins(pluginsDir string) {
    // 扫描 pluginsDir 下所有 .so 文件
    // 动态加载并注册
}

func (gp *gangPlugin) preemptableFn(preemptor *api.TaskInfo, preemptees []*api.TaskInfo) ([]*api.TaskInfo, int) {
    var victims []*api.TaskInfo
    occupiedMap := make(map[api.JobID]int32)

    for _, preemptee := range preemptees {
        job := ssn.Jobs[preemptee.Job]
        occupiedMap[job.UID]++
        // 只有超过 minMember 的 Task 才可以被抢占
        if occupiedMap[job.UID] > job.MinAvailable {
            victims = append(victims, preemptee)
        }
    }
    return victims, Permit
}

plugins:
  - name: nodeorder
    arguments:
      nodeaffinity.weight: 10
      podaffinity.weight: 10
      tainttoleration.weight: 10
      leastrequested.weight: 1

plugins:
  - name: binpack
    arguments:
      binpack.weight: 10
      binpack.cpu: 1
      binpack.memory: 1
      binpack.resources: "nvidia.com/gpu"
      binpack.resources.nvidia.com/gpu: 2

tiers:
  - plugins:
      - name: gang          # Tier 1 - 最高优先级
        enabledJobOrder: true
        enabledJobReady: true
        enabledJobPipelined: true
      - name: priority
        enabledJobOrder: true
        enabledTaskOrder: true
  - plugins:
      - name: drf           # Tier 2
        enabledJobOrder: true
        enabledPreemptable: true
      - name: proportion
        enabledQueueOrder: true
        enabledReclaimable: true
  - plugins:
      - name: predicates    # Tier 3
        enabledPredicate: true
      - name: nodeorder
        enabledNodeOrder: true
      - name: binpack
        enabledNodeOrder: true

package myplugin

import (
    "volcano.sh/volcano/pkg/scheduler/api"
    "volcano.sh/volcano/pkg/scheduler/framework"
)

const PluginName = "myplugin"

type myPlugin struct {
    arguments framework.Arguments
}

func New(arguments framework.Arguments) framework.Plugin {
    return &myPlugin{arguments: arguments}
}

func (mp *myPlugin) Name() string {
    return PluginName
}

func (mp *myPlugin) OnSessionOpen(ssn *framework.Session) {
    // 注册扩展点
    ssn.AddNodeOrderFn(mp.Name(), func(task *api.TaskInfo, node *api.NodeInfo) (float64, error) {
        // 自定义节点打分逻辑
        return score, nil
    })
}

func (mp *myPlugin) OnSessionClose(ssn *framework.Session) {
    // 清理
}

import "volcano.sh/volcano/pkg/scheduler/plugins/myplugin"

func init() {
    // ...
    framework.RegisterPluginBuilder("myplugin", myplugin.New)
}

tiers:
  - plugins:
      - name: myplugin
        arguments:
          mykey: myvalue
        enabledNodeOrder: true

func (p *myPlugin) OnSessionOpen(ssn *framework.Session) {
    weight := 1.0
    if w, ok := p.arguments["weight"]; ok {
        weight = w.(float64)
    }
}