(refactor)kafka: add support for liveness containers to be specified explicitly (#948)

Signed-off-by: ksatchit <ksatchit@mayadata.io>

experiments/kafka/kafka-broker-disk-failure/kafka-broker-disk-failure-ansible-logic.yml

@@ -15,6 +15,7 @@
     kafka_instance: "{{ lookup('env','KAFKA_INSTANCE_NAME') }}"
     kafka_broker: "{{ lookup('env','KAFKA_BROKER') }}"
     kafka_stream: "{{ lookup('env','KAFKA_LIVENESS_STREAM') }}"
+    kafka_liveness_image: "{{ lookup('env','KAFKA_LIVENESS_IMAGE') }}" 
     kafka_consumer_timeout: "{{ lookup('env','KAFKA_CONSUMER_TIMEOUT') }}" 
     kafka_service: "{{ lookup('env','KAFKA_SERVICE') }}"
     kafka_port: "{{ lookup('env','KAFKA_PORT') }}" 

experiments/kafka/kafka-broker-disk-failure/kafka-broker-disk-failure-k8s-job.yml

@@ -28,6 +28,9 @@ spec:
           - name: KAFKA_LIVENESS_STREAM
             value: 'enabled'
 
+          - name: KAFKA_LIVENESS_IMAGE
+            value: 'litmuschaos/kafka-client:ci'
+
           # in milliseconds
           - name: KAFKA_CONSUMER_TIMEOUT
             value: '30000'

experiments/kafka/kafka-broker-pod-failure/kafka-broker-pod-failure-ansible-logic.yml

@@ -14,6 +14,7 @@
     kafka_kind: "{{ lookup('env','KAFKA_KIND') }}"
     kafka_broker: "{{ lookup('env','KAFKA_BROKER') }}"
     kafka_stream: "{{ lookup('env','KAFKA_LIVENESS_STREAM') }}" 
+    kafka_liveness_image: "{{ lookup('env','KAFKA_LIVENESS_IMAGE') }}" 
     kafka_consumer_timeout: "{{ lookup('env','KAFKA_CONSUMER_TIMEOUT') }}" 
     kafka_service: "{{ lookup('env','KAFKA_SERVICE') }}"
     kafka_port: "{{ lookup('env','KAFKA_PORT') }}" 

experiments/kafka/kafka-broker-pod-failure/kafka-broker-pod-failure-k8s-job.yml

@@ -27,6 +27,9 @@ spec:
           - name: KAFKA_LIVENESS_STREAM
             value: 'enabled'
 
+          - name: KAFKA_LIVENESS_IMAGE
+            value: 'litmuschaos/kafka-client:ci'
+
           # in milliseconds
           - name: KAFKA_CONSUMER_TIMEOUT
             value: '30000'

utils/apps/kafka/kafka_liveness.j2

@@ -9,7 +9,7 @@ spec:
   restartPolicy: Never
   initContainers:
   - name: kafka-topic-creator
-    image: litmuschaos/kafka-client:ci
+    image: {{ kafka_liveness_image }}
     imagePullPolicy: Always
     env: 
       - name: TOPIC_NAME
@@ -28,7 +28,7 @@ spec:
       - "./topic.sh"
   containers:
   - name: kafka-producer
-    image: litmuschaos/kafka-client:ci
+    image: {{ kafka_liveness_image }}
     imagePullPolicy: Always
     env:
       - name: TOPIC_NAME
@@ -42,7 +42,7 @@ spec:
       - -c
       - "stdbuf -oL ./producer.sh"
   - name: kafka-consumer
-    image: litmuschaos/kafka-client:ci
+    image: {{ kafka_liveness_image }}
     imagePullPolicy: Always
     env:
       - name: KAFKA_CONSUMER_TIMEOUT