100 lines
2.5 KiB
Go
100 lines
2.5 KiB
Go
|
package kube_inventory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
corev1 "k8s.io/api/core/v1"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf"
|
||
|
|
)
|
||
|
|
|
||
|
|
func collectNodes(ctx context.Context, acc telegraf.Accumulator, ki *KubernetesInventory) {
|
||
|
|
list, err := ki.client.getNodes(ctx, ki.NodeName)
|
||
|
|
if err != nil {
|
||
|
|
acc.AddError(err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
gatherNodeCount(len(list.Items), acc)
|
||
|
|
|
||
|
|
for i := range list.Items {
|
||
|
|
ki.gatherNode(&list.Items[i], acc)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func gatherNodeCount(count int, acc telegraf.Accumulator) {
|
||
|
|
fields := map[string]interface{}{"node_count": count}
|
||
|
|
tags := make(map[string]string)
|
||
|
|
|
||
|
|
acc.AddFields(nodeMeasurement, fields, tags)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ki *KubernetesInventory) gatherNode(n *corev1.Node, acc telegraf.Accumulator) {
|
||
|
|
fields := make(map[string]interface{}, len(n.Status.Capacity)+len(n.Status.Allocatable)+1)
|
||
|
|
tags := map[string]string{
|
||
|
|
"node_name": n.Name,
|
||
|
|
"cluster_namespace": n.Annotations["cluster.x-k8s.io/cluster-namespace"],
|
||
|
|
"version": n.Status.NodeInfo.KubeletVersion,
|
||
|
|
}
|
||
|
|
|
||
|
|
for resourceName, val := range n.Status.Capacity {
|
||
|
|
switch resourceName {
|
||
|
|
case "cpu":
|
||
|
|
fields["capacity_cpu_cores"] = ki.convertQuantity(val.String(), 1)
|
||
|
|
fields["capacity_millicpu_cores"] = ki.convertQuantity(val.String(), 1000)
|
||
|
|
case "memory":
|
||
|
|
fields["capacity_memory_bytes"] = ki.convertQuantity(val.String(), 1)
|
||
|
|
case "pods":
|
||
|
|
fields["capacity_pods"] = atoi(val.String())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for resourceName, val := range n.Status.Allocatable {
|
||
|
|
switch resourceName {
|
||
|
|
case "cpu":
|
||
|
|
fields["allocatable_cpu_cores"] = ki.convertQuantity(val.String(), 1)
|
||
|
|
fields["allocatable_millicpu_cores"] = ki.convertQuantity(val.String(), 1000)
|
||
|
|
case "memory":
|
||
|
|
fields["allocatable_memory_bytes"] = ki.convertQuantity(val.String(), 1)
|
||
|
|
case "pods":
|
||
|
|
fields["allocatable_pods"] = atoi(val.String())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, val := range n.Status.Conditions {
|
||
|
|
conditiontags := map[string]string{
|
||
|
|
"status": string(val.Status),
|
||
|
|
"condition": string(val.Type),
|
||
|
|
}
|
||
|
|
for k, v := range tags {
|
||
|
|
conditiontags[k] = v
|
||
|
|
}
|
||
|
|
running := 0
|
||
|
|
nodeready := 0
|
||
|
|
if val.Status == "True" {
|
||
|
|
if val.Type == "Ready" {
|
||
|
|
nodeready = 1
|
||
|
|
}
|
||
|
|
running = 1
|
||
|
|
} else if val.Status == "Unknown" {
|
||
|
|
if val.Type == "Ready" {
|
||
|
|
nodeready = 0
|
||
|
|
}
|
||
|
|
running = 2
|
||
|
|
}
|
||
|
|
conditionfields := map[string]interface{}{
|
||
|
|
"status_condition": running,
|
||
|
|
"ready": nodeready,
|
||
|
|
}
|
||
|
|
acc.AddFields(nodeMeasurement, conditionfields, conditiontags)
|
||
|
|
}
|
||
|
|
|
||
|
|
unschedulable := 0
|
||
|
|
if n.Spec.Unschedulable {
|
||
|
|
unschedulable = 1
|
||
|
|
}
|
||
|
|
fields["spec_unschedulable"] = unschedulable
|
||
|
|
|
||
|
|
acc.AddFields(nodeMeasurement, fields, tags)
|
||
|
|
}
|