From 606bffec8b0ef047cce41ae006a26005cac63bab Mon Sep 17 00:00:00 2001
From: Matthew Fuller <mf4192@att.com>
Date: Fri, 21 May 2021 22:20:43 +0000
Subject: [PATCH] Add liveness and readiness probes to jumphost

Adds TCP liveness and readiness probes for the jumphost
container's sshd service.

Closes: #6
Change-Id: Ifa2e63bbea7773aa5c9154d03df32e4da3b57994
---
 pkg/services/jumphost.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/pkg/services/jumphost.go b/pkg/services/jumphost.go
index 3607bc6..f44a1ea 100644
--- a/pkg/services/jumphost.go
+++ b/pkg/services/jumphost.go
@@ -29,6 +29,7 @@ import (
 	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
+	"k8s.io/apimachinery/pkg/util/intstr"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
 	airshipv1 "sipcluster/pkg/api/v1"
@@ -202,6 +203,24 @@ func (jh jumpHost) generateDeployment(instance string, labels map[string]string,
 									SubPath:   authorizedKeysFile,
 								},
 							},
+							LivenessProbe: &corev1.Probe{
+								Handler: corev1.Handler{
+									TCPSocket: &corev1.TCPSocketAction{
+										Port: intstr.FromString("ssh"),
+									},
+								},
+								InitialDelaySeconds: 15,
+								PeriodSeconds:       10,
+							},
+							ReadinessProbe: &corev1.Probe{
+								Handler: corev1.Handler{
+									TCPSocket: &corev1.TCPSocketAction{
+										Port: intstr.FromString("ssh"),
+									},
+								},
+								InitialDelaySeconds: 5,
+								PeriodSeconds:       20,
+							},
 						},
 					},
 					SecurityContext: &corev1.PodSecurityContext{