如何創(chuàng)建自動(dòng)化腳本

在前面的 bash 初學(xué)者系列文章中，我們介紹了關(guān)于 bash 的一些基礎(chǔ)知識(shí)，在了解了這些基礎(chǔ)知識(shí)之后，可以嘗試創(chuàng)建一些自動(dòng)化腳本，來完成一些相對(duì)枯燥重復(fù)的管理任務(wù)。

今天我們介紹一下如何創(chuàng)建自動(dòng)化腳本。

使用 bash 腳本自動(dòng)化用戶管理

在服務(wù)器上創(chuàng)建用戶可能是系統(tǒng)管理員經(jīng)常要做的事情，這是一個(gè)相對(duì)乏味的任務(wù)，所以我們可以創(chuàng)建一個(gè)自動(dòng)化的腳本來進(jìn)行這項(xiàng)任務(wù)。

首先，創(chuàng)建一個(gè)文本文件，其中包含要添加用戶的所有服務(wù)器主機(jī)名或IP地址。

比如，下面我們創(chuàng)建了一個(gè)名為 servers.txt 的文本文件，在其中添加了5臺(tái)服務(wù)器：

$ cat servers.txt
server1
server2
server3
server4
server5

在這里我使用的是主機(jī)名，因?yàn)橐呀?jīng)在 /etc/hosts 文件中包含了IP地址（另外也可以使用 SSH 配置文件）。

然后創(chuàng)建一個(gè)腳本文件 adduser.sh，代碼如下：

#!/bin/bash


servers=$(cat servers.txt)


echo -n "Enter the username: "
read name
echo -n "Enter the user id: "
read uid


for i in $servers; do
    echo $i
    ssh $i "sudo useradd -m -u $uid ansible"
    if [ $? -eq 0 ]; then
        echo "User $name added on $i"
    else
        echo "Error on $i"
    fi
done

上述腳本首先要求我們輸入要添加的用戶的用戶名和用戶 ID，然后回循環(huán)并連接到 servers.txt 中列出的服務(wù)器，并添加用戶。

運(yùn)行結(jié)果如下：

通過上述例子，我們需要了解以下幾點(diǎn)：

• 可以使用 ssh 代理或空的 ssh 密碼，這樣在運(yùn)行的時(shí)候系統(tǒng)不會(huì)要求輸入密碼；
• 必須擁有在所有服務(wù)器上具有超級(jí)用戶訪問權(quán)限（無(wú)需密碼）的有效帳戶。

可以想象，如果你有100臺(tái)服務(wù)器需要添加同一個(gè)新用戶，使用上述腳本肯定會(huì)節(jié)省很多時(shí)間。

使用 bash 腳本自動(dòng)備份

備份是我們經(jīng)常做的事情，那寫個(gè)自動(dòng)備份的腳本來自動(dòng)化實(shí)現(xiàn)，是非常方便的。看如下 backup.sh 腳本：

#!/bin/bash


backup_dirs=("/etc" "/home" "/boot")
dest_dir="/backup"
dest_server="server1"
backup_date=$(date +%b-%d-%y)


echo "Starting backup of: ${backup_dirs[@]}"


for i in "${backup_dirs[@]}"; do
    sudo tar -Pczf /tmp/$i-$backup_date.tar.gz $i
    if [ $? -eq 0 ]; then
        echo "$i backup succeeded."
    else
        echo "$i backup failed."
    fi
    scp /tmp/$i-$backup_date.tar.gz $dest_server:$dest_dir
    if [ $? -eq 0 ]; then
        echo "$i transfer succeeded."
    else
        echo "$i transfer failed."
    fi
done


sudo rm /tmp/*.gzecho "Backup is done."

上面腳本，首先創(chuàng)建了一個(gè)名為 backup_dirs 的數(shù)組，用來保存要備份的目錄。然后又創(chuàng)建了另外三個(gè)變量：

• dest_dir ：要備份到哪個(gè)目錄；
• dest_server ：備份服務(wù)器；
• backup_time ：備份日期。

然后，對(duì)于數(shù)組 backup_dirs 中的目錄，會(huì)在被壓縮為 tar 歸檔文件，保存在 /tmp 目錄下，然后使用 scp 命令將 tar 文件發(fā)送到備份服務(wù)器，最后，刪除 /tmp 中的 tar 文件。

運(yùn)行一下，結(jié)果如下：

$ ./backup.sh
Starting backup of: /etc /home /boot
/etc backup succeeded.
etc-Aug-30-20.tar.gz 100% 1288KB 460.1KB/s   00:02
/etc transfer succeeded.
/home backup succeeded.
home-Aug-30-20.tar.gz 100% 2543KB 547.0KB/s   00:04
/home transfer succeeded.
/boot backup succeeded.
boot-Aug-30-20.tar.gz 100%  105MB 520.2KB/s   03:26
/boot transfer succeeded.
Backup is done.

如果你希望每天在某個(gè)時(shí)間點(diǎn)（比如晚上）都自動(dòng)運(yùn)行備份腳本，可以制定 cron 任務(wù)：

$ crontab -e
00*** /home/kabary/scripts/backup.sh

監(jiān)視可用磁盤空間

我們每天做各種操作，產(chǎn)生的文件都會(huì)占用磁盤空間。在空間被占滿之前及時(shí)刪除不必要的臨時(shí)文件是有必要的。我們可以使用 df 命令查看系統(tǒng)的剩余空間：

$ df -h / /apps /database
Filesystem Size  Used Avail Use% Mounted on
/dev/sda5 20G  7.9G   11G  44% /
/dev/mapper/vg1-applv 4.9G  2.4G  2.3G  52% /apps
/dev/mapper/vg1-dblv 4.9G  4.5G  180M  97% /database

在 /database 目錄下其使用率為 97%，幾乎沒有剩余空間了。如果我使用 awk 命令可以只顯示第 5 個(gè)字段，就是使用率。

現(xiàn)在我們創(chuàng)建腳本 disk_space.sh，如下：

#!/bin/bash
filesystems=("/" "/apps" "/database")
for i in ${filesystems[@]}; do
    usage=$(df -h $i | tail -n 1 | awk '{print $5}' | cut -d % -f1)
    if [ $usage -ge 90 ]; then
        alert="Running out of space on $i, Usage is: $usage%"
        echo "Sending out a disk space alert email."
        echo $alert | mail -s "$i is $usage% full" your_email
    fi
done

上述代碼，首先創(chuàng)建了一個(gè)數(shù)組 filesystems，用于保存要監(jiān)視的目錄，然后便利這個(gè)數(shù)組，查看其中每個(gè)目錄的使用率是否大于 90%，如果使用率超過 90%，就會(huì)發(fā)送一封警告電子郵件，提示文件空間不足。

這里需要注意，帶啊中的 your_email 需要替換為真實(shí)的郵件。

運(yùn)行一下，結(jié)果如下：

$ ./disk_space.sh
Sending out a disk space alert email.

然后收到的電子郵件如下所示：

同樣，也可以為其設(shè)置定時(shí)任務(wù)：

$ crontab -e
0*/6 *** /home/kabary/scripts/disk_space.sh