如何在 Linux Shell 程式設計中定義和使用函式 | Linux 中國

函式是一段可復用的程式碼。我們通常把重覆的程式碼放進函式中並且在不同的地方去呼叫它。庫是函式的集合。我們可以在庫中定義經常使用的函式，這樣其它指令碼便可以不再重覆程式碼而使用這些函式。

本文我們將討論諸多關於函式的內容和一些使用技巧。為了方便演示，我將在 Ubuntu 系統上使用 Bourne Again SHell (Bash)。

呼叫函式

在 Shell 中呼叫函式和呼叫其它命令是一模一樣的。例如，如果你的函式名稱為 my_func，你可以在命令列中像下麵這樣執行它：

$ my_func

如果你的函式接收多個引數，那麼可以像下麵這樣寫（類似命令列引數的使用）：

$ my_func arg1 arg2 arg3

定義函式

我們可以用下麵的語法去定義一個函式：

function function_name {
    Body of function
}

函式的主體可以包含任何有效的命令、迴圈陳述句和其它函式或指令碼。現在讓我們建立一個簡單的函式，它向螢幕上顯示一些訊息（註：直接在命令列裡寫）。

function print_msg {
    echo "Hello, World"
}

現在，讓我們執行這個函式：

$ print_msg
Hello, World

不出所料，這個函式在螢幕上顯示了一些訊息。

在上面的例子中，我們直接在終端裡建立了一個函式。這個函式也可以儲存到檔案中。如下麵的例子所示。

#!/bin/bash
function print_msg {
    echo "Hello, World"
}
print_msg

我們已經在 function.sh 檔案中定義了這個函式。現在讓我們執行這個指令碼：

$ chmod +x function.sh
$ ./function.sh
Hello, World

你可以看到，上面的輸出和之前的是一模一樣的。

更多函式用法

在上一小節中我們定義了一個非常簡單的函式。然而在軟體開發的過程中，我們需要更多高階的函式，它可以接收多個引數並且帶有傳回值。在這一小節中，我們將討論這種函式。

向函式傳遞引數

我們可以像呼叫其它命令那樣給函式提供引數。我們可以在函式裡使用美元 $ 符號訪問到這些引數。例如，$1 表示第一個引數，$2 代表第二個引數，以此類推。

讓我們修改下之前的函式，讓它以引數的形式接收資訊。修改後的函式就像這樣：

function print_msg {
    echo "Hello $1"
}

在上面的函式中我們使用 $1 符號訪問第一個引數。讓我們執行這個函式：

$ print_msg "LinuxTechi"

執行完後，生成如下資訊：

Hello LinuxTechi

從函式中傳回數值

跟其它程式語言一樣，Bash 提供了傳回陳述句讓我們可以向呼叫者傳回一些數值。讓我們舉例說明：

function func_return_value {
    return 10
}

上面的函式向呼叫者傳回 10。讓我們執行這個函式：

$ func_return_value
$ echo "Value returned by function is: $?"

當你執行完，將會產生如下的輸出結果：

Value returned by function is: 10

提示：在 Bash 中使用 $? 去獲取函式的傳回值。

函式技巧

目前我們已經對 Bash 中的函式有了一些瞭解。現在讓我們建立一些非常有用的 Bash 函式，它們可以讓我們的生活變得更加輕鬆。

Logger

讓我們建立一個 logger 函式，它可以輸出帶有日期和時間的 log 資訊。

function log_msg {
    echo "[`date '+ %F %T'` ]: $@"
}

執行這個函式：

$ log_msg "This is sample log message"

執行完，就會生成如下資訊：

[ 2018-08-16 19:56:34 ]: This is sample log message

顯示系統資訊

讓我們建立一個顯示 GNU/Linux 資訊的函式

function system_info {
    echo "### OS information ###"
    lsb_release -a
    echo
    echo "### Processor information ###"
    processor=`grep -wc "processor" /proc/cpuinfo`
    model=`grep -w "model name" /proc/cpuinfo  | awk -F: '{print $2}'`
    echo "Processor = $processor"
    echo "Model     = $model"
    echo
    echo "### Memory information ###"
    total=`grep -w "MemTotal" /proc/meminfo | awk '{print $2}'`
    free=`grep -w "MemFree" /proc/meminfo | awk '{print $2}'`
    echo "Total memory: $total kB"
    echo "Free memory : $free kB"
}

### OS information ###
No LSB modules are available.
Distributor ID:           Ubuntu
Description:   Ubuntu 18.04.1 LTS
Release:         18.04
Codename:    bionic
### Processor information ###
Processor = 1
Model     =  Intel(R) Core(TM) i7-7700HQ CPU @ 2.80GHz
### Memory information ###
Total memory: 4015648 kB
Free memory : 2915428 kB

function search {
    find . -name $1
}

$ search dir4

./dir1/dir2/dir3/dir4

function digital_clock {
    clear
    while [ 1 ]
    do
        date +'%T'
        sleep 1
        clear
    done
}

$ source utils.sh