文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Shell多线程操作及线程数控制实例

2022-06-04 21:37

关注

前言

在业务开发过程中,经常会在后台写一些shell脚本处理数据,但估计很多人不知道shell脚本也可以支持多线程,而且非常简单。本篇文章主要就是介绍shell实现多进程以及进程数量控制。

需求

为了更好的说明问题,我们结合例子讲解,假设需求就是扫描url.txt文件,然后判断里面的URL是否失效。url.txt文件的内容是一行一个URL,如:

http://www.baidu.com

http://www.google.com

http://www.lsjlt.com

单进程实现

那么shell脚本scanUrl.sh可以这样写:

#!/bin/bash

#判断是否有参数

if [ $# != 1 ] ;then

        echo "The parameters you enter is not correct !";

        exit -1;

fi

#循环读出URL并判断状态码 while read line do { isok=`curl -I -o /dev/null -s -w %{http_code} $line` if [ "$isok" = "200" ]; then echo $line "OK" else echo $line "no" fi } done < $1 echo "执行结束"


那么可以执行下面的命令扫描:
/bin/sh scanUrl.sh url.txt

但这样脚本执行非常慢,一万个URL几个小时都扫描不完。

多进程实现


改成多进程实现非常简单,只需要在do后面的大括号加 & 符号,在done后面加一个wait,表示父进程等待子进程退出后再退出

#!/bin/bash

#判断是否有参数

if [ $# != 1 ] ;then

        echo "The parameters you enter is not correct !";

        exit -1;

fi

#循环读出URL并判断状态码 while read line do { isok=`curl -I -o /dev/null -s -w %{http_code} $line` if [ "$isok" = "200" ]; then echo $line "OK" else echo $line "no" fi } }& done < $1 wait echo "执行结束"


这样就能多进程并发执行了,但有个问题是进程会一下子非常多,几百上千,超过系统限制报错,下面我们就加上进程数控制。

多进程实现并控制进程数

#!/bin/bash

#允许的进程数

THREAD_NUM=200

#定义描述符为9的管道

mkfifo tmp

exec 9<>tmp

#预先写入指定数量的换行符,一个换行符代表一个进程

for ((i=0;i<$THREAD_NUM;i++))

do

    echo -ne "n" 1>&9

done

if [ $# != 1 ] ;then echo "The parameters you enter is not correct !"; exit -1; fi

while read line do { #进程控制 read -u 9 { #isok=`curl -I -o /dev/null -s -w %{http_code} $line` if [ "$isok" = "200" ]; then echo $line "OK" else echo $line "no" fi echo -ne "n" 1>&9 }& } done < $1 wait echo "执行结束" rm tmp


上面的代码就可以保证子进程在指定数量了,其进程控制原理是通过管道实现的,当管道无内容可读时就不会执行
{

    #isok=`curl -I -o /dev/null -s -w %{http_code} $line`

    if [ "$isok" = "200" ]; then

        echo $line "OK" 

    else

        echo $line "no" 

    fi

    #写入一个换行符

    echo -ne "n" 1>&9

}&

而且每个进程执行完成后都会向管道写入一个换行符,从而保证进程数是指定的。

这样就能达到我们的目的了。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-服务器
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