1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// pkg/diskAvailable/diskAvailable.go
package diskAvailable
import (
"context"
"encoding/json"
"fmt"
"io/ioutil"
"k8s.io/apimachinery/pkg/runtime"
"net/http"
"strconv"
v1 "k8s.io/api/core/v1"
"k8s.io/kubernetes/pkg/scheduler/framework"
)
const (
Name = "DiskAvailable" // 插件名称改为 DiskAvailable
RequiredLabel = "use-disk-score" // 固定标签键
)
type DiskAvailable struct {
handle framework.Handle
}
var _ framework.ScorePlugin = &DiskAvailable{}
// New 初始化插件实例
func New(_ context.Context, plArgs runtime.Object, h framework.Handle) (framework.Plugin, error) {
d := &DiskAvailable{
handle: h,
}
return d, nil
}
// Name 返回插件名称
func (d *DiskAvailable) Name() string {
return Name
}
// Score 计算节点得分
func (d *DiskAvailable) Score(ctx context.Context, state *framework.CycleState, pod *v1.Pod, nodeName string) (int64, *framework.Status) {
// 检查 Pod 是否包含指定标签
if _, exists := pod.Labels[RequiredLabel]; !exists {
// 如果标签不存在，返回默认分数 0
return 0, framework.NewStatus(framework.Success, "Pod does not have the required label")
}
// 从 Prometheus 获取磁盘大小
diskSize, err := fetchDiskSizeFromPrometheus(nodeName)
if err != nil {
return 0, framework.NewStatus(framework.Error, fmt.Sprintf("Failed to fetch disk size for node %s: %v", nodeName, err))
}
// 假设磁盘大小范围为 [0, 100] GB，将其映射到 [0, 100] 分数
score := int64((diskSize / 250.0) * 100)
if score > 100 {
score = 100
}
return score, nil
}
// ScoreExtensions 返回 nil，因为我们不需要扩展
func (d *DiskAvailable) ScoreExtensions() framework.ScoreExtensions {
return nil
}
// 从 Prometheus 获取磁盘大小
func fetchDiskSizeFromPrometheus(nodeName string) (float64, error) {
// Prometheus 查询 URL
var hostIP string
if nodeName == "node2" {
hostIP = "192.168.1.202"
} else if nodeName == "node3" {
hostIP = "192.168.1.203"
}
prometheusQueryURL := fmt.Sprintf("http://kube-prometheus-stack-prometheus.kube-prometheus-stack.svc.cluster.local:9090/api/v1/query?query=node_filesystem_avail_bytes{instance='%s:9100',mountpoint='/'}", hostIP)
resp, err := http.Get(prometheusQueryURL)
if err != nil {
return 0, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return 0, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
}
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return 0, err
}
// Parse Prometheus response
var result struct {
Data struct {
Result []struct {
Value []interface{} `json:"value"`
} `json:"result"`
} `json:"data"`
}
if err := json.Unmarshal(body, &result); err != nil {
return 0, err
}
if len(result.Data.Result) == 0 {
return 0, fmt.Errorf("no data found for node %s", nodeName)
}
// Extract disk size (in bytes) from the result
diskSizeBytes, ok := result.Data.Result[0].Value[1].(string)
if !ok {
return 0, fmt.Errorf("invalid data format for node %s", nodeName)
}
// Convert to GB
diskSizeGB := parseDiskSize(diskSizeBytes) / (1024 * 1024 * 1024)
return diskSizeGB, nil
}
// 解析磁盘大小
func parseDiskSize(sizeStr string) float64 {
size, _ := strconv.ParseFloat(sizeStr, 64)
return size
}
|