Linux中國

改進你的腳本程序的 5 個方法

系統管理員經常寫腳本程序,不論長短,這些腳本可以完成某種任務。

你是否曾經查看過某個軟體發行方提供的安裝用的 腳本 script 程序?為了能夠適應不同用戶的系統配置,順利完成安裝,這些腳本程序經常包含很多函數和邏輯分支。多年來,我積累了一些改進腳本程序的一些技巧,這裡分享幾個,希望能對朋友們也有用。這裡列出一組短腳本示例,展示給大家做腳本樣本。

初步嘗試

我嘗試寫一個腳本程序時,原始程序往往就是一組命令行,通常就是調用標準命令完成諸如更新網頁內容之類的工作,這樣可以節省時間。其中一個類似的工作是解壓文件到 Apache 網站伺服器的主目錄里,我的最初腳本程序大概是下面這樣:

cp january_schedule.tar.gz /usr/apache/home/calendar/
cd /usr/apache/home/calendar/
tar zvxf january_schedule.tar.gz

這幫我節省了時間,也減少了鍵入多條命令操作。時日久了,我掌握了另外的技巧,可以用 Bash 腳本程序完成更難的一些工作,比如說創建軟體安裝包、安裝軟體、備份文件系統等工作。

1、條件分支結構

和眾多其他編程語言一樣,腳本程序的條件分支結構同樣是強大的常用技能。條件分支結構賦予了計算機程序邏輯能力,我的很多實例都是基於條件邏輯分支。

基本的條件分支結構就是 if 條件分支結構。通過判定是否滿足特定條件,可以控制程序選擇執行相應的腳本命令段。比如說,想要判斷系統是否安裝了 Java ,可以通過判斷系統有沒有一個 Java 庫目錄;如果找到這個目錄,就把這個目錄路徑添加到可運行程序路徑,也就可以調用 Java 庫應用了。

if [ -d "$JAVA_HOME/bin" ] ; then
    PATH="$JAVA_HOME/bin:$PATH"

2、限定運行許可權

你或許想只允許特定的用戶才能執行某個腳本程序。除了 Linux 的許可權許可管理,比如對用戶和用戶組設定許可權、通過 SELinux 設定此類的保護許可權等,你還可以在腳本里設置邏輯判斷來設置執行許可權。類似的情況可能是,你需要確保只有網站程序的所有者才能執行相應的網站初始化操作腳本。甚至你可以限定只有 root 用戶才能執行某個腳本。這個可以通過在腳本程序里設置邏輯判斷實現,Linux 提供的幾個環境變數可以幫忙。其中一個是保存用戶名稱的變數 $USER, 另一個是保存用戶識別碼的變數 $UID 。在腳本程序里,執行用戶的 UID 值就保存在 $UID 變數里。

用戶名判別

第一個例子里,我在一個帶有幾個應用伺服器實例的多用戶環境里指定只有用戶 jboss1 可以執行腳本程序。條件 if 語句主要是判斷,「要求執行這個腳本程序的用戶不是 jboss1 嗎?」當此條件為真時,就會調用第一個 echo 語句,接著是 exit 1,即退出這個腳本程序。

if [ "$USER" != 'jboss1' ]; then
     echo "Sorry, this script must be run as JBOSS1!"
     exit 1
fi
echo "continue script"

根用戶判別

接下來的例子是要求只有根用戶才能執行腳本程序。根用戶的用戶識別碼(UID)是 0,設置的條件判斷採用大於操作符(-gt),所有 UID 值大於 0 的用戶都被禁止執行該腳本程序。

if [ "$UID" -gt 0 ]; then
     echo "Sorry, this script must be run as ROOT!"
     exit 1
fi
echo "continue script"

3、帶參數執行程序

可執行程序可以附帶參數作為執行選項,命令行腳本程序也是一樣,下面給出幾個例子。在這之前,我想告訴你,能寫出好的程序並不只是寫出我們想要它執行什麼的程序,程序還需要不執行我們不要它執行的操作。如果運行程序時沒有提供參數造成程序缺少足夠信息,我願意腳本程序不要做任何破壞性的操作。因而,程序的第一步就是確認命令行是否提供了參數,判定的條件就是參數數量 $# 是否為 0 ,如果是(意味著沒有提供參數),就直接終止腳本程序並退出操作。

