- [root@station1 ~]# cat a.sh
- #!/bin/bash
- for((i=0;i<20;i++))
- do
- sleep 1
- echo "bingfa"
- done
- [root@station1 ~]# time bash a.sh
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- real 0m20.067s
- user 0m0.016s
- sys 0m0.031s
- [root@station1 ~]#
- [root@station1 ~]# cat b.sh
- #!/bin/bash
- for((i=0;i<20;i++))
- do
- {
- sleep 1
- echo "bingfa"
- }&
- done
- wait
- [root@station1 ~]# time bash b.sh
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- bingfa
- real 0m1.060s
- user 0m0.005s
- sys 0m0.057s
- [root@station1 ~]#
- [root@station1 ~]# cat c.sh
- #!/bin/bash
- exec 6<>tmpfile
- echo "1\n1\n1" &>6
- for((i=0;i<20;i++))
- do
- read -u 6
- {
- sleep 1
- echo "$REPLY"
- echo "1" 1>&6
- }&
- done
- wait
- [root@station1 ~]# time bash c.sh
- 111
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- real 0m1.074s
- user 0m0.012s
- sys 0m0.031s
- [root@station1 ~]#
- [root@station1 ~]# cat d.sh
- #!/bin/bash
- mkfifo fd2
- exec 9<>fd2
- echo -n -e "1\n1\n1\n" 1>&9
- for((i=0;i<20;i++))
- do
- read -u 9
- { #your process
- sleep 1
- echo "$REPLY"
- echo -ne "1\n" 1>&9
- } &
- done
- wait
- rm -f fd2
- [root@station1 ~]# time bash d.sh
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- 1
- real 0m7.075s
- user 0m0.018s
- sys 0m0.044s
- [root@station1 ~]#
shell如何实现多线程?
情景
shell脚本的执行效率虽高,但当任务量巨大时仍然需要较长的时间,尤其是需要执行一大批的命令时。因为默认情况下,shell脚本中的命令是串行执行的。如果这些命令相互之间是独立的,则可以使用“并发”的方式执行这些命令,这样可以更好地利用系统资源,提升运行效率,缩短脚本执行的时间。如果命令相互之间存在交互,则情况就复杂了,那么不建议使用shell脚本来完成多线程的实现。
为了方便阐述,使用一段测试代码。在这段代码中,通过seq
命令输出1到10,使用for...in
语句产生一个执行10次的循环。每一次循环都执行sleep 1
,并echo
出当前循环对应的数字。
注意:
- 真实的使用场景下,循环次数不一定等于10,或高或低,具体取决于实际的需求。
- 真实的使用场景下,循环体内执行的语句往往比较耗费系统资源,或比较耗时等。
请根据真实场景的各种情况理解本文想要表达的内容。
$ cat test1.sh
#/bin/bashall_num=10a=$(date +%H%M%S)for num in `seq 1 ${all_num}`do sleep 1 echo ${num}doneb=$(date +%H%M%S)echo -e "startTime:\t$a"echo -e "endTime:\t$b"
通过上述代码可知,为了体现执行的时间,将循环体开始前后的时间打印了出来。
运行结果:
$ sh test1.sh
12345678910startTime: 193649endTime: 193659
10次循环,每次sleep 1秒,所以总执行时间10s。
方案
方案1:使用"&"使命令后台运行
在linux中,在命令的末尾加上&
符号,则表示该命令将在后台执行,这样后面的命令不用等待前面的命令执行完就可以开始执行了。示例中的循环体内有多条命令,则可以以{}
括起来,在大括号后面添加&
符号。
$ cat test2.sh
#/bin/bashall_num=10a=$(date +%H%M%S)for num in `seq 1 ${all_num}`do{ sleep 1 echo ${num}} &doneb=$(date +%H%M%S)echo -e "startTime:\t$a"echo -e "endTime:\t$b"
运行结果:
sh test2.sh
startTime: 194147endTime: 194147[j-tester@merger142 ~/bin/multiple_process]$ 12345678910
通过结果可知,程序没有先打印数字,而是直接输出了开始和结束时间,然后显示出了命令提示符[j-tester@merger142 ~/bin/multiple_process]$
(出现命令提示符表示脚本已运行完毕),然后才是数字的输出。这是因为循环体内的命令全部进入后台,所以均在sleep了1秒以后输出了数字。开始和结束时间相同,即循环体的执行时间不到1秒钟,这是由于循环体在后台执行,没有占用脚本主进程的时间。
方案2:命令后台运行+wait
命令
解决上面的问题,只需要在上述循环体的done语句后面加上wait
命令,该命令等待当前脚本进程下的子进程结束,再运行后面的语句。
$ cat test3.sh
#/bin/bashall_num=10a=$(date +%H%M%S)for num in `seq 1 ${all_num}`do{ sleep 1 echo ${num}} &donewaitb=$(date +%H%M%S)echo -e "startTime:\t$a"echo -e "endTime:\t$b"
运行结果:
$ sh test3.sh
12345679810startTime: 194221endTime: 194222
但这样依然存在一个问题:
因为&
使得所有循环体内的命令全部进入后台运行,那么倘若循环的次数很多,会使操作系统在瞬间创建出所有的子进程,这会非常消耗系统的资源。如果循环体内的命令又很消耗系统资源,则结果可想而知。 最好的方法是并发的进程是可配置的。
方案3:使用文件描述符控制并发数
$ cat test4.sh
#/bin/bashall_num=10# 设置并发的进程数thread_num=5a=$(date +%H%M%S)# mkfifotempfifo="my_temp_fifo"mkfifo ${tempfifo}# 使文件描述符为非阻塞式exec 6<>${tempfifo}rm -f ${tempfifo}# 为文件描述符创建占位信息for ((i=1;i<=${thread_num};i++))do{ echo }done >&6 # for num in `seq 1 ${all_num}`do{ read -u6 { sleep 1 echo ${num} echo "" >&6 } & } done wait# 关闭fd6管道exec 6>&-b=$(date +%H%M%S)echo -e "startTime:\t$a"echo -e "endTime:\t$b"
运行结果:
$ sh test4.sh
13245678910startTime: 195227endTime: 195229
方案4:使用xargs -P
控制并发数
xargs命令有一个-P
参数,表示支持的最大进程数,默认为1。为0时表示尽可能地大,即方案2
的效果。
$ cat test5.sh
#/bin/bashall_num=10thread_num=5a=$(date +%H%M%S)seq 1 ${all_num} | xargs -n 1 -I {} -P ${thread_num} sh -c "sleep 1;echo {}"b=$(date +%H%M%S)echo -e "startTime:\t$a"echo -e "endTime:\t$b"
运行结果:
$ sh test5.sh
12345687910startTime: 195257endTime: 195259
方案5:使用GNU parallel
命令控制并发数
GNU parallel
命令是非常强大的并行计算命令,使用-j
参数控制其并发数量。
$ cat test6.sh
#/bin/bashall_num=10thread_num=6a=$(date +%H%M%S)parallel -j 5 "sleep 1;echo {}" ::: `seq 1 10`b=$(date +%H%M%S)echo -e "startTime:\t$a"echo -e "endTime:\t$b"
运行结果:
$ sh test6.sh
12345678910startTime: 195616endTime: 195618
总结
“多线程”的好处不言而喻,虽然shell中并没有真正的多线程,但上述解决方案可以实现“多线程”的效果,重要的是,在实际编写脚本时应有这样的考虑和实现。
另外:方案3、4、5虽然都可以控制并发数量,但方案3显然写起来太繁琐。方案4和5都以非常简洁的形式完成了控制并发数的效果,但由于方案5的parallel命令非常强大,所以十分建议系统学习下。方案3、4、5设置的并发数均为5,实际编写时可以将该值作为一个参数传入。参考文章
相关知识点
- wait命令
&
后台运行- 文件描述符、mkfifo等
- xargs命令
- parallel命令
一个入门级可控多线程shell脚本方案
说到shell可控多线程,网上分享的大部分是管道控制的方案。这种方案,张戈博客也曾经实战并分享过一次:《》,感兴趣的朋友可以看看。
下面张戈博客再分享另一种更容易理解的入门级可控多线程shell脚本方案:任务切割、各个击破。
先来 1 段场景描述:
某日,在鹅厂接到了这个任务,需要在Linux服务器中,对几千个IP进行一次Ping检测,只要取得ping可达的IP就好。如果单个IP去ping测试,虽然也可以完成任务,几千个IP还好了,如果更多呢?
鉴于这个case简单程度,第一时间先放弃了以前用过的管道方案,而是采用了各个击破的思想。
简单思路:
按照任务切割的“战略思想”,我先将这几千IP存入一个iplist文件,然后写一个分割函数,将这个文件分成多份临时IP清单,最后,用多线程遍历这些临时IP文件即可变相实现多线程了。
具体代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | #!/bin/sh #文本分割函数:将文本$1按份数$2进行分割 SplitFile() { linenum=`wc -l $1 |awk '{print $1}'` if [[ $linenum -le $2 ]] then echo "The lines of this file is less then $2, Are you kidding me..." exit fi Split=`expr $linenum / $2` Num1=1 FileNum=1 test -d SplitFile || mkdir -p SplitFile rm -rf SplitFile/* while [ $Num1 -lt $linenum ] do Num2=`expr $Num1 + $Split` sed -n "${Num1}, ${Num2}p " $1 > SplitFile/$1-$FileNum Num1=`expr $Num2 + 1` FileNum=`expr $FileNum + 1` done } #Define some variables SPLIT_NUM=${ 1:-10} #参数1表示分割成多少份即,开启多少个线程,默认10个 FILE=${ 2:-iplist} #参数2表示分割的对象,默认iplist文件 #分割文件 SplitFile $FILE $SPLIT_NUM #循环遍历临时IP文件 for iplist in $(ls ./SplitFile/*) do #循环ping测试临时IP文件中的ip(丢后台) cat $iplist | while read ip do ping -c 4 -w 4 $ip >/dev/null && echo $ip | tee -ai okip.log #ping 可达的IP则写入日志 done & #在while循环后面加上&符号,让这个嵌套循环在后台执行 done |
将代码保存为ping.sh之后,执行 sh ping.sh iplist 100 的过程如下:
先将iplist切割成100份,存放在 SplitFile 文件夹中
然后,通过for循环读取这些分割文件,并在后台使用while循环对其中ip执行ping命令。
由于while是丢后台的, 所以for循环会一次性执行100个while,相当于开启了100个线程,速度自然不可同日而语矣。
其中,切割的份数即你想要开启的多线程数量,很明显,这种任务分割的思路虽然没有管道方案来的高大上,但是其思想更加简单易懂,而且通用性也更好,适合入门级的简单多线程任务。