• shell 函数


    一、shell函数的作用

      shell将需要多次使用的代码集合起来组合成函数,方便n次使用,减少代码量,调用函数使之方便,整洁。函数可以在shell 脚本当中做一个类似自定义执行命令,最大的功能就是可以简化很多的程序代码。需要注意的是shell脚本的执行方式是由上而下/由左而右,因此在shellscript当中的function的设置一定要在程序的最前面,这样才能够在执行时被找到可用的程序段。

    二、shell函数结构

    1、函数结构

    function 函数名(){
    	函数体
    	return n
    	}
    

    使用return或exit可以显式地结束函数

    2、返回值

    函数返回码是指函数最后一条命令的状态码,可以用于函数返回值
    使用return命令手动指定返回值:

    #!/bin/bash
    function() {
        echo "这个函数会对输入的两个数字进行相加运算..."
        echo "输入第一个数字: "
        read    Num1
        echo "输入第二个数字: "
        read   Num2
        echo "两个数字分别为 $Num1 和 $Num2 !"
        return $(($Num1 + $Num2))
    }
    function
    echo "输入的两个数字之和为 $? !"
    

    注:1、return结束函数,同时赋予函数运行结果($?  查看结束返回值恰好为计算结果;2、read 和 read -p,后者会打印出提示。

    3、向函数传参

    sum() {
    echo "第一个参数为 $1 !"
    echo "第二个参数为 $2 !"
    echo "第十个参数为 $10 !"
    echo "第十个参数为 ${10} !"
    echo "第十一个参数为 ${11} !"
    echo "参数总数有 $# 个!"
    echo "作为一个字符串输出所有参数 $* !"
    }
    sum  1 2 32 430 23 89 23  120 498 30  22
    
    # sum $1 $2 $3 $4 $5 $6 $7 $8 ${9} ${10}  ${11}  #此写法需要执行脚本时输入参数
    

    注意,$10 不能获取第十个参数,获取第十个参数需要${10}。当n>=10时,需要使用${n}来获取参数

    参数处理说明
    $# 传递到脚本或函数的参数个数
    $* 以一个单字符串显示所有向脚本传递的参数
    $$ 脚本运行的当前进程ID号
    $! 后台运行的最后一个进程的ID号
    $@ 与$*相同,但是使用时加引号,并在引号中返回每个参数。
    $- 显示Shell使用的当前选项,与set命令功能相同。
    $? 显示最后命令的退出状态。0表示没有错误,其他任何值表明有错误。

    三、常见的结构和例子

    1、在shell文件内部定义函数并引用

    #!/bin/bash
    function factorial
    {
    factorial=1
    for (( i=1;i <= $1;i++ ))
            do
            factorial=$[ $factorial * $i ]
            done
    echo $1的阶乘是:$factorial
    }
    echo '程序名':$0,用于求阶乘
    factorial $1
    

    2、将函数用在case判断中,减少代码量。

    #!/bin/bash
    PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/loacl/sbin:~/bin
    export PATH
     
    #函数1
    function printInfo ()
    {
            echo -n "你的选择是: "
    }
    
    #函数2
    function dotr()
    {
            tr 'a-z' 'A-Z'
    }
    
    read -p "Please input your choice(one|two|three|four):" num
    
    #用case做条件判断
    case $num in
            "one")
                    printInfo; echo $num | dotr
                    ;;
            "two")
                    printInfo; echo $num | dotr
                    ;;
            "Three")
                    printInfo; echo $num | dotr
                    ;;
            "four") printInfo; echo $num | dotr
                    ;;
    esac
    exit 0
    

    四、速度优化

    1、巧用main函数

    我们知道,像java,C这样的编译型语言都会有一个函数入口,这种结构使得代码可读性很强,我们知道哪些直接执行,那些是函数。但是脚本不一样,脚本属于解释性语言,从第一行直接执行到最后一行,如果在这当中命令与函数糅杂在一起,那就非常难读了。
    用python的朋友都知道,一个合乎标准的python脚本大体上至少是这样的:

    #!/usr/bin/env python
    
    def func1():
        pass
    
    def func2():
        pass
    
    if __name__=='__main__':
        func1()
        func2()
    

    他用一个很巧妙的方法实现了我们习惯的main函数,使得代码可读性更强。
    在shell中,我们也有类似的小技巧:

    #!/usr/bin/env bash
    
    func1(){
        #do sth
    }
    
    func2(){
        #do sth
    }
    
    main(){
        func1
        func2
    }
    
    main "$@"
    

    2、命令并行化,提高执行速度

    当我们需要充分考虑执行效率时,我们可能需要在执行命令的时候考虑并行化。

    shell中最简单的并行化是通过"&"以及"wait"命令来做:

    • 在每个进程中使用&符号进行让脚本在后台运行,无需等待当前进程结束。
    • 为了确保每个进程都执行完成,最后务必使用wait关键字,用来确保每一个子进程都执行完成。
    func(){
        #do sth
    }
    for((i=0;i<10;i++))do
        func &
    done
    wait
    

    当然,这里并行的次数不能太多,否则机器会卡死。稍微正确的做法比较复杂,如果图省事可以使用parallel命令来做,或者是xargs来处理。

    3、awk的执行效率会高于shell很多,能用awk替换的尽量替换

    sum=0
    for((i=0;i<100000;i++))
    do
    sum=$(($sum+$i))
    done
    echo $sum 
    
    time bash  sum.sh   
    4999950000
    
    real	0m0.502s
    user	0m0.475s
    sys	0m0.026s
    
    
    time awk 'BEGIN {sum=0; for (i=0;i<100000;i++) sum=sum+i;print sum;}'
    4999950000
    
    real	0m0.014s
    user	0m0.014s
    sys	0m0.000s
    

    4、使用新写法

    这里的新写法不是指有多厉害,而是指我们可能更希望使用较新引入的一些语法,更多是偏向代码风格的,比如

    1. 尽量使用func(){}来定义函数,而不是func{}
    2. 尽量使用[[]]来代替[]
    3. 尽量使用$()将命令的结果赋给变量,而不是反引号
    4. 在复杂的场景下尽量使用printf代替echo进行回显
  • 相关阅读:
    第02组 Beta冲刺(4/4)
    第02组 Beta冲刺(2/4)
    第02组 Beta冲刺(3/4)
    第02组 Beta冲刺(1/4)
    第02组 Alpha事后诸葛亮
    第02组 Alpha冲刺(4/4)
    第02组 Alpha冲刺(3/4)
    第02组 Alpha冲刺(2/4)
    第02组 Alpha冲刺(1/4)
    第02组 Beta版本演示
  • 原文地址:https://www.cnblogs.com/zjz20/p/14787960.html
Copyright © 2020-2023  润新知