如何自定义 Kube-Scheduler 插件

k8s的发展越来越像是一个框架，然后把各种扩展的能力留给开发者。开发者可以基于这些接口结合自己的业务场景，实现自己的场景化需求。其中kube scheduler 就是充分体现了这个特质，关于kube scheduler 本身的介绍参加之前的文章，今天我想介绍如何给scheduler 添加一个调度plugin。

我们首先通过yaml定义这个plugin

1. apiVersion: kubescheduler.config.k8s.io/v1beta1 
2. kind: KubeSchedulerConfiguration 
3. clientConnection: 
4.   kubeconfig: "/etc/kubernetes/scheduler.conf"
5. profiles: 
6. - schedulerName: default-scheduler 
7.   plugins: 
8.     score: 
9.       enabled: 
10.       - name: HelloWorldPlugin 
11.       disabled: 
12.       - name: "*"
13.   pluginConfig: 
14.   - name: HelloWorldPlugin 
15.     args: 
16.       xxx: "xxx"
17.       yyy: "123"
18.       zzz: 3 

我们定义了一个 HelloWorldPlugin 的插件，并且定义了这个插件的启动参数。然后需要修改kube scheduler启动参数通过 --config 指定上面的配置文件。

接下来我们就需要实现这个插件，scheduler是通过每个插件的打分的方式确定调度的主机。所以我们需要实现一个打分的接口

1. type ScorePlugin interface { 
2.   Plugin 
3.   // 打分 
4.   Score(ctx context.Context, state *CycleState, p *v1.Pod, nodeName string) (int64, *Status) 
5.    
6.   ScoreExtensions() ScoreExtensions 
7. } 
8.  
9. type ScoreExtensions interface { 
10.   // 打分归一化，保证每个插件的公平性 
11.   NormalizeScore(ctx context.Context, state *CycleState, p *v1.Pod, scores NodeScoreList) *Status 
12. } 

我们根据自己的业务需求实现这个接口，譬如下面这个例子，基于主机网络带宽的调度：首先通过promethues获取主机的网络流量，打分依据网络流量大小。

1. func (n *HelloWorldPlugin) Score(ctx context.Context, state *framework.CycleState, p *v1.Pod, nodeName string) (int64, *framework.Status) { 
2.     nodeBandwidth, err := n.prometheus.GetNodeBandwidthMeasure(nodeName) 
3.     if err != nil { 
4.         return 0, framework.NewStatus(framework.Error, fmt.Sprintf("error getting node bandwidth measure: %s", err)) 
5.     } 
6.  
7.     klog.Infof("[NetworkTraffic] node '%s' bandwidth: %s", nodeName, nodeBandwidth.Value) 
8.     return int64(nodeBandwidth.Value), nil 
9. } 

我们希望网络流量越大，得分越少，于是在归一化处理的时候，我们通过下面简单公式转化成最终的分数。

1. func (n *HelloWorldPlugin) NormalizeScore(ctx context.Context, state *framework.CycleState, pod *v1.Pod, scores framework.NodeScoreList) *framework.Status { 
2.     for i, node := range scores { 
3.         scores[i].Score = framework.MaxNodeScore - (node.Score * framework.MaxNodeScore / higherScore) 
4.     } 
5.  
6.     klog.Infof("[NetworkTraffic] Nodes final score: %v", scores) 
7.     return nil 
8. } 

这样一个简单的，基于网络流量调度的插件就实现了。

原文链接：https://www.toutiao.com/i7046924965886591502/