load-ipvs.sh

load-ipvs.sh

#!/bin/bash
# Author: Delweng Zheng <delweng@gmail.com>

if $(lsmod | grep -q "ip_vs"); then
    exit 0
fi

if [ -f /etc/redhat-release ]; then
    RELEASE="centos"
elif cat /etc/issue | grep -Eqi "debian"; then
    RELEASE="debian"
elif cat /etc/issue | grep -Eqi "ubuntu"; then
    RELEASE="ubuntu"
elif cat /etc/issue | grep -Eqi "centos|red hat|redhat"; then
    RELEASE="centos"
elif cat /proc/version | grep -Eqi "debian"; then
    RELEASE="debian"
elif cat /proc/version | grep -Eqi "ubuntu"; then
    RELEASE="ubuntu"
elif cat /proc/version | grep -Eqi "centos|red hat|redhat"; then
    RELEASE="centos"
fi

if [[ "${RELEASE}" == "centos" ]]; then
    cat << EOF > /etc/sysconfig/modules/ipvs.modules
#!/bin/sh

##LOAD_IP_VS
modprobe ip_vs
modprobe ip_vs_wrr
modprobe ip_vs_rr
modprobe ip_vs_dh
modprobe ip_vs_sh
EOF
    chmod +x /etc/sysconfig/modules/ipvs.modules
elif [[ "${RELEASE}" == "ubuntu" ]]; then
    sed -i '/ip_vs/Id' /etc/modules
    cat << EOF > /etc/modules
##LOAD_IP_VS
modprobe ip_vs
modprobe ip_vs_wrr
modprobe ip_vs_rr
modprobe ip_vs_dh
modprobe ip_vs_sh
EOF
else
    echo "Not supported release: ${RELEASE}, by now only support centos|ubuntu"
    exit 1
fi

modprobe ip_vs
modprobe ip_vs_wrr
modprobe ip_vs_rr
modprobe ip_vs_dh
modprobe ip_vs_sh