#!/bin/env bash
#tidb_monitor_ready.sh
#主集群部署多套监控,主集群两套监控,备集群一套监控
#0为正常,1为异常
TidbHome=$(awk -F: -v un="tidb" '{if($1==un){print $6;exit}}' /etc/passwd 2>/dev/null)
clsName=$(tiup cluster list 2>/dev/null|awk '{if($3~"v" && $4~"tiup"){printf "%s\n",$1;exit}}') 
prom_cnt=$(tiup cluster display ${clsName} 2>/dev/null|awk '/prometheus/{p++}END{printf "%s",p}') 
graf_cnt=$(tiup cluster display ${clsName} 2>/dev/null|awk '/grafana/{p++}END{printf "%s",p}') 
HostName=$(hostname)
#主机名第3位是1的为主集群
if [ "${HostName:2:1}" = "1" ]
then
    HostRole="P"
elif [ "${HostName:2:1}" = "2" ]
then
    HostRole="S"
else
    HostRole="P"
fi

if [ "${HostRole}" = "P" -a "${prom_cnt}" -ge 2 -a "${graf_cnt}" -ge 2 ]
then
    echo "kv:主集群监控节点数目"
    echo "kv:0"
elif [ "${HostRole}" = "P" ]
then
    echo "kv:主集群监控节点数目"
    echo "kv:1"
elif [ "${HostRole}" = "S" -a "${prom_cnt}" -ge 1 -a "${graf_cnt}" -ge 1 ]
then
    echo "kv:备集群监控节点数目"
    echo "kv:0"
elif [ "${HostRole}" = "S" ]
then
    echo "kv:备集群监控节点数目"
    echo "kv:1"
else
    echo "kv:集群监控节点数目"
    echo "kv:1"    
fi