shell腳本自動磁區、創建邏輯卷

#!/bin/bash
# vim:et:ts=4:sw=4:
<<INFO
SCRIPYT:autodiskmount.sh
AUTHOR:anqixiang
DATE:2020-10-18
DESCRIBE:把磁盤分為一個區，并做邏輯卷
SYSTEM:CentOS 7、RedHat 7、Ubuntu16/18
WARNING:LV_NAME、DISK_CAP、MOUNT_DIR三個陣列的元素個數需一致
VERSION:1.1.0
MODIFY:
INFO

set -e

cecho(){
    echo -e "\033[$1m$2\033[0m"
}

#定義邏輯卷的相關引數
VG_NAME="data"          #卷組名
LV_NAME=(${VG_NAME}_app ${VG_NAME}_mysql)   #以陣列的形式存盤邏輯卷名字
DISK_CAP=(30% 70%)                          #代表第一個邏輯卷占30%磁盤空間，第二個占70%
MOUNT_DIR=(/data/app /data/mysql)           #邏輯卷的掛載路徑

######################功能函式######################
#檢測磁盤
Check_Disk()
{
    local disk_num=$(fdisk -l |grep "${DISK_NMAE}" |wc -l)
    if [ ${disk_num} -eq 0 ];then
        cecho 31 "${DISK_NMAE}不存在" && exit 1
    elif [ ${disk_num} -gt 1 ];then
        cecho 31 "${DISK_NMAE}已磁區,請先格式化" && exit 1
    fi
    cecho 32 "#磁區磁盤為:${DISK_NMAE}"
}

#磁盤磁區
Divi_Disk()
{
    fdisk ${DISK_NMAE} << EOF
n
p
1


wq
EOF
    fdisk -l |grep "${DISK_NMAE}"
}

#創建邏輯卷
Create_Lvm()
{
    #創建卷組
    if ! vgs | grep -w "${VG_NAME}" >/dev/null;then  
        vgcreate "${VG_NAME}" "${DISK_NMAE}1"
    fi
    #創建邏輯卷
    for((i=0;i<${#LV_NAME[@]};i++))
    do
        if ! lvs | grep -w "${LV_NAME[i]}" >/dev/null;then
            lvcreate -n ${LV_NAME[i]} -l ${DISK_CAP[i]}VG ${VG_NAME} -y
            mkfs.ext4 /dev/${VG_NAME}/${LV_NAME[i]} >/dev/null           
        fi
    done
    lvs
}

#掛載邏輯卷
Mount_Lvm()
{
    for((i=0;i<${#MOUNT_DIR[@]};i++))
    do
        [ ! -d ${MOUNT_DIR[i]} ] && mkdir -p ${MOUNT_DIR[i]}
        local uuid=$(blkid | grep -w "${VG_NAME}-${LV_NAME[i]}" |awk '{print $2}')
        [ ! -f /etc/fstab.bak ] && cp /etc/fstab{,.bak}
        sed -ri "s#(.*)${MOUNT_DIR[i]}(.*)##" /etc/fstab        #防止多次寫入相同的掛載內容
        echo "${uuid} ${MOUNT_DIR[i]} ext4 defaults 0 0" >> /etc/fstab
        sed -i '/^$/d' /etc/fstab
        mount -a
        if df -h | grep -w "${MOUNT_DIR[i]}";then
            cecho 96 "${LV_NAME[i]}邏輯卷創建成功"
        else
            cecho 31 "${LV_NAME[i]}邏輯卷創建失敗"
        fi
    done
}

#卸載邏輯卷
Umount_Lvm()
{
    for((i=0;i<${#MOUNT_DIR[@]};i++))
    do
        df -h |grep -w "${MOUNT_DIR[i]}" && umount ${MOUNT_DIR[i]}
        [ -d ${MOUNT_DIR[i]} ] && rm -rf ${MOUNT_DIR[i]}
        if lvs | grep -w "${LV_NAME[i]}" >/dev/null;then
            lvremove -f /dev/${VG_NAME}/${LV_NAME[i]}
        fi
        sed -ri "s#(.*)${MOUNT_DIR[i]}(.*)##" /etc/fstab
    done
    if vgs | grep -w "${VG_NAME}" >/dev/null;then
        vgremove ${VG_NAME}
    fi
    sed -i '/^$/d' /etc/fstab
    mount -a
    #洗掉磁盤磁區
    fdisk ${DISK_NMAE} << EOF
d
w
EOF
    local disk_num=$(fdisk -l |grep "${DISK_NMAE}" |wc -l)
    if [ ${disk_num} -eq 1 ];then
        cecho 96 "邏輯卷卸載成功"
    else
        cecho 31 "邏輯卷卸載失敗"
    fi
}

#幫助資訊
Help(){
	cat << EOF
Usage: 
=======================================================================
EXAMPLE:
    bash autodiskmount.sh /dev/sdb mount     #創建邏輯卷
    bash autodiskmount.sh /dev/sdb umount    #卸載邏輯卷
EOF
}

######################主函式######################
[ "$#" -ne 2 ] && Help && exit 1
DISK_NMAE=$1
if [ "x$2" == "xmount" ];then
    Check_Disk
    Divi_Disk
    Create_Lvm
    Mount_Lvm
elif [ "x$2" == "xumount" ];then
    Umount_Lvm
else
    Help && exit 1
fi
set +e