shell 实现进程池

[bringtree]

来源

$0	当前脚本的文件名
$n	传递给脚本或函数的参数。n 是一个数字，表示第几个参数。例如，第一个参数是$1，第二个参数是$2。
$#	传递给脚本或函数的参数个数。
$*	传递给脚本或函数的所有参数。
$@	传递给脚本或函数的所有参数。被双引号(" ")包含时，与 $* 稍有不同，下面将会讲到。
$?	上个命令的退出状态，或函数的返回值。
$$	当前Shell进程ID。对于 Shell 脚本，就是这些脚本所在的进程ID。

[bringtree]

来源

[ -a FILE ] 如果 FILE 存在则为真。

[ -b FILE ] 如果 FILE 存在且是一个块特殊文件则为真。

[ -c FILE ] 如果 FILE 存在且是一个字特殊文件则为真。

[ -d FILE ] 如果 FILE 存在且是一个目录则为真。

[ -e FILE ] 如果 FILE 存在则为真。

[ -f FILE ] 如果 FILE 存在且是一个普通文件则为真。

[ -g FILE ] 如果 FILE 存在且已经设置了SGID则为真。

[ -h FILE ] 如果 FILE 存在且是一个符号连接则为真。

[ -k FILE ] 如果 FILE 存在且已经设置了粘制位则为真。

[ -p FILE ] 如果 FILE 存在且是一个名字管道(F如果O)则为真。

[ -r FILE ] 如果 FILE 存在且是可读的则为真。

[ -s FILE ] 如果 FILE 存在且大小不为0则为真。

[ -t FD ] 如果文件描述符 FD 打开且指向一个终端则为真。

[ -u FILE ] 如果 FILE 存在且设置了SUID (set user ID)则为真。

[ -w FILE ] 如果 FILE 如果 FILE 存在且是可写的则为真。

[ -x FILE ] 如果 FILE 存在且是可执行的则为真。

[ -O FILE ] 如果 FILE 存在且属有效用户ID则为真。

[ -G FILE ] 如果 FILE 存在且属有效用户组则为真。

[ -L FILE ] 如果 FILE 存在且是一个符号连接则为真。

[ -N FILE ] 如果 FILE 存在 and has been mod如果ied since it was last read则为真。

[ -S FILE ] 如果 FILE 存在且是一个套接字则为真。

if ! [ -z "$data" ]; then xxxxx fi 当data变量不是一个空字符串的时候，执行 xxxxx [ -z "xxx" ] 的用处是判断 "xxx" 是否为空字符串，加上 ! 就是取“否”的意思。 所以连起来就是当 "$data" 非空时执行if语句。

[ FILE1 -nt FILE2 ] 如果 FILE1 has been changed more recently than FILE2, or 如果 FILE1 exists and FILE2 does not则为真。

[ FILE1 -ot FILE2 ] 如果 FILE1 比 FILE2 要老, 或者 FILE2 存在且 FILE1 不存在则为真。

[ FILE1 -ef FILE2 ] 如果 FILE1 和 FILE2 指向相同的设备和节点号则为真。

[ -o OPTIONNAME ] 如果 shell选项 “OPTIONNAME” 开启则为真。

[ -z STRING ] “STRING” 的长度为零则为真。 字符串为空即NULL时为真。

[ -n STRING ] or [ STRING ] “STRING” 的长度为非零 non-zero则为真。加-n与不加-n结果相同。

[ STRING1 == STRING2 ] 如果2个字符串相同。 “=” may be used instead of “==” for strict POSIX compliance则为真。

[ STRING1 != STRING2 ] 如果字符串不相等则为真。

[ STRING1 < STRING2 ] 如果 “STRING1” sorts before “STRING2” lexicographically in the current locale则为真。

[ STRING1 > STRING2 ] 如果 “STRING1” sorts after “STRING2” lexicographically in the current locale则为真。

[ ARG1 OP ARG2 ] “OP” is one of -eq, -ne, -lt, -le, -gt or -ge. These arithmetic binary operators return true if “ARG1” is equal to, not equal to, less than, less than or equal to, greater than, or greater than or equal to “ARG2”, respectively. “ARG1” and “ARG2” are integers.

[bringtree]

来源 原理 利用 CheckPool中的while 卡住 执行的流程

#!/usr/bin/env bash

#set -euo pipefail
export LANG=en_US.UTF-8
#[ 0 = $UID ] || exit 1
mkdir -p /tmp/agent/results || exit 1
cd /tmp/agent && rm -f results/* || exit 1
# 进程池容量
readonly POOL_SIZE=4
# 进程池数组，存储子进程 pid
POOL=()

# 功能: 显示执行结果
function ShowResult {
    local status=
    cd results
    for task in $(ls); do
        status=$(sed -n '$p' $task)
        if [ 0 -eq $status ]; then
            echo -en "$task > \033[32;1mNORMAL\033[0m > "
        else
            echo -en "$task > \033[31;1mERROR\033[0m > "
        fi
        sed -n -e '$d' -e '1,$p' $task | tr '\n' ' '
        echo -e "\033[0m"
    done
}

# 功能: 执行一个任务
# 根据业务需求自己实现
function ExecTask {
    # for 循环模拟一次任务
    for n in $(seq 1 $1); do
        echo 'a'
        sleep 1
    done &>> results/$1 #记录输出结果
    # 记录命令返回码
    echo $? >> results/$1
}

# 功能: 检查进程池状
# 参数: 一个非负整数
# 参数为 0 时检查进程池是否为空
# 参数为正整数是检查进程池是否已满
function CheckPool {
    while :; do
#        clear
        for pid in ${!POOL[@]}; do
            if [ -d /proc/$pid ]; then
                echo "Handling ${POOL[$pid]} ..."
            else
                #echo "Finished ${POOL[$pid]}"
                unset POOL[$pid]
            fi
        done
        if [ 0 -eq $1 ]; then
            [ 0 -eq ${#POOL[@]} ] && break
        else
            [ $POOL_SIZE -gt ${#POOL[@]} ] && break
        fi
        sleep 0.5
    done
}

# 测试开始
for task in $(seq 11 18); do # 这里的多任务用几个整数模拟
    CheckPool $POOL_SIZE # 检查进程池是否已满
    ExecTask $task & # 后台启动任务
    POOL[$!]="$task" # 记录任务 pid 到进程池
done
CheckPool 0 # 检查进程池是否已空
ShowResult
# 测试结束