scheme

Signed-off-by: Prashanth Dintyala <[email protected]>
NVIDIA · Sep 29, 2023 · fa966e5 · fa966e5
1 parent 2df5cd3
commit fa966e5
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/operator/pkg/resources/target/statefulset.go b/operator/pkg/resources/target/statefulset.go
@@ -131,7 +131,7 @@ func NewTargetSS(ais *aisv1.AIStore) *apiv1.StatefulSet {
 							VolumeMounts:    volumeMounts(ais),
 							Lifecycle:       cmn.NewAISNodeLifecycle(),
 							LivenessProbe:   cmn.NewAISLivenessProbe(),
-							ReadinessProbe:  readinessProbe(ais.Spec.TargetSpec.ServicePort),
+							ReadinessProbe:  readinessProbe(ais.Spec.TargetSpec.ServicePort, ais.Spec.TLSSecretName != nil),
 						},
 					},
 					ServiceAccountName: cmn.ServiceAccountName(ais),
@@ -157,12 +157,18 @@ func volumeMounts(ais *aisv1.AIStore) []corev1.VolumeMount {
 	return vols
 }
 
-func readinessProbe(port intstr.IntOrString) *corev1.Probe {
+func readinessProbe(port intstr.IntOrString, useHTTPS bool) *corev1.Probe {
+	scheme := corev1.URISchemeHTTP
+	if useHTTPS {
+		scheme = corev1.URISchemeHTTPS
+	}
+
 	return &corev1.Probe{
 		ProbeHandler: corev1.ProbeHandler{
 			HTTPGet: &corev1.HTTPGetAction{
-				Path: "/v1/health?readiness=true",
-				Port: port,
+				Path:   "/v1/health?readiness=true",
+				Port:   port,
+				Scheme: scheme,
 			},
 		},
 		InitialDelaySeconds: 15,