feat: set startup probe to prevent from unnecessary restarts (#490)

Signed-off-by: arkbriar <arkbriar@gmail.com>
risingwavelabs · Aug 7, 2023 · 30bd7f8 · 30bd7f8
1 parent 10771c8
commit 30bd7f8
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pkg/factory/risingwave_object_factory.go b/pkg/factory/risingwave_object_factory.go
@@ -1005,7 +1005,17 @@ func basicSetupRisingWaveContainer(container *corev1.Container, component *risin
 	}
 
 	// Set the default probes.
-	container.StartupProbe = nil
+	container.StartupProbe = &corev1.Probe{
+		InitialDelaySeconds: 5,
+		PeriodSeconds:       5,
+		TimeoutSeconds:      5,
+		FailureThreshold:    12,
+		ProbeHandler: corev1.ProbeHandler{
+			TCPSocket: &corev1.TCPSocketAction{
+				Port: intstr.FromString(consts.PortService),
+			},
+		},
+	}
 	container.LivenessProbe = &corev1.Probe{
 		InitialDelaySeconds: 2,
 		PeriodSeconds:       10,