Merge pull request #257 from jichenjc/bug/237-1

rootfs · web-flow · commit 069fced2252e · 2022-09-29T12:06:31.000-04:00
add healthy for exporter
diff --git a/cmd/exporter.go b/cmd/exporter.go
@@ -42,6 +42,14 @@ var (
 	enabledEBPFCgroupID = flag.Bool("enable-cgroup-id", true, "whether enable eBPF to collect cgroup id (must have kernel version >= 4.18 and cGroup v2)")
 )
 
+func healthProbe(w http.ResponseWriter, req *http.Request) {
+	w.WriteHeader(http.StatusOK)
+	_, err := w.Write([]byte(`ok`))
+	if err != nil {
+		klog.Fatalf("%s", fmt.Sprintf("failed to write response: %v", err))
+	}
+}
+
 func main() {
 	klog.InitFlags(nil)
 	flag.Parse()
@@ -80,6 +88,7 @@ func main() {
 	}
 
 	http.Handle(*metricsPath, promhttp.Handler())
+	http.HandleFunc("/healthz", healthProbe)
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		_, err = w.Write([]byte(`<html>
 			<head><title>Energy Stats Exporter</title></head>
diff --git a/manifests/kubernetes/deployment.yaml b/manifests/kubernetes/deployment.yaml
@@ -78,6 +78,16 @@ spec:
         - containerPort: 9102
           hostPort: 9102
           name: http
+        livenessProbe:
+          failureThreshold: 5
+          httpGet:
+            path: /healthz
+            port: 9102
+            scheme: HTTP
+          initialDelaySeconds: 10
+          periodSeconds: 60
+          successThreshold: 1
+          timeoutSeconds: 10
         volumeMounts:
         - mountPath: /lib/modules
           name: lib-modules