if [ $# -eq 0 ]; then
    echo "No arguments provided"
    exit 1
fi
echo "arguments found: $#"

多個運行參數

可以傳遞給腳本程序的參數不止一個。腳本使用內部變數指代這些參數,內部變數名用非負整數遞增標識,也就是 $1$2$3 等等遞增。我只是擴展前面的程序,並在下面一行輸出顯示用戶提供的前三個參數。顯然,要針對所有的每個參數有對應的響應需要更多的邏輯判斷,這裡的例子只是簡單展示參數的使用。

echo $1 $2 $3

我們在討論這些參數變數名,你或許有個疑問,「參數變數名怎麼跳過了 $0,(而直接從$1 開始)?」

是的,是這樣,這是有原因的。變數名 $0 確實存在,也非常有用,它儲存的是被執行的腳本程序的名稱。

echo $0

程序執行過程中有一個變數名指代程序名稱,很重要的一個原因是,可以在生成的日誌文件名稱里包含程序名稱,最簡單的方式應該是調用一個 echo 語句。

echo test >> $0.log

當然,你或許要增加一些代碼,確保這個日誌文件存放在你希望的路徑,日誌名稱包含你認為有用的信息。

4、交互輸入

腳本程序的另一個好用的特性是可以在執行過程中接受輸入,最簡單的情況是讓用戶可以輸入一些信息。

echo "enter a word please:"
read word
echo $word

這樣也可以讓用戶在程序執行中作出選擇。

read -p "Install Software ?? [Y/n]: " answ
if [ "$answ" == 'n' ]; then
    exit 1
fi
    echo "Installation starting..."

5、出錯退出執行

幾年前,我寫了個腳本,想在自己的電腦上安裝最新版本的 Java 開發工具包(JDK)。這個腳本把 JDK 文件解壓到指定目錄,創建更新一些符號鏈接,再做一下設置告訴系統使用這個最新的版本。如果解壓過程出現錯誤,在執行後面的操作就會使整個系統上的 Java 破壞不能使用。因而,這種情況下需要終止程序。如果解壓過程沒有成功,就不應該再繼續進行之後的更新操作。下面語句段可以完成這個功能。

tar kxzmf jdk-8u221-linux-x64.tar.gz -C /jdk --checkpoint=.500; ec=$?
if [ $ec -ne 0 ]; then
     echo "Installation failed - exiting."
     exit 1
fi

下面的單行語句可以給你快速展示一下變數 $? 的用法。

ls T; ec=$?; echo $ec

先用 touch T 命令創建一個文件名為 T 的文件,然後執行這個單行命令,變數 ec 的值會是 0。然後,用 rm T 命令刪除文件,再執行該單行命令,變數 ec 的值會是 2,因為文件 T 不存在,命令 ls 找不到指定文件報錯。

在邏輯條件里利用這個出錯標識,參照前文我使用的條件判斷,可以使腳本文件按需完成設定操作。

結語

要完成複雜的功能,或許我們覺得應該使用諸如 Python、C 或 Java 這類的高級編程語言,然而並不盡然,腳本編程語言也很強大,可以完成類似任務。要充分發揮腳本的作用,有很多需要學習的,希望這裡的幾個例子能讓你意識到腳本編程的強大。

via: https://opensource.com/article/20/1/improve-bash-scripts

作者:Alan Formy-Duval 選題:lujun9972 譯者:fisherue 校對:wxy

本文由 LCTT 原創編譯,Linux中國 榮譽推出


本文轉載來自 Linux 中國: https://github.com/Linux-CN/archive

對這篇文章感覺如何?

太棒了
0
不錯
0
愛死了
0
不太好
0
感覺很糟
0
雨落清風。心向陽

    You may also like

    Leave a reply

    您的電子郵箱地址不會被公開。 必填項已用 * 標註

    此站點使用Akismet來減少垃圾評論。了解我們如何處理您的評論數據

    More in:Linux中國