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

shell脚本专题(05):传参

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

\$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

shell脚本专题(05):传参

只有传入的参数个数为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

shell脚本专题(05):传参

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

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

shell脚本专题(05):传参

使用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

shell脚本专题(05):传参

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
中断执行

shell脚本专题(05):传参

-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! 

shell脚本专题(05):传参

-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

shell脚本专题(05):传参

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

五、本文总结

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

 

 

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

更多请点击:shell专题

shell脚本专题(05):传参

上一篇:使用supervisor进行进程管理


下一篇:shell变量的删除、替换和替代