shell脚本部署redis以及redis主从复制和redis-cluster集群

# 关于脚本：
# 使用root用户执行此脚本，提前关闭selinux；
# 执行脚本之前，hostsIP内的IP修改成自己的机器IP；
# hostsIp内的IP数量如果有增加或者减少，for循环的变量也需要增减；
# sshCopy内的root密码改成自己的root密码；
# installRedis内配置了redis的密码，自行修改

# 关于环境：
# 此脚本部署的是redis-6.0.5版本
# 脚本中的命令需要用到net-tools软件，如果没有网络，将ifconfig注释，打开ip a这个命令的变量即可；
# 因为是虚拟机测试，所以redis-cluster使用的是3节点6实例，具体环境请根据自身需求修改，部署redis也是3主机的形式，如果是单节点，需要修改hostsIP和sshCopy内的内容（ps：单节点没测试过，不确定可用性，可以直接执行installRedis内的命令）；
# 生产环境，测试完之后再使用，如果生产环境没有网络，需要提前准备好redis的tar包；
# 如果是自己准备的redis tar包，需要修改installRedis内的脚本，注释掉wget字段，打开scp字段，注意自己路径和tar包名字；

# 关于主从复制：
# 如果需要使用redis 5以下版本的主从复制，需要修改redisBGSAVE内的replicaof，5.0版本以下的，需要改成slaveof

# 脚本执行方式：sh redis.sh install （编译安装redis）
             # sh redis.sh bgsave (部署redis-主从复制)
             # sh redis.sh cluster (部署redis-cluster)

#!/bin/env bash
# from:半癫

set -euxo pipefail
redisHome=/opt/redis-6.0.5
redisClusterHome=/opt/redis-cluster

# 关闭防火墙，检查selinux
stopFirewalld (){
systemctl disable firewalld --now
sedisabled=$(sestatus | awk '{print $NF}')
if [ $sedisabled = disabled ]
then
    printf "\e[1;32m===========================selinux is disabled==================================\e[0m\n"
else
    printf "\e[1;32m===========================请关闭selinux后再操作================================\e[0m\n"
exit 1
fi
}

# 安装net-tools，ifconfig命令需要用到，已有的情况下，不会再次yum
netTools(){
tools=$(rpm -qa net-tools | wc -l)
if [ $tools -gt 0 ]
then
    printf "\e[1;32m===========================net-tools 已安装=====================================\e[0m\n"
else
    printf "\e[1;32m===========================net-tools 未安装，开始安装===========================\e[0m\n"
    yum -y install net-tools
    if [ $? -ne 0 ]
    then
        printf "\e[1;32m===========================net-tools 安装失败===================================\e[0m\n"
    	exit 2
    else
        printf "\e[1;32m===========================net-tools 安装成功===================================\e[0m\n"
    fi
fi
}

# 检查网络是否通
checkNetwork(){
ping -c1 www.baidu.com > /dev/null 2>&1
if [ $? -eq 0 ]
then
    printf "\e[1;32m===========================网络正常=============================================\e[0m\n"
else
    printf "\e[1;32m===========================请检查网络===========================================\e[0m\n"
    exit 3
fi
}

checkNetwork
stopFirewalld
netTools

# 配置hosts
hostsIp(){
ip=$(ifconfig eth0 | grep netmask | awk '{print $2}')
# ip=$(ifconfig eth0 | grep netmask | awk '{print $2}')
ip1=192.168.10.134
ip2=192.168.10.135
cat >> /etc/hosts <<eof
$ip redis01
$ip1 redis02
$ip2 redis03
eof
}

# ssh配置免密
sshCopy(){
yum -y install sshpass > /dev/null 2>&1
ssh-keygen -t rsa -P '' -f ~/.ssh/id_rsa
for cluster in redis01 redis02 redis03
do
  sshpass -p '123.com' ssh-copy-id  $cluster  -o StrictHostKeyChecking=no
  scp /etc/hosts $cluster:/etc/hosts
  if [ $? -eq 0 ];then
      printf "\e[1;32m===========================$cluster 秘钥copy完成================================\e[0m\n"
  else
      printf "\e[1;32m===========================$cluster 秘钥copy失败================================\e[0m\n"
	  exit 4
  fi
done
}

