#!/bin/bash
######################################################################
#### function: install zookeeper
####author:danial_chen
####create date:2019-11-07
####modify date:2019-11-07
######################################################################
#useage: sh installzk.sh "zk_ip1 zk_ip2 zk_ip3" zkserver_id "zookeeper_username" "zookeeper_password"
#eg. sh installzk.sh "192.168.1.11 192.168.1.12 191.168.1.13" 2 "zookeeper" "****"
if [ -z "$1" ] || [ -z "$2" ]; then
  echo "Useage : sh $0 "zk_ip1 zk_ip2 zk_ip3" zkserver_id"
  echo "Default: sh $0 "192.168.1.11 192.168.1.12 191.168.1.13" 2"
  exit 1
fi

set +e
## ENV ##
source /etc/profile
set -e
pd=$PWD
zookeeper_ip=($1)
zkserver_id=$2
zookeeper_username=($3)
zookeeper_password=($4)
if [ -z "$zookeeper_password" ]; then
  echo "zookeeper_password is empty"
  exit 10
fi
environment_type=$5
jvm_mem="2048"
if [ "$environment_type" != "prod" ]; then
  jvm_mem="256"
fi
zookeeper_version="zookeeper-3.5.9"
#zookeeper_version="apache-zookeeper-3.9.1-bin"
export dataDir=${dataDir:-$pd/data/zk_2181/data}
export dataLogDir=${dataLogDir:-$pd/data/zk_2181/logs}

## install zookeeper ##
if [ -z "$JAVA_HOME" ]; then
  echo "jdk not be installed,please install jdk first!"
  exit 2
fi
tar -zxf $pd/$zookeeper_version.tar.gz
chmod -R 755 $pd/$zookeeper_version
if [ $? -ne 0 ]; then
  echo "tar $zookeeper_version.tar.gz failed!"
fi
mkdir -p $dataDir
mkdir -p $dataLogDir

## Modify config file ##
#cp -rf $pd/$zookeeper_version/conf/zoo_sample.cfg $pd/$zookeeper_version/conf/zoo.cfg
sed -i "/^tickTime=/c tickTime=5000" $pd/$zookeeper_version/conf/zoo.cfg
sed -i "/^dataDir=/c dataDir=$dataDir" $pd/$zookeeper_version/conf/zoo.cfg
sed -i "/^dataLogDir=/c dataLogDir=$dataLogDir" $pd/$zookeeper_version/conf/zoo.cfg
echo ' ' >> $pd/$zookeeper_version/conf/zoo.cfg
echo "admin.serverPort=2388" >> $pd/$zookeeper_version/conf/zoo.cfg
echo 'maxClientCnxns=1200' >> $pd/$zookeeper_version/conf/zoo.cfg
echo "admin.enableServer=false" >> $pd/$zookeeper_version/conf/zoo.cfg
echo "autopurge.snapRetainCount=3" >> $pd/$zookeeper_version/conf/zoo.cfg
echo "autopurge.purgeInterval=1" >> $pd/$zookeeper_version/conf/zoo.cfg
echo "4lw.commands.whitelist=*" >> $pd/$zookeeper_version/conf/zoo.cfg

if [ ${#zookeeper_ip[@]} -gt 1 ]; then
  echo $zkserver_id > $dataDir/myid
  for ((i = ${#zookeeper_ip[@]}; i > 0; i--)); do
    b=$((${i} - 1))
    sed -i "/^clientPort/a server.${i}=${zookeeper_ip[$b]}:2888:3888" $pd/$zookeeper_version/conf/zoo.cfg
  done
fi

## Modify zkEnv.sh
sed -i "s/ZK_SERVER_HEAP:-1000/ZK_SERVER_HEAP:-${jvm_mem}/g" $pd/$zookeeper_version/bin/zkEnv.sh

# 先删除原来的
sed -i '/ZK_HOME/d' /etc/profile
## Add system environment variable ##
export ZK_HOME=$pd/$zookeeper_version
if ! grep -q "ZK_HOME=$pd/$zookeeper_version" /etc/profile; then
  echo "export ZK_HOME=$pd/$zookeeper_version" >> /etc/profile
  echo "PATH=\$ZK_HOME/bin:\$PATH:\$HOME/bin" >> /etc/profile
fi
set +e
source /etc/profile
set -e

## Add zookeeper user ##
start_zk_user="zookeeper"
groupadd -f $start_zk_user
IF_START_USER=$(sed -n '/^'${start_zk_user}':/p' /etc/passwd)
if [ -z "${IF_START_USER}" ]; then
  useradd -g $start_zk_user -M -s /sbin/nologin $start_zk_user
fi
chown -R $start_zk_user. $pd

## register systemd ##
systemctl stop zookeeper.service || echo
cat > /usr/lib/systemd/system/zookeeper.service << EOF
[Unit]
Description=zookeeper
After=syslog.target network.target

[Service]
Type=forking
User=$start_zk_user
Group=$start_zk_user
Environment=ZOO_LOG_DIR=$dataLogDir
Environment=ZOO_LOG4J_PROP=ERROR,ROLLINGFILE
Environment="PATH=$JAVA_HOME/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/root/bin:/root/bin"
ExecStart=$pd/$zookeeper_version/bin/zkServer.sh start
ExecStop=$pd/$zookeeper_version/bin/zkServer.sh stop
Restart=always
RestartSec=5
StartLimitInterval=0
StartLimitBurst=5

[Install]
WantedBy=multi-user.target
EOF

## start server ##
systemctl daemon-reload
systemctl enable zookeeper
systemctl restart zookeeper

echo "install zookeeper success!"