#!/bin/bash
#一鍵部署zookeeper集群服務器腳本
# 判斷/opt目錄下是否有指定安裝包
if [ -e /opt/apache-zookeeper-3.5.7-bin.tar.gz ];then
    continue
else
    echo -e '\033[35m /opt目錄中沒有安裝包[apache-zookeeper-3.5.7-bin.tar.gz]\033[0m'
    exit
fi
# 配置yum本地源
yumm (){
        if ls /etc/yum.repos.d/ | grep repo.bak &> /dev/null
        then
            mv /etc/yum.repos.d/*.repo /etc/yum.repos.d/repo.bak/
        else
            mkdir /etc/yum.repos.d/repo.bak/
            mv /etc/yum.repos.d/*.repo /etc/yum.repos.d/repo.bak/
        fi
        echo -e "[local]\nname=local\nbaseurl=file:///mnt\nenable=1\ngpgcheck=0" > /etc/yum.repos.d/local.repo
        if df -h | grep /dev/sr0 &> /dev/null
        then
            umount /dev/sr0
            mount /dev/cdrom /mnt &> /dev/null
        else
            mount /dev/cdrom /mnt &> /dev/null
            if [ $? -ne 0 ]
            then
                echo "光盤不存在，檢查是否加載鏡像，狀態是否已連接！"
                exit
            fi
        fi
}
yumm
if yum clean all &> /dev/null
then
    if yum makecache &> /dev/null
    then
        echo "yum本地源配置完成！可以正常安裝程序！"
    else
        echo "yum緩存建立失敗！"
    fi
else
    echo "yum緩存清除失敗......"        
fi
read -p "當前服務器的myid為：" num
read -p "輸入要設置的zookeeper服務器1的ip：" ip1
read -p "輸入要設置的zookeeper服務器2的ip：" ip2
read -p "輸入要設置的zookeeper服務器3的ip：" ip3
yum install -y java-1.8.0-openjdk java-1.8.0-openjdk-devel
cd /opt/
tar zxvf apache-zookeeper-3.5.7-bin.tar.gz
mv apache-zookeeper-3.5.7-bin /usr/local/zookeeper-3.5.7 &> /dev/null
cd /usr/local/zookeeper-3.5.7/conf/ 
/usr/bin/cp zoo_sample.cfg zoo.cfg
sed -i '12 s/tmp\/zookeeper/usr\/local\/zookeeper-3.5.7\/data/' /usr/local/zookeeper-3.5.7/conf/zoo.cfg
sed -i '13c dataLogDir=\/usr\/local\/zookeeper-3.5.7\/logs' /usr/local/zookeeper-3.5.7/conf/zoo.cfg
sed -i '15c server.1='$ip1':3188:3288' /usr/local/zookeeper-3.5.7/conf/zoo.cfg
sed -i '16c server.2='$ip2':3188:3288' /usr/local/zookeeper-3.5.7/conf/zoo.cfg
sed -i '17c server.3='$ip3':3188:3288' /usr/local/zookeeper-3.5.7/conf/zoo.cfg
mkdir /usr/local/zookeeper-3.5.7/data &> /dev/null
mkdir /usr/local/zookeeper-3.5.7/logs &> /dev/null
echo $num > /usr/local/zookeeper-3.5.7/data/myid
echo '#!/bin/bash
#chkconfig:2345 20 90
#description: Zookeeper Service Control Script
ZK_HOME='/usr/local/zookeeper-3.5.7'
case $1 in
start)
echo "-----zookeeper啟動-----"
$ZK_HOME/bin/zkServer.sh start
;;
stop)
echo "----zookeeper停止-------"
$ZK_HOME/bin/ zkServer.sh stop
;;
restart)
echo "----zookeeper重啟-------"
$ZK_HOME/bin/zkServer.sh restart
;;
status)
echo  "-----zookeeper狀態------"
$ZK_HOME/bin/zkServer.sh status
;;
*)
echo "Usage: $0 {start|stop|restart|status}"
esac' > /etc/init.d/zookeeper
chmod +x /etc/init.d/zookeeper
chkconfig --add zookeeper
echo "所有服務執行完腳本后執行 service zookeeper start 開啟服務"
echo "所有服務執行完腳本后執行 service zookeeper status 查看狀態"