改進你的腳本程序的五個方法
- 巧用 Bash 腳本程序能幫助你完成很多極具挑戰(zhàn)的任務(wù)。
系統(tǒng)管理員經(jīng)常寫腳本程序，不論長短，這些腳本可以完成某種任務(wù)。
你是否曾經(jīng)查看過某個軟件發(fā)行方提供的安裝用的腳本script程序?為了能夠適應(yīng)不同用戶的系統(tǒng)配置，順利完成安裝，這些腳本程序經(jīng)常包含很多函數(shù)和邏輯分支。多年來，我積累了一些改進腳本程序的一些技巧，這里分享幾個，希望能對朋友們也有用。這里列出一組短腳本示例，展示給大家做腳本樣本。
初步嘗試
我嘗試寫一個腳本程序時，原始程序往往就是一組命令行，通常就是調(diào)用標準命令完成諸如更新網(wǎng)頁內(nèi)容之類的工作，這樣可以節(jié)省時間。其中一個類似的工作是解壓文件到 Apache 網(wǎng)站服務(wù)器的主目錄里，我的最初腳本程序大概是下面這樣：
- cp january_schedule.tar.gz /usr/apache/home/calendar/
- cd /usr/apache/home/calendar/
- tar zvxf january_schedule.tar.gz
這幫我節(jié)省了時間，也減少了鍵入多條命令操作。時日久了，我掌握了另外的技巧，可以用 Bash 腳本程序完成更難的一些工作，比如說創(chuàng)建軟件安裝包、安裝軟件、備份文件系統(tǒng)等工作。
1、條件分支結(jié)構(gòu)
和眾多其他編程語言一樣，腳本程序的條件分支結(jié)構(gòu)同樣是強大的常用技能。條件分支結(jié)構(gòu)賦予了計算機程序邏輯能力，我的很多實例都是基于條件邏輯分支。
基本的條件分支結(jié)構(gòu)就是 if 條件分支結(jié)構(gòu)。通過判定是否滿足特定條件，可以控制程序選擇執(zhí)行相應(yīng)的腳本命令段。比如說，想要判斷系統(tǒng)是否安裝了 Java ，可以通過判斷系統(tǒng)有沒有一個 Java 庫目錄;如果找到這個目錄，就把這個目錄路徑添加到可運行程序路徑，也就可以調(diào)用 Java 庫應(yīng)用了。
if [ -d "$JAVA_HOME/bin" ] ; then PATH="$JAVA_HOME/bin:$PATH"
2、限定運行權(quán)限
你或許想只允許特定的用戶才能執(zhí)行某個腳本程序。除了 Linux 的權(quán)限許可管理，比如對用戶和用戶組設(shè)定權(quán)限、通過 SELinux 設(shè)定此類的保護權(quán)限等，你還可以在腳本里設(shè)置邏輯判斷來設(shè)置執(zhí)行權(quán)限。類似的情況可能是，你需要確保只有網(wǎng)站程序的所有者才能執(zhí)行相應(yīng)的網(wǎng)站初始化操作腳本。甚至你可以限定只有 root 用戶才能執(zhí)行某個腳本。這個可以通過在腳本程序里設(shè)置邏輯判斷實現(xiàn)，Linux 提供的幾個環(huán)境變量可以幫忙。其中一個是保存用戶名稱的變量 $USER， 另一個是保存用戶識別碼的變量 $UID 。在腳本程序里，執(zhí)行用戶的 UID 值就保存在 $UID 變量里。
用戶名判別
第一個例子里，我在一個帶有幾個應(yīng)用服務(wù)器實例的多用戶環(huán)境里指定只有用戶 jboss1 可以執(zhí)行腳本程序。條件 if 語句主要是判斷，“要求執(zhí)行這個腳本程序的用戶不是 jboss1 嗎?”當此條件為真時，就會調(diào)用第一個 echo 語句，接著是 exit 1，即退出這個腳本程序。
- if [ "$USER" != 'jboss1' ]; then
- echo "Sorry, this script must be run as JBOSS1!"
- exit 1
- fi
- echo "continue script"
根用戶判別
接下來的例子是要求只有根用戶才能執(zhí)行腳本程序。根用戶的用戶識別碼(UID)是 0，設(shè)置的條件判斷采用大于操作符(-gt)，所有 UID 值大于 0 的用戶都被禁止執(zhí)行該腳本程序。
- if [ "$UID" -gt 0 ]; then
- echo "Sorry, this script must be run as ROOT!"
- exit 1
- fi
- echo "continue script"
3、帶參數(shù)執(zhí)行程序
可執(zhí)行程序可以附帶參數(shù)作為執(zhí)行選項，命令行腳本程序也是一樣，下面給出幾個例子。在這之前，我想告訴你，能寫出好的程序并不只是寫出我們想要它執(zhí)行什么的程序，程序還需要不執(zhí)行我們不要它執(zhí)行的操作。如果運行程序時沒有提供參數(shù)造成程序缺少足夠信息，我愿意腳本程序不要做任何破壞性的操作。因而，程序的第一步就是確認命令行是否提供了參數(shù)，判定的條件就是參數(shù)數(shù)量 $# 是否為 0 ，如果是(意味著沒有提供參數(shù))，就直接終止腳本程序并退出操作。
- if [ $# -eq 0 ]; then
- echo "No arguments provided"
- exit 1
- fi
- echo "arguments found: $#"
多個運行參數(shù)
可以傳遞給腳本程序的參數(shù)不止一個。腳本使用內(nèi)部變量指代這些參數(shù)，內(nèi)部變量名用非負整數(shù)遞增標識，也就是 $1、$2、$3 等等遞增。我只是擴展前面的程序，并在下面一行輸出顯示用戶提供的前三個參數(shù)。顯然，要針對所有的每個參數(shù)有對應(yīng)的響應(yīng)需要更多的邏輯判斷，這里的例子只是簡單展示參數(shù)的使用。
- echo $1 $2 $3
我們在討論這些參數(shù)變量名，你或許有個疑問，“參數(shù)變量名怎么跳過了 $0，(而直接從$1 開始)?”
是的，是這樣，這是有原因的。變量名 $0 確實存在，也非常有用，它儲存的是被執(zhí)行的腳本程序的名稱。
- echo $0
程序執(zhí)行過程中有一個變量名指代程序名稱，很重要的一個原因是，可以在生成的日志文件名稱里包含程序名稱，最簡單的方式應(yīng)該是調(diào)用一個 echo 語句。
- echo test >> $0.log
當然，你或許要增加一些代碼，確保這個日志文件存放在你希望的路徑，日志名稱包含你認為有用的信息。
4、交互輸入
腳本程序的另一個好用的特性是可以在執(zhí)行過程中接受輸入，最簡單的情況是讓用戶可以輸入一些信息。
- echo "enter a word please:"
- read word
- echo $word
這樣也可以讓用戶在程序執(zhí)行中作出選擇。
- read -p "Install Software ?? [Y/n]: " answ
- if [ "$answ" == 'n' ]; then
- exit 1
- fi
- echo "Installation starting..."
5、出錯退出執(zhí)行
幾年前，我寫了個腳本，想在自己的電腦上安裝最新版本的 Java 開發(fā)工具包(JDK)。這個腳本把 JDK 文件解壓到指定目錄，創(chuàng)建更新一些符號鏈接，再做一下設(shè)置告訴系統(tǒng)使用這個最新的版本。如果解壓過程出現(xiàn)錯誤，在執(zhí)行后面的操作就會使整個系統(tǒng)上的 Java 破壞不能使用。因而，這種情況下需要終止程序。如果解壓過程沒有成功，就不應(yīng)該再繼續(xù)進行之后的更新操作。下面語句段可以完成這個功能。
- 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 命令創(chuàng)建一個文件名為 T 的文件，然后執(zhí)行這個單行命令，變量 ec 的值會是 0。然后，用 rm T 命令刪除文件，再執(zhí)行該單行命令，變量 ec 的值會是 2，因為文件 T 不存在，命令 ls 找不到指定文件報錯。
在邏輯條件里利用這個出錯標識，參照前文我使用的條件判斷，可以使腳本文件按需完成設(shè)定操作。
結(jié)語
要完成復(fù)雜的功能，或許我們覺得應(yīng)該使用諸如 Python、C 或 Java 這類的高級編程語言，然而并不盡然，腳本編程語言也很強大，可以完成類似任務(wù)。要充分發(fā)揮腳本的作用，有很多需要學習的，希望這里的幾個例子能讓你意識到腳本編程的強大。
