【Shell脚本编程（二）】函数的高级用法

最新推荐文章于 2021-06-02 16:12:32 发布

原创最新推荐文章于 2021-06-02 16:12:32 发布 · 363 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#Shell脚本编程

Linux 专栏收录该内容

24 篇文章

订阅专栏

1. 函数定义和使用

1.1. 函数介绍

Linux Shell 中的函数和大多数编程语言中的函数一样；
将相似的任务或代码封装到函数中，供其他地方调用；

1.2. 语法格式

第一种格式：

name()
{
    command1
    command2
    ......
    commandn
}

第二种格式：

function name
{
    command1
    command2
    ......
    commandn
}

1.3. 如何调用函数

直接使用函数名调用，可以将其想象成Shell中的一条命令；
函数内部可以直接使用参数 $1 、 ¥ 2 . . .$ n
调用函数：function_name $1 $2

test()
{
    echo "test function"
}

test  #结果为test function


function greeting
{
    echo "hello,zhangsan"
}

greeting  #结果为hello,zhangsan

需求描述：写一个监控nginx的脚本；如果Nginx服务宕掉，则该脚本可以检测到并将进程启动；如果正常运行，则不做任何处理。

vim nginx_daemon.sh

############脚本文件内容##############

#!/bin/bash
#

ps -ef | grep nginx | grep -v grep

echo $?
############脚本文件内容##############

#将nginx停掉后执行
sh nginx_daemon.sh  #结果为0，是因为虽然过滤掉grep nginx，但是脚本文件名称仍然存在nginx（nginx_daemon.sh）

#修改脚本文件名称
mv nginx_daemon.sh daemon.sh
sh daemon.sh  #结果为1

#或者在脚本中这样实现
#!/bin/bash
#

this_pid=$$

ps -ef | grep nginx | grep -v grep | grep -v $this_pid

echo $?

#完整实现脚本
#!/bin/bash
#

this_pid=$$

while true
do
ps -ef | grep nginx | grep -v grep | grep -v $this_pid &> /dev/null

if [ $? -eq 0];then
    echo "Nginx is running well"
    sleep 3
else
    systemctl start nginx
    echo "Nginx is down,Start it..."
fi
done
##############脚本文件内容################

#如果不希望终端打印提示信息则执行
nohup sh nginx_daemon.sh &

#查看打印的日志信息
tail -f nohup.out

2. 向函数传递参数

2.1. 高级语言传参

int example_1(int arg1, int arg2)
{
    arg1=arg2
    ......
    ......
    return null
}

2.2. 高级语言函数调用

int num1=10;
int num2=20;

#调用函数形式一：
int num3=example_1(num1,num2);

#调用函数形式二：
int num4;
num4=example_1(num1,num2);

2.3. Shell中传参

function name
{
    echo "Hello $1"
    echo "Hello $2"
}

2.4. Shell中函数调用

函数调用：name Lily Allen

funciton greeting
{
    echo "Hello $1"
}

greeting zhangsan  #结果为Hello Zhangsan
greeting Lisi  #结果为Hello Lisi

例子1:

需求描述：写一个脚本，该脚本可以实现计算器的功能，可以进行±*/四种计算。例如：sh calculate.sh 30 + 40 | sh calculate.sh 30 - 40 | sh calculate.sh 等。

#计算器脚本
vim calculate.sh
#############脚本文件内容##############
#!/bin/bash
#

function calcu
{
    case $2 in
            +)
                echo "`expr $1 + $3`"
                ;;
            -)
                echo "`expr $1 - $3`"
                ;;
            \*)
                echo "`expr $1 \* $3`"
                ;;
            /)
                echo "`expr $1 / $3`"
                ;;
    esac
}

calcu $1 $2 $3

3. 函数返回值

3.1. 返回值的方式

方法	代码
方法一	return
方法二	echo

3.2. 使用return返回值

使用return返回值，只能返回1-255的整数
函数使用return返回值，通常只是用来供其他地方调用获取状态，因此通常仅返回0或1；0表示成功，1表示失败；

3.3. 使用echo返回值

使用echo可以返回任何字符串结果；
通常用于返回数据，比如一个字符串值或者列表值；

vim nginx.sh

#!/bin/bash
#

this_pid=$$

function is_nginx_running
{
    ps -ef | grep nginx | grep -v grep | grep -v $this_pid &> /dev/null
    if [ $? -eq 0];then
        return
    else
        return 1
    fi
}

is_nginx_running && echo "Nginx is running" || echo "Nginx is stoped"

vim get_sys_user.sh

#!/bin/bash
#

function get_users
{
    users=`cat /etc/passwd | cut -d: -f1`
    echo $users
}

user_list=`get_users`

index=1
for u in $user_list
do 
    echo "The $index user is : $u"
    index=$(($index+1))
done

4. 局部变量和全局变量

4.1. 全局变量

不做特殊声明，Shell中变量都是全局变量；
Tips：大型脚本程序中函数中慎用全局变量；

4.2. 局部变量

定义变量时，使用local关键字；
函数内和外若存在同名变量，则函数内部变量覆盖外部变量；

vim var.sh

#!/bin/bash
#
var1="Hello world"

function test
{
    local var2=87
}

test
echo $var1
echo $var2

5. 函数库

5.1. 为什么要定义函数库

经常使用的重复代码封装成函数文件；
一般不直接执行，而是由其他脚本调用；

函数库：

定义一个函数库，该函数库实现以下几个函数：

加法函数add；
减法函数reduce；
乘法函数multiple；
除法函数divide；
打印系统运行情况的函数sys_load，该函数可以显示内存运行情况，磁盘使用情况；

vim base_function

function add
{
    echo "`expr $1 + $2`"
}

function reduce
{
    echo "`expr $1 - $2`"
}

function multiple
{
    echo "`expr $1 \* $2`"
}

function divide
{
    echo "`expr $1 / $2`"
}

function sys_load
{
    echo "Memory Info"
    echo 
    free -m
    echo
    
    echo "Disk Usage"
    echo
    df -h
    echo
}


###############################################
vim calculate.sh

#!/bin/bash
#

. /root/lesson/3.5/lib/base_function  #引入函数库（函数库文件所在的绝对路径）

add 12 23

reduce 90 30
 
multiple 12 12

divide 12 2

sys_load