shell脚本专题(05)：传参

一、专题背景

最近使用了个自动化平台（详见自动化运维平台Spug测试）进行每周的变更，效果很不错，平台将大量重复繁琐的操作通过脚本分发方式标准化自动化了，平台核心是下发到各个服务器的shell脚本，感觉有必要对shell脚本做个总结，所以有了写本专题的想法。本专题将结合运维实际介绍shell脚本的各项用法，预计10篇左右，将包括系统巡检、监控、ftp上传下载、数据库查询、日志清理、时钟同步、定时任务等，里面会涉及shell常用语法、注意事项、调试排错等。

二、本文前言

本文是该专题的第五篇。

文章主要介绍如何在脚本执行时传参、如何对传入的参数做基本的处理、如何处理选项、怎样执行用户输出。

三、环境说明

主机名	操作系统版本	ip	备注
shell	Centos 7.6.1810	172.16.7.100	shell测试服务器

四、脚本测试

1.命令行参数

[root@shell param]# more test1.sh 
#!/bin/bash

name=$(basename $0)
total=$[ $4 * $5 ]

echo 这是第一个参数: $1
echo 这是第二个参数: $2
echo 这是第三个参数: $3
echo 这是第四个参数: $4
echo 这是第五个参数: $5

echo 脚本名称为 $name
echo $4乘以$5的值为 $total

[root@shell param]# ./test1.sh 1 a "b c" 4 5
这是第一个参数: 1
这是第二个参数: a
这是第三个参数: b c
这是第四个参数: 4
这是第五个参数: 5
脚本名称为 test1.sh
4乘以5的值为 20

传入的参数可以是数字、字符和字符串

$0 是脚本名， $1是第一个参数， $2 是第二个参数，依次类推，直到第九个参数 $9

2.判断参数个数

[root@shell param]# ./test1.sh 
./test1.sh:行4: *  : 语法错误: 期待操作数 （错误符号是 "*  "）
这是第一个参数:
这是第二个参数:
这是第三个参数:
这是第四个参数:
这是第五个参数:
脚本名称为 test1.sh
乘以的值为

当脚本执行时未加参数则执行报错，此时需要对脚本进行优化，可以使用特殊变量$#判断传入脚本的参数个数。

[root@shell param]# more test1.sh 
#!/bin/bash

if [ $# -ne 5 ]
then
    echo 参数个数有误，请重新输入
else
    name=$(basename $0)
    total=$[ $4 * $5 ]
    echo 这是第一个参数: $1
    echo 这是第二个参数: $2
    echo 这是第三个参数: $3
    echo 这是第四个参数: $4
    echo 这是第五个参数: $5
    echo 脚本名称为 $name
    echo $4乘以$5的值为 $total
fi
[root@shell param]# ./test1.sh 
参数个数有误，请重新输入
[root@shell param]# ./test1.sh  1 2 3 4
参数个数有误，请重新输入
[root@shell param]# ./test1.sh  1 2 3 
参数个数有误，请重新输入
[root@shell param]# ./test1.sh  1 2 3 4 5 6 
参数个数有误，请重新输入
[root@shell param]# ./test1.sh 1 a "b c" 4 5
这是第一个参数: 1
这是第二个参数: a
这是第三个参数: b c
这是第四个参数: 4
这是第五个参数: 5
脚本名称为 test1.sh
4乘以5的值为 20

只有传入的参数个数为5时脚本才能正常运行，避免因为参数个数传入有误造成脚本执行报错。

3.获取所有的参数

[root@shell param]# more test2.sh 
#!/bin/bash
i=1
for param in "$*"
do
echo "参数$i = $param"
i=$[ $i + 1 ]
done

j=1
for param in "$@"
do
echo "参数$j = $param"
j=$[ $j + 1 ]
done
[root@shell param]# ./test2.sh  a b c
参数1 = a b c
参数1 = a
参数2 = b
参数3 = c

$*和$@都可以获取到左右的传入参数，区别是 $*变量会将所有参数当成单个参数，而 $@变量会单独处理每个参数。

4.遍历参数

使用shift命令默认情况下它会将每个参数变量向左移动一个位置，可以使用这个方法遍历所有传入的参数。

[root@shell param]# more test3.sh 
#!/bin/bash
i=1
while [ -n "$1" ]
do
  echo "参数$i = $1"
  i=$[ $i + 1 ]
  shift
done
[root@shell param]# ./test3.sh  a b c
参数1 = a
参数2 = b
参数3 = c

使用while先判断输入的参数1是否存在，如果非空则执行下面的操作：先输出参数$1，然后使用shift将参数$2移动为新的$1，如此循环，直至所有参数被遍历。

5.选项处理

[root@shell param]# more test4.sh
#!/bin/bash

name=$(basename $0)

case $1  in
'start')
        systemctl start docker
        ;;
