qwt

#!/bin/bash

set -o errexit
set -o pipefail
set -o nounset

export KAFKA_OPTS="-Djava.security.auth.login.config=/etc/kafka/kafka_server_jaas.conf"

#Global Variables
topic_name="test-k1"
cert="/mnt/cifsConfluentPlatform/ssl_certs/clients/ITERGO_ANSIBLE"
kb="/etc/systemd/system/multi-user.target.wants/confluent-kafka.service"
zk="/etc/systemd/system/multi-user.target.wants/confluent-zookeeper.service"
cs="/etc/systemd/system/multi-user.target.wants/confluent-schema-registry.service"
#zookeeper_server="{% for host in groups['zookeeper'] %}{% if loop.index > 1%},{% endif %}{{ host }}:{{zookeeper.clientPort}}{% endfor %}"

_check_zookeeper() {
    resultCreateZnode=$(zookeeper-shell $hostname:2181 create /readinessCheck OK 2>&1 | grep -E "Node already exists: /readinessCheck|Created /readinessCheck")
    [[ ! ${resultCreateZnode-} ]] && { echo "ERROR Something wrong with ZooKeeper... exiting"; exit; }
    resultCheckZnode=$(zookeeper-shell $hostname:2181 get /readinessCheck 2>&1 | grep "^OK$")
    [[ ! ${resultCheckZnode-} ]] && { echo "ERROR Something wrong with ZooKeeper... exiting"; exit; }
    resultDeleteZnode=$(zookeeper-shell $hostname:2181 delete /readinessCheck)
}

_check_schema() {

    #Listing available schemas

    schema="$(curl -k --cert $cert/client.ITERGO_ANSIBLE.cer --key $cert/client.ITERGO_ANSIBLE.kexstore.key -X POST -H "Content-Type: application/vnd.schemaregistry.v1+json" --data '{"schema": "{\"type\": \"string\"}"}'  https://$HOSTNAME:8081/subjects/ITERGO_A_Test-key/versions)"
    schema="$(curl -k --cert $cert/client.ITERGO_ANSIBLE.cer --key $cert/client.ITERGO_ANSIBLE.kexstore.key -X GET https://$HOSTNAME:8081/subjects/ITERGO_A_Test-key/versions)"
    if [[ $schema != "[1]"  ]]; then
        echo "schema not found"
        exit 1
    fi
    schema="$(curl -k --cert $cert/client.ITERGO_ANSIBLE.cer --key $cert/client.ITERGO_ANSIBLE.kexstore.key -X DELETE https://$HOSTNAME:8081/subjects/ITERGO_A_Test-key/versions/1)"
    if [[ $schema -ne 1 ]]; then
        echo "Failed to delete schema"
        exit 1
    fi
}

_check_broker() {

    partition="$(curl -s http://localhost:{{kafkabroker.port_jmx_http}} | grep "^kafka_server_replicamanager_underreplicatedpartitions" | awk '{print $2}')"
    while [ $partition != "0.0" ]; do
        sleep 1;
    done
    if [[ $? -ne 0 ]]; then
        echo -e "check failed"
        exit 1
    fi
}

if [ -f $kb ]; then
    _check_broker()
elif [ -f $zk ]; then
    _check_zookeeper()
elif [ -f $cs ]; then
    _check_schema()
else
    echo "Script Fails"
    exit 1;
fi

echo -e "Success"