LivenessProbe for FE, BE and Broker (#22)

* Refactor prometheus deployment to statefulset * Refactor grafana deployment to statefulset * Refactor loki deployment to statefulset * update deploy resources * fixbug: Incorrect loki volume mounting * Add livenessProbe for fe, be, and broker * Refactor Doris monitor K8s app resources (#21) * Refactor prometheus deployment to statefulset * Refactor grafana deployment to statefulset * Refactor loki deployment to statefulset * update deploy resources * Refactor prometheus deployment to statefulset * Refactor grafana deployment to statefulset * Refactor loki deployment to statefulset
linsoss · Nov 25, 2023 · 616387d · 616387d
1 parent 94139d7
commit 616387d
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 0 deletions.
diff --git a/internal/transformer/be_resources.go b/internal/transformer/be_resources.go
@@ -259,6 +259,14 @@ func MakeBeStatefulSet(cr *dapi.DorisCluster, scheme *runtime.Scheme) *appv1.Sta
 			SuccessThreshold: 1,
 			FailureThreshold: 3,
 		},
+		LivenessProbe: &corev1.Probe{
+			ProbeHandler:        util.NewTcpSocketProbeHandler(GetBeHeartbeatServicePort(cr)),
+			InitialDelaySeconds: 20,
+			TimeoutSeconds:      1,
+			PeriodSeconds:       5,
+			SuccessThreshold:    1,
+			FailureThreshold:    5,
+		},
 	}
 	// pod template: init container
 	privileged := true

diff --git a/internal/transformer/broker_resources.go b/internal/transformer/broker_resources.go
@@ -176,6 +176,14 @@ func MakeBrokerStatefulSet(cr *dapi.DorisCluster, scheme *runtime.Scheme) *appv1
 			SuccessThreshold: 1,
 			FailureThreshold: 3,
 		},
+		LivenessProbe: &corev1.Probe{
+			ProbeHandler:        util.NewTcpSocketProbeHandler(GetBrokerIpcPort(cr)),
+			InitialDelaySeconds: 20,
+			TimeoutSeconds:      1,
+			PeriodSeconds:       5,
+			SuccessThreshold:    1,
+			FailureThreshold:    5,
+		},
 	}
 	// pod template: merge additional pod containers configs defined by user
 	mainContainer.Env = append(mainContainer.Env, cr.Spec.Broker.AdditionalEnvs...)

diff --git a/internal/transformer/cn_resources.go b/internal/transformer/cn_resources.go
@@ -236,6 +236,14 @@ func MakeCnStatefulSet(cr *dapi.DorisCluster, scheme *runtime.Scheme) *appv1.Sta
 			SuccessThreshold: 1,
 			FailureThreshold: 3,
 		},
+		LivenessProbe: &corev1.Probe{
+			ProbeHandler:        util.NewTcpSocketProbeHandler(GetCnHeartbeatServicePort(cr)),
+			InitialDelaySeconds: 20,
+			TimeoutSeconds:      1,
+			PeriodSeconds:       5,
+			SuccessThreshold:    1,
+			FailureThreshold:    5,
+		},
 	}
 	// pod template: init container
 	privileged := true

diff --git a/internal/transformer/fe_resources.go b/internal/transformer/fe_resources.go
@@ -279,6 +279,14 @@ func MakeFeStatefulSet(cr *dapi.DorisCluster, scheme *runtime.Scheme) *appv1.Sta
 			SuccessThreshold:    1,
 			FailureThreshold:    3,
 		},
+		LivenessProbe: &corev1.Probe{
+			ProbeHandler:        util.NewTcpSocketProbeHandler(GetFeEditLogPort(cr)),
+			InitialDelaySeconds: 20,
+			TimeoutSeconds:      1,
+			PeriodSeconds:       5,
+			SuccessThreshold:    1,
+			FailureThreshold:    5,
+		},
 	}
 	// pod template: merge additional pod containers configs defined by user
 	mainContainer.Env = append(mainContainer.Env, cr.Spec.FE.AdditionalEnvs...)