Shell简单实现多线程

 
 

一、目的

解决Shell脚本单线程下效率低下的问题

二、适用场景

需要在Linux系统执行同一项命令,但是针对不同的对象,例如PING检测主机,当然可以延展,只要是命令之间不会产生冲突就可以了

三、场景模拟

3.1、我为啥写这个?

学习这个东西的原因是因为朋友需要对多达几百个IP进行Telnet检测,然后询问我能不能写脚本实现,一开始我没考虑太多,直接开始写,脚本内容如下:

#!/bin/bash
##tenlnet扫描
if [ -f "/usr/bin/nc" ];then
echo '开始检测'
else
if [ ! -f "/usr/bin/apt" ];then
apt-get install -y nc
else
if [ -f "/usr/bin/dnf" ];then
dnf install -y nc
else
if [ -f "/usr/bin/yum" ];then
yum install -y nc
else
echo "请自己安装nc"
fi
fi
fi
fi
rm -f disables.txt ips.txt
read -p "请输入IP列表文件路径" File
if [[ -f "$File" ]];then
echo -e "正在获取IP列表"
else
echo "找不到这个文件"
exit 22
fi
read -p "请输入端口,多个请使用空格分开" Drop
ips=`cat $File | wc -l`
echo "一共有 ${ips}"
for((i=1;i<=${ips};i++));do
ip=`cat $File | awk "NR==$i{print }"`
echo -e "正在检测${ip}"
for d in $Drop;do
services="$ip $d"
nc -v -z $services
if [[ "$?" == "0" ]];then
echo -e "$services" >> ips.txt
else
echo "$services" >> disables.txt
fi
done
done
echo "开启的IP列表在 ips.txt"
echo "没有开启的IP列表在disables.txt"

后来发现虽然能实现检测,但是效率太低了,于是就想到Python的多线程处理,然后就去百度一下Shell的多线程,结果真的有方法实现“多线程”,下面开始安排多线程检测IP通讯的脚本(Telnet还没写)。

3.2、多种方式实现“多线程”

3.2.1、第一个方案:命令后台运行+wait命令

之所以第一个写这个方案,是因为我现在只理解这个,其他的方法还在研究中,下面先附上脚本内容吧

#!/bin/bash
# 设置并发的进程数
#echo "请输入线程数"
#read s
rm -f ips.txt disables.txt
read -p "请输入IP列表文件路径" File
if [[ -f "$File" ]];then
echo -e "正在获取IP列表"
else
echo "找不到这个文件"
exit 22
fi
ips=`cat $File | wc -l`
echo -e "当前文件存在${ips}个IP地址"
for((i=1;i<=${ips};i++));do
{
ip=`cat $File | awk "NR==$i{print }"`
sleep 5
ping $ip -c 5 >/dev/null
if [[ "$?" == "0" ]];then
echo -e "$ip" >> ips.txt
else
echo "$ip" >> disables.txt
fi
} &
done
wait
connects=`cat ips.txt | wc -l`
disconnects=`cat disables.txt | wc -l`
echo -e "完成\n存活IP已记录到: ips.txt\n未使用[也可能禁PING]的已记录到:disables.txt"
echo "当前可PING通的IP有: ${connects}"
echo "当前不可PING通的IP有: ${disconnects}"

  

Shell简单实现多线程
 
上一篇:Linux shell模拟多线程拷贝


下一篇:实现断点续传的FTP下载类(支持多线程多任务下载)