Volcano调度器对PreemptionPolicy的支持实现分析

背景介绍

在Kubernetes生态系统中，Volcano作为一款面向高性能计算场景的批处理调度器，其调度能力直接影响着集群资源利用率和作业执行效率。其中，优先级抢占机制是调度器的重要功能之一，它允许高优先级任务抢占低优先级任务的资源。Kubernetes原生支持通过PriorityClass定义优先级，并在1.19版本引入了preemptionPolicy字段，用于控制是否允许抢占行为。

问题发现

在Volcano的现有实现中，虽然支持基本的优先级抢占功能，但尚未完整实现Kubernetes PriorityClass中的preemptionPolicy策略。具体表现为：

1. Volcano缓存中仅将PriorityClass简化为一个数值，丢失了preemptionPolicy等附加信息
2. 调度逻辑中没有考虑preemptionPolicy为Never的情况
3. 对Volcano Job级别的preemptionPolicy支持不完整

技术方案设计

核心修改点

1. 缓存层改造：需要扩展Volcano调度器缓存中的PriorityClass表示方式，从单纯的数值扩展为包含preemptionPolicy等完整信息的结构体。

2. 调度逻辑增强：

   • 在抢占逻辑中增加preemptionPolicy检查
   • 当preemptionPolicy为Never时，跳过该任务的抢占行为
   • 同时考虑Pod级别和Job级别的preemptionPolicy

3. 控制器协调：确保vc-controller在创建Pod时正确传递Job级别的PriorityClass配置，包括preemptionPolicy。

实现细节

在抢占逻辑的关键路径上，需要添加如下判断逻辑：

if task.Pod.Spec.PreemptionPolicy != nil && *task.Pod.Spec.PreemptionPolicy == v1.PreemptNever {
    // 记录日志并跳过抢占
    klog.V(3).Infof("Task %s/%s is not eligible to preempt other tasks due to preemptionPolicy is Never", task.Namespace, task.Name)
    return
}

应用场景

preemptionPolicy的支持为集群管理员提供了更精细的资源管控能力：

1. 关键任务保护：将关键业务设置为高优先级但禁止抢占，既保证资源分配又避免影响其他服务
2. 服务质量分级：实现多级服务质量保障，区分可抢占和不可抢占的工作负载
3. 资源利用率优化：允许非关键业务灵活抢占，提高整体资源利用率

实现考量

在技术方案讨论过程中，社区对几个关键问题进行了深入探讨：

1. Pod级别与Job级别的支持：虽然Kubernetes原生仅支持Pod级别，但考虑到Volcano主要面向批处理场景，最终决定同时支持两个级别的preemptionPolicy。

2. 优先级继承机制：确保Job中创建的Pod能够正确继承Job的PriorityClass配置，包括preemptionPolicy。

3. 与现有功能兼容：保持与Volcano已有抢占、回收等功能的兼容性，确保平滑升级。

总结

Volcano对preemptionPolicy的完整支持，使其调度能力更加贴近生产环境需求，为用户提供了更灵活的资源管控手段。这一改进不仅完善了与Kubernetes的兼容性，也增强了在高性能计算场景下的调度精细度。通过合理的preemptionPolicy配置，用户可以更好地平衡资源利用率和业务稳定性，满足多样化的业务需求。