Add Zookeeper readiness and liveness check #3

kafka-inmemory/resources/openshift-template.yaml

@@ -138,6 +138,18 @@ objects:
           volumeMounts:
           - mountPath: /var/lib/zookeeper
             name: zookeeper-storage
+          livenessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
+          readinessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
         dnsPolicy: ClusterFirst
         restartPolicy: Always
         volumes:

kafka-inmemory/resources/zookeeper.yaml

@@ -22,6 +22,18 @@ spec:
           volumeMounts:
             - name: zookeeper-storage
               mountPath: /var/lib/zookeeper
+          livenessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
+          readinessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
       volumes:
         - name: zookeeper-storage
           emptyDir: {}

kafka-statefulsets/resources/openshift-template.yaml

@@ -158,6 +158,18 @@ objects:
           volumeMounts:
           - mountPath: /var/lib/zookeeper
             name: zookeeper-storage
+          livenessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
+          readinessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
         dnsPolicy: ClusterFirst
         restartPolicy: Always
     volumeClaimTemplates:

kafka-statefulsets/resources/zookeeper.yaml

@@ -29,6 +29,18 @@ spec:
           volumeMounts:
             - name: zookeeper-storage
               mountPath: /var/lib/zookeeper
+          livenessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
+          readinessProbe:
+            exec:
+              command:
+              - "/opt/zookeeper/zookeeper_healthcheck.sh"
+            initialDelaySeconds: 15
+            timeoutSeconds: 5
   volumeClaimTemplates:
     - metadata:
         name: zookeeper-storage

zookeeper/Dockerfile

@@ -1,6 +1,6 @@
 FROM fedora:26
 
-RUN dnf -y install java-1.8.0-openjdk-headless gettext hostname && dnf clean all -y
+RUN dnf -y install java-1.8.0-openjdk-headless gettext hostname nmap-ncat && dnf clean all -y
 
 # set Kafka home folder
 ENV ZOOKEEPER_HOME=/opt/zookeeper
@@ -32,7 +32,7 @@ EXPOSE 2181 2888 3888
 # copy template configuration files
 COPY ./config/zookeeper.properties.template $ZOOKEEPER_HOME/config/
 # copy scripts for starting Kafka and Zookeeper
-COPY ./scripts/zookeeper_run.sh $ZOOKEEPER_HOME
+COPY ./scripts/ $ZOOKEEPER_HOME
 
 USER zookeeper:zookeeper
 

zookeeper/scripts/zookeeper_healthcheck.sh

@@ -0,0 +1,8 @@
+#!/bin/bash
+
+OK=$(echo ruok | nc 127.0.0.1 2181)
+if [ "$OK" == "imok" ]; then
+    exit 0
+else
+    exit 1
+fi