Linux Shell——函数的使用

文/一介书生,一枚码农。

scripts are for lazy people.

Linux Shell

函数是存在内存里的一组代码的命名的元素。函数创建于脚本运行环境之中,并且可以执行。
函数的语法结构为:

function <function-name> {
    <code to execute>
}

创建函数不需要dodone这些关键字,只需要大括号来界定函数的作用范围。
我们可以在命令行中创建一个简单的函数:

function show_system {
	echo "The uptime is:"
	uptime
	echo
	
	echo "Current dir:"
	pwd
	echo
	
	echo "Date:"
	date
}

在命令行定义上面的函数:

function in command line

然后执行函数:

show_system

最后,我们可以使用type命令查看函数的定义:

type show_system

方法可以有传递的参数。在此之前,我们先简单介绍 sed命令,下面的脚本用来移除文件中的注释行和空白行。

sed -i.bak '/^\s*#/d;/^$/d' <filename>

详细介绍一下上面的脚本:

  • sed -i.bak:sed 命令会创建一个以.bak结尾的备份文件,用来备份原始的文件。
  • /^:表示行首。
  • \s*:匹配任何不可见字符,包括空格、制表符、换页符等。
  • #/:shell 脚本中的注释标记。 /^\s*#/一起使用用来查找注释行和带有不可见字符的注释行。
  • d:用来删除匹配的行。
  • ;/^$/d;是表达式的分隔符,后面的表达式跟前面的表达式意思很像,用来删除空白行, "$"匹配输入字符串的结束位置。

我们把上面的脚本移到函数中,创建一个名为clean_file的函数:

function clean_file {
  sed -i.bak '/^\s*#/d;/^$/d' "$1"
}

上面的代码中,我们使用位置参数$1来接收命令行里传递的参数,需要注意的是$1 用大括号来括起来。

下面,我们完善clean_file方法,新建一个脚本文件 clean.sh。代码如下:

#!/bin/bash
# Script will prompt for filename
# And remove commented and blank lines

function is_file {
    if [ ! -f "$1" ] ; then
        echo "$1 does not seem to be a file"
        exit 2
    fi
}

function clean_file {
    is_file "$1"
    BEFORE=$(wc -l "$1")
    echo "The file $1 starts with $BEFORE"
    sed -i.bak '/^\s*#/d;/^$/d' "$1"
    AFTER=$(wc -l "$1")
    echo "The file $1 is now $AFTER"
}

read -p "Enter a file to clean: "
clean_file "$REPLY"
exit 1

其中,wc -l命令用来打印文件行数。
现在,我们新建一个用来测试的文件 test.txt,里面随便写些内容:

# This file is just for testing.

Hello.

World.

# new comment.

$

The end.

在 test 文件所在的目录下,执行 clean.sh脚本:

clean.sh

然后查看 test 文件和 test.bak 文件,发现空白行和注释行已经删除。

cat test

这里需要注意的是,shell 脚本是从上至下来执行的,如果在一个函数中调用了另一个函数,那么被调用的函数要定义在调用函数的前面。

当然,除了函数可以传递参数外,还可以有返回值。默认情况下,在执行完函数内的最后一行代码后,方法会返回一个状态的数字,你可以使用$?变量来查看函数执行的状态。如果返回值为0,表示方法正常退出,非0表示程序发生错误或其他非正常退出。
我们使用return关键字来返回一个整数。这里需要注意两点:

  • 函数一旦执行完就会返回状态代码。
  • 状态代码的范围为0到255。

如果不想返回函数的状态代码,而是想返回一个字符串或是其他类型,可以使用 echo 变量的方式来返回值。

下面创建脚本文件 translate.sh,用来把文件中的大写字符转换为小写字符:

#!/bin/bash
# Script that translates upper letter to low letter

function to_lower ()
{
    input="$1"
    output=$(tr '[A-Z]' '[a-z]' <<< "$input")
    echo $output
}

while true
do
    read -p "Enter c to continue or q to exit: "
    result=$(to_lower $REPLY)
    echo "After translate: $result"
    
    if [ $result = "q" ] ; then
        break
    fi

done
echo "Finished"

代码会把输入的大写字符变成小写字符,当输入"Q"或"q"时程序退出。

posted @ 2017-05-02 15:50  林本托  阅读(16665)  评论(0编辑  收藏  举报