# tar包编译安装redis，如果没有网络的情况下，可以提前下载tar包，注释掉wget这一行，关闭scp这一行的注释即可
installRedis(){
if [ $(find / -iname 'redis-server' | wc -l) -gt 0 ]
then
    echo 'redis 已安装'
else
    echo 'redis 未安装，开始安装'
	for k in redis01 redis02 redis03
	do
        ssh root@$k "yum -y install gcc glibc glibc-kernheaders  glibc-common glibc-devel make"
        ssh root@$k "yum -y install centos-release-scl"
        ssh root@$k "yum -y install devtoolset-9-gcc devtoolset-9-gcc-c++ devtoolset-9-binutils"
        ssh root@$k "echo "source /opt/rh/devtoolset-9/enable" >> /etc/profile.d/gcc9.sh && source /etc/profile.d/gcc9.sh"
        ssh root@$k "wget -O /opt/redis.tar.gz http://download.redis.io/releases/redis-6.0.5.tar.gz"
	# scp redis-6.0.5.tar.gz root@$k:/opt/
        ssh root@$k "cd /opt/ && tar xf redis.tar.gz && cd redis-6.0.5 && make && make install"
        ssh root@$k "sed -i 's/bind.*/bind 0.0.0.0/g' $redisHome/redis.conf"
	    ssh root@$k "sed -i 's/protected-mode yes/protected-mode no/g' $redisHome/redis.conf"
        ssh root@$k "sed -i 's/daemonize no/daemonize yes/g' $redisHome/redis.conf"
        ssh root@$k "sed -i 's/# requirepass.*/requirepass redis/g' $redisHome/redis.conf"
        ssh root@$k "$redisHome/src/redis-server $redisHome/redis.conf"
        ssh root@$k "ss -nltp | grep 6379"
        if [ $? -eq 0 ]
        then
            printf "\e[1;32m===========================$k 启动成功=======================================\e[0m\n"
        else
            printf "\e[1;32m===========================$k 启动失败=======================================\e[0m\n"
        	exit 5
        fi
	done
fi
}

# redis主从复制
redisBGSAVE(){
ip=$(ifconfig eth0 | grep netmask | awk '{print $2}')
# ip=$(ip a | grep global | awk '{print $2}' | awk -F '/' '{print $1}')
egrep -v "^$|#" $redisHome/redis.conf > $redisHome/redis.conf.bak 
sed -i "s/# replicaof.*/replicaof $ip 6379/g" $redisHome/redis.conf.bak
sed -i "s/# masterauth.*/masterauth redis/g" $redisHome/redis.conf.bak
$redisHome/src/redis-server $redisHome/redis.conf
    if [ $? -eq 0 ]
    then
        printf "\e[1;32m===========================redis01 启动成功=====================================\e[0m\n"
    else
        printf "\e[1;32m===========================redis01 启动失败=====================================\e[0m\n"
        exit 6
    fi

for m in redis02 redis03
do
    scp $redisHome/redis.conf.bak root@$m:$redisHome/redis.conf
    ssh root@$m "$redisHome/src/redis-server $redisHome/redis.conf"
    ssh root@$m "ss -nltp | grep 6379"
    if [ $? -eq 0 ]
    then
        printf "\e[1;32m===========================$m 启动成功=====================================     \e[0m\n"
    else
        printf "\e[1;32m===========================$m 启动失败=====================================     \e[0m\n"
    	exit 7
    fi
done
}

# redis-cluster集群
redisCluster(){
node1=6001
node2=6002
ulimit=$(ulimit -a | grep 'open files' | awk '{print $NF}')

if [ $ulimit -le 10032 ]
then
    ulimit -n 10032
else
    printf "\e[1;32m===========================开始部署集群====================================     \e[0m\n"
fi

mkdir -p $redisClusterHome/{$node1,$node2}/{conf,data,log}
egrep -v "^$|#" $redisHome/redis.conf > $redisClusterHome/$node1/conf/redis-$node1.conf
sed -i "s/port.*/port $node1/g" $redisClusterHome/$node1/conf/redis-$node1.conf
sed -i "s#dir.*#dir $redisClusterHome/$node1/data#g" $redisClusterHome/$node1/conf/redis-$node1.conf
sed -i "s#pidfile.*#pidfile $redisClusterHome/$node1/#g" $redisClusterHome/$node1/conf/redis-$node1.conf
sed -i "s#logfile.*#logfile $redisClusterHome/$node1/log/redis.log#g" $redisClusterHome/$node1/conf/redis-$node1.conf
cat >> $redisClusterHome/$node1/conf/redis-$node1.conf <<EOF
cluster-enabled yes
cluster-config-file $redisClusterHome/$node1/conf/cluster-$node1.conf
cluster-node-timeout 5000
EOF
sed "s/$node1/$node2/g" $redisClusterHome/$node1/conf/redis-$node1.conf > $redisClusterHome/$node2/conf/redis-$node2.conf
scp -r $redisClusterHome root@redis02:/opt/
scp -r $redisClusterHome root@redis03:/opt/
for n in redis01 redis02 redis03
do
    for node in $node1 $node2
    do
    ssh root@$n "$redisHome/src/redis-server $redisClusterHome/$node/conf/redis-$node.conf"
    ssh root@$n "ss -nltp | grep $node"
    if [ $? -eq 0 ]
    then
        printf "\e[1;32m===========================$node 启动成功==================================     \e[0m\n"
    else
        printf "\e[1;32m===========================$node 启动失败==================================     \e[0m\n"
    	exit 8 
    fi
    done
done
}

# 引用位置变量，选择自己需要的功能，如果已有redis环境，需要配置cluster集群，需要自己根据环境修改脚本
case $1 in
'install')
    hostsIp
    sshCopy
    installRedis
	;;
'bgsave')
    redisBGSAVE
	;;
'cluster')
    redisCluster
	;;
'*')
    printf "\e[1;32m===========================输入错误，请检查拼写=================================\e[0m\n"
	exit 9 
	;;
esac