mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-20 23:46:23 +01:00
f325410c26
- Return meta-labels for the discovered targets via promutils.Labels instead of map[string]string. This improves the speed of generating meta-labels for discovered targets by up to 5x. - Remove memory allocations in hot paths during ScrapeWork generation. The ScrapeWork contains scrape settings for a single discovered target. This improves the service discovery speed by up to 2x.
287 lines
9.7 KiB
Go
287 lines
9.7 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
|
)
|
|
|
|
func TestParsePodListFailure(t *testing.T) {
|
|
f := func(s string) {
|
|
t.Helper()
|
|
r := bytes.NewBufferString(s)
|
|
objectsByKey, _, err := parsePodList(r)
|
|
if err == nil {
|
|
t.Fatalf("expecting non-nil error")
|
|
}
|
|
if len(objectsByKey) != 0 {
|
|
t.Fatalf("unexpected non-empty objectsByKey: %v", objectsByKey)
|
|
}
|
|
}
|
|
f(``)
|
|
f(`[1,23]`)
|
|
f(`{"items":[{"metadata":1}]}`)
|
|
f(`{"items":[{"metadata":{"labels":[1]}}]}`)
|
|
}
|
|
|
|
const testPodsList = `
|
|
{
|
|
"kind": "PodList",
|
|
"apiVersion": "v1",
|
|
"metadata": {
|
|
"selfLink": "/api/v1/pods",
|
|
"resourceVersion": "72425"
|
|
},
|
|
"items": [
|
|
{
|
|
"metadata": {
|
|
"name": "etcd-m01",
|
|
"namespace": "kube-system",
|
|
"selfLink": "/api/v1/namespaces/kube-system/pods/etcd-m01",
|
|
"uid": "9d328156-75d1-411a-bdd0-aeacb53a38de",
|
|
"resourceVersion": "22318",
|
|
"creationTimestamp": "2020-03-16T20:44:30Z",
|
|
"labels": {
|
|
"component": "etcd",
|
|
"tier": "control-plane"
|
|
},
|
|
"annotations": {
|
|
"kubernetes.io/config.hash": "3ec997b76fb6ed3b78da8e0b5676dac4",
|
|
"kubernetes.io/config.mirror": "3ec997b76fb6ed3b78da8e0b5676dac4",
|
|
"kubernetes.io/config.seen": "2020-03-16T20:44:26.538136233Z",
|
|
"kubernetes.io/config.source": "file"
|
|
},
|
|
"ownerReferences": [
|
|
{
|
|
"apiVersion": "v1",
|
|
"kind": "Node",
|
|
"name": "m01",
|
|
"uid": "b48dd901-ead0-476a-b209-d2d908d65109",
|
|
"controller": true
|
|
}
|
|
]
|
|
},
|
|
"spec": {
|
|
"volumes": [
|
|
{
|
|
"name": "etcd-certs",
|
|
"hostPath": {
|
|
"path": "/var/lib/minikube/certs/etcd",
|
|
"type": "DirectoryOrCreate"
|
|
}
|
|
},
|
|
{
|
|
"name": "etcd-data",
|
|
"hostPath": {
|
|
"path": "/var/lib/minikube/etcd",
|
|
"type": "DirectoryOrCreate"
|
|
}
|
|
}
|
|
],
|
|
"containers": [
|
|
{
|
|
"name": "etcd",
|
|
"image": "k8s.gcr.io/etcd:3.4.3-0",
|
|
"command": [
|
|
"etcd",
|
|
"--advertise-client-urls=https://172.17.0.2:2379",
|
|
"--cert-file=/var/lib/minikube/certs/etcd/server.crt",
|
|
"--client-cert-auth=true",
|
|
"--data-dir=/var/lib/minikube/etcd",
|
|
"--initial-advertise-peer-urls=https://172.17.0.2:2380",
|
|
"--initial-cluster=m01=https://172.17.0.2:2380",
|
|
"--key-file=/var/lib/minikube/certs/etcd/server.key",
|
|
"--listen-client-urls=https://127.0.0.1:2379,https://172.17.0.2:2379",
|
|
"--listen-metrics-urls=http://127.0.0.1:2381",
|
|
"--listen-peer-urls=https://172.17.0.2:2380",
|
|
"--name=m01",
|
|
"--peer-cert-file=/var/lib/minikube/certs/etcd/peer.crt",
|
|
"--peer-client-cert-auth=true",
|
|
"--peer-key-file=/var/lib/minikube/certs/etcd/peer.key",
|
|
"--peer-trusted-ca-file=/var/lib/minikube/certs/etcd/ca.crt",
|
|
"--snapshot-count=10000",
|
|
"--trusted-ca-file=/var/lib/minikube/certs/etcd/ca.crt"
|
|
],
|
|
"resources": {
|
|
|
|
},
|
|
"ports": [
|
|
{
|
|
"name": "foobar",
|
|
"containerPort": 1234,
|
|
"protocol": "TCP"
|
|
}
|
|
],
|
|
"volumeMounts": [
|
|
{
|
|
"name": "etcd-data",
|
|
"mountPath": "/var/lib/minikube/etcd"
|
|
},
|
|
{
|
|
"name": "etcd-certs",
|
|
"mountPath": "/var/lib/minikube/certs/etcd"
|
|
}
|
|
],
|
|
"livenessProbe": {
|
|
"httpGet": {
|
|
"path": "/health",
|
|
"port": 2381,
|
|
"host": "127.0.0.1",
|
|
"scheme": "HTTP"
|
|
},
|
|
"initialDelaySeconds": 15,
|
|
"timeoutSeconds": 15,
|
|
"periodSeconds": 10,
|
|
"successThreshold": 1,
|
|
"failureThreshold": 8
|
|
},
|
|
"terminationMessagePath": "/dev/termination-log",
|
|
"terminationMessagePolicy": "File",
|
|
"imagePullPolicy": "IfNotPresent"
|
|
}
|
|
],
|
|
"restartPolicy": "Always",
|
|
"terminationGracePeriodSeconds": 30,
|
|
"dnsPolicy": "ClusterFirst",
|
|
"nodeName": "test-node",
|
|
"hostNetwork": true,
|
|
"securityContext": {
|
|
|
|
},
|
|
"schedulerName": "default-scheduler",
|
|
"tolerations": [
|
|
{
|
|
"operator": "Exists",
|
|
"effect": "NoExecute"
|
|
}
|
|
],
|
|
"priorityClassName": "system-cluster-critical",
|
|
"priority": 2000000000,
|
|
"enableServiceLinks": true
|
|
},
|
|
"status": {
|
|
"phase": "Running",
|
|
"conditions": [
|
|
{
|
|
"type": "Initialized",
|
|
"status": "True",
|
|
"lastProbeTime": null,
|
|
"lastTransitionTime": "2020-03-20T13:30:29Z"
|
|
},
|
|
{
|
|
"type": "Ready",
|
|
"status": "True",
|
|
"lastProbeTime": null,
|
|
"lastTransitionTime": "2020-03-20T13:30:32Z"
|
|
},
|
|
{
|
|
"type": "ContainersReady",
|
|
"status": "True",
|
|
"lastProbeTime": null,
|
|
"lastTransitionTime": "2020-03-20T13:30:32Z"
|
|
},
|
|
{
|
|
"type": "PodScheduled",
|
|
"status": "True",
|
|
"lastProbeTime": null,
|
|
"lastTransitionTime": "2020-03-20T13:30:29Z"
|
|
}
|
|
],
|
|
"hostIP": "172.17.0.2",
|
|
"podIP": "172.17.0.2",
|
|
"podIPs": [
|
|
{
|
|
"ip": "172.17.0.2"
|
|
}
|
|
],
|
|
"startTime": "2020-03-20T13:30:29Z",
|
|
"containerStatuses": [
|
|
{
|
|
"name": "etcd",
|
|
"state": {
|
|
"running": {
|
|
"startedAt": "2020-03-20T13:30:30Z"
|
|
}
|
|
},
|
|
"lastState": {
|
|
"terminated": {
|
|
"exitCode": 0,
|
|
"reason": "Completed",
|
|
"startedAt": "2020-03-17T18:56:24Z",
|
|
"finishedAt": "2020-03-20T13:29:54Z",
|
|
"containerID": "docker://24eea6f192d4598fcc129b5f163a02d1457137f4ec34e8c80c6049a65604cb07"
|
|
}
|
|
},
|
|
"ready": true,
|
|
"restartCount": 2,
|
|
"image": "k8s.gcr.io/etcd:3.4.3-0",
|
|
"imageID": "docker-pullable://k8s.gcr.io/etcd@sha256:4afb99b4690b418ffc2ceb67e1a17376457e441c1f09ab55447f0aaf992fa646",
|
|
"containerID": "docker://a28f0800855008485376c1eece1cf61de97cb7026b9188d138b0d55d92fc2f5c",
|
|
"started": true
|
|
}
|
|
],
|
|
"qosClass": "BestEffort"
|
|
}
|
|
}
|
|
]
|
|
}
|
|
`
|
|
|
|
func TestParsePodListSuccess(t *testing.T) {
|
|
r := bytes.NewBufferString(testPodsList)
|
|
objectsByKey, meta, err := parsePodList(r)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
expectedResourceVersion := "72425"
|
|
if meta.ResourceVersion != expectedResourceVersion {
|
|
t.Fatalf("unexpected resource version; got %s; want %s", meta.ResourceVersion, expectedResourceVersion)
|
|
}
|
|
sortedLabelss := getSortedLabelss(objectsByKey)
|
|
expectedLabelss := []*promutils.Labels{
|
|
promutils.NewLabelsFromMap(map[string]string{
|
|
"__address__": "172.17.0.2:1234",
|
|
|
|
"__meta_kubernetes_namespace": "kube-system",
|
|
"__meta_kubernetes_node_label_node_label": "xyz",
|
|
"__meta_kubernetes_node_labelpresent_node_label": "true",
|
|
"__meta_kubernetes_node_name": "test-node",
|
|
"__meta_kubernetes_pod_name": "etcd-m01",
|
|
"__meta_kubernetes_pod_ip": "172.17.0.2",
|
|
"__meta_kubernetes_pod_container_image": "k8s.gcr.io/etcd:3.4.3-0",
|
|
"__meta_kubernetes_pod_container_name": "etcd",
|
|
"__meta_kubernetes_pod_container_port_name": "foobar",
|
|
"__meta_kubernetes_pod_container_port_number": "1234",
|
|
"__meta_kubernetes_pod_container_port_protocol": "TCP",
|
|
"__meta_kubernetes_pod_ready": "true",
|
|
"__meta_kubernetes_pod_phase": "Running",
|
|
"__meta_kubernetes_pod_node_name": "test-node",
|
|
"__meta_kubernetes_pod_host_ip": "172.17.0.2",
|
|
"__meta_kubernetes_pod_uid": "9d328156-75d1-411a-bdd0-aeacb53a38de",
|
|
"__meta_kubernetes_pod_controller_kind": "Node",
|
|
"__meta_kubernetes_pod_controller_name": "m01",
|
|
"__meta_kubernetes_pod_container_init": "false",
|
|
|
|
"__meta_kubernetes_pod_label_component": "etcd",
|
|
"__meta_kubernetes_pod_label_tier": "control-plane",
|
|
|
|
"__meta_kubernetes_pod_labelpresent_component": "true",
|
|
"__meta_kubernetes_pod_labelpresent_tier": "true",
|
|
|
|
"__meta_kubernetes_pod_annotation_kubernetes_io_config_hash": "3ec997b76fb6ed3b78da8e0b5676dac4",
|
|
"__meta_kubernetes_pod_annotation_kubernetes_io_config_mirror": "3ec997b76fb6ed3b78da8e0b5676dac4",
|
|
"__meta_kubernetes_pod_annotation_kubernetes_io_config_seen": "2020-03-16T20:44:26.538136233Z",
|
|
"__meta_kubernetes_pod_annotation_kubernetes_io_config_source": "file",
|
|
|
|
"__meta_kubernetes_pod_annotationpresent_kubernetes_io_config_hash": "true",
|
|
"__meta_kubernetes_pod_annotationpresent_kubernetes_io_config_mirror": "true",
|
|
"__meta_kubernetes_pod_annotationpresent_kubernetes_io_config_seen": "true",
|
|
"__meta_kubernetes_pod_annotationpresent_kubernetes_io_config_source": "true",
|
|
}),
|
|
}
|
|
if !areEqualLabelss(sortedLabelss, expectedLabelss) {
|
|
t.Fatalf("unexpected labels:\ngot\n%v\nwant\n%v", sortedLabelss, expectedLabelss)
|
|
}
|
|
}
|