'stop')
        systemctl stop docker
        ;;
'restart')
        systemctl restart docker
        ;;
'status')
        systemctl status docker
        ;;
*)    
        echo "Usage: sh $name {start|stop|restart|status}"
        exit 3
esac
[root@shell param]# ./test4.sh
Usage: sh test4.sh {start|stop|restart|status}
[root@shell param]# ./test4.sh abc
Usage: sh test4.sh {start|stop|restart|status}
[root@shell param]# ./test4.sh stop
[root@shell param]# ./test4.sh start
[root@shell param]# ./test4.sh status

case 语句会检查每个参数是不是有效选项。如果是的话，就运行对应 case 语句中的命令。

当参数为start、stop、restart和status时执行对应的命令，若为其它参数则提醒需输入正确命令并退出。

6.用户输入

6.1基本读取

使用read可以从标准输入（键盘）或另一个文件描述符中接受输入，在收到输入后， read 命令会将数据放进一个变量。

[root@shell param]# more test5.sh 
#!/bin/bash
read  -n1 -p  "是否继续 Enter[Y/N]? " option

case $option in
Y | y) echo
echo "继续执行下一步: Hello World!";;
N | n) echo
echo "中断执行"
exit;;
esac
[root@shell param]# ./test5.sh 
是否继续 Enter[Y/N]? y
继续执行下一步: Hello World!
[root@shell param]# ./test5.sh 
是否继续 Enter[Y/N]? N
中断执行

-p 选项可以指定提示符“是否继续 Enter[Y/N]?”，-n 选项和值 1 一起使用，告诉 read 命令在接受单个字符后退出。

6.2隐藏读取

[root@shell param]# more test6.sh 
#!/bin/bash
read -t 5 -s -p "Enter your password: " pass
echo
echo "显示输入的密码： $pass "
[root@shell param]# ./test6.sh 
Enter your password: 
显示输入的密码： abc123! 

-s 选项可以避免在 read 命令中输入的数据出现在显示器上（实际上，数据会被显示，只是read 命令会将文本颜色设成跟背景色一样）

-t 选项来指定一个计时器，即5秒钟后没输入退出程序。

6.3文件读取

[root@shell param]# more test7.sh 
#!/bin/bash
i=1
cat myfile | while read myline
do
echo "第$i行: $myline"
i=$[ $i + 1]
done
[root@shell param]# more myfile 
abc123!
123abc!
!123abc
abc!123
123!abc
!abc123
[root@shell param]# ./test7.sh 
第1行: abc123!
第2行: 123abc!
第3行: !123abc
第4行: abc!123
第5行: 123!abc
第6行: !abc123

本例使用 read 命令来读取文件数据，对文件使用 cat 命令，将结果通过管道直接传给含有 read 的 while 命令，也可以使用输出重定向方式，参见shell脚本专题(04)：循环中的“2.批量新增用户”

五、本文总结

参数传入是脚本执行时的一个重要的方式，可以让脚本的执行更加灵活方便，交互的界面使脚本执行更具可读性。本文介绍了三种交互方式：命令行参数、选项和read方式。掌握了这三种方式会让脚本更优雅功能性更强。

 

 

本文所有脚本和安装包文件已上传github：shell-scripts-05

更多请点击：shell专题