From ba71f1ded49aab6ff6aaa1f2b0dd8bdaffcffa16 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B0=A2=E9=87=91=E8=99=8E?= <1050780355@qq.com>
Date: Mon, 27 Nov 2023 15:57:47 +0800
Subject: [PATCH] add kafka monitor (#4349)

---
 service/database/api/req.go     | 8 ++++++++
 service/database/request/req.go | 2 ++
 2 files changed, 10 insertions(+)

diff --git a/service/database/api/req.go b/service/database/api/req.go
index cf696b8717b..19a07b23e62 100644
--- a/service/database/api/req.go
+++ b/service/database/api/req.go
@@ -118,4 +118,12 @@ var (
 		"minio_bucket_traffic_received_bytes": "sum(minio_bucket_traffic_received_bytes{bucket=\"@\"}) by (bucket, instance, job, namespace)",
 		"minio_bucket_traffic_sent_bytes":     "sum(minio_bucket_traffic_sent_bytes{bucket=\"@\"}) by (bucket, instance, job, namespace)",
 	}
+
+	Kafka = map[string]string{
+		"cpu":           "round(max by (pod) (rate(container_cpu_usage_seconds_total{namespace=~\"#\",pod=~\"@-kafka-\\\\d\" ,container=\"kafka\"}[5m])) / on (pod) (max by (pod) (container_spec_cpu_quota{namespace=~\"#\", pod=~\"@-kafka-\\\\d\",container=\"kafka\"} / 100000)) * 100,0.01)",
+		"memory":        "round(max by (pod)(container_memory_usage_bytes{namespace=~\"#\",pod=~\"@-kafka-\\\\d\",container=\"kafka\" })/ on (pod) (max by (pod) (container_spec_memory_limit_bytes{namespace=~\"#\", pod=~\"@-kafka-\\\\d\",container=\"kafka\"})) * 100,0.01)",
+		"disk_capacity": "(max by (persistentvolumeclaim,namespace) (kubelet_volume_stats_capacity_bytes {namespace=~\"#\", persistentvolumeclaim=~\"data-@-kafka-\\\\d\"}))",
+		"disk":          "round((max by (persistentvolumeclaim,namespace) (kubelet_volume_stats_used_bytes {namespace=~\"#\", persistentvolumeclaim=~\"data-@-kafka-\\\\d\"})) / (max by (persistentvolumeclaim,namespace) (kubelet_volume_stats_capacity_bytes {namespace=~\"#\", persistentvolumeclaim=~\"data-@-kafka-\\\\d\"})) * 100, 0.01)",
+		"disk_used":     "(max by (persistentvolumeclaim,namespace) (kubelet_volume_stats_used_bytes {namespace=~\"#\", persistentvolumeclaim=~\"data-@-kafka-\\\\d\"}))",
+	}
 )
diff --git a/service/database/request/req.go b/service/database/request/req.go
index e01d1148ac0..1f1a90c0c4d 100644
--- a/service/database/request/req.go
+++ b/service/database/request/req.go
@@ -77,6 +77,8 @@ func GetQuery(query *api.PromRequest) (string, error) {
 		result = api.Pgsql[query.Query]
 	case "minio":
 		result = api.Minio[query.Query]
+	case "kafka":
+		result = api.Kafka[query.Query]
 	default:
 		fmt.Println(query.Type)
 	}