☰

如何限制平行作業

tag-icon • tag-icon bash expect
如何限制平行作業

我有一個包含文件的主機/tmp/hostlist，它需要使用者名稱和密碼才能登錄，我正在使用 Expect 命令來登入主機並執行命令，並在我的 bash 腳本中如下所示。

)

VAR=$(expect -c "$script")

echo "$VAR" >/tmp/outexp
-----------------

我能夠處理並行執行 -

while IFS= read -r i
do
        ( export server_name=`echo $i`;echo "connecting to $i";expect -c 
wait

有沒有辦法限制沒有。一次有多少個主機？例如，假設我在一個文件中有 1000 個主機..並且我想一次在一組 100 個主機中執行以完成 1000 個主機..

答案1

使用 GNU Parallel 你會做類似的事情：

[compute $script]
export script
run_one() {
  i="$1"
  export server_name=`echo $i`
  echo "connecting to $i"
  expect -c "$script"
  echo "Job completed on $i"
  echo "-----------------------------------"
}
export -f run_one

cat "$file" | parallel -j100 run_one '2>&1' >"$log2"

或者：

[compute $script]
export script

cat "$file" |
  parallel -j100 --tag 'i={} expect -c "$script" 2>&1' >"$log2"

答案2

您可以檢查有多少作業正在運行並等待它們完成。

這是一個範例腳本：

#!/bin/bash

i=0
while [[ $i -lt 50 ]]; do

    n=$(jobs | grep Running | wc -l)
    if [[ $n -ge 10 ]]; then
        echo "waiting for jobs to finish ($n running)"
        sleep 1
    else
        echo start next $i
        bash -c "sleep $(( $RANDOM % 3 )); echo $i finished" &
        let i+=1
    fi
done
wait

您可以像這樣使用它來將並行主機限制為 100 個：

while IFS= read -r i
do
    n=$(jobs | grep Running | wc -l)
    if [[ $n -ge 100 ]]; then
        echo "waiting for jobs to finish ($n running)"
        sleep 1
    else
        ( export server_name=`echo $i`;echo "connecting to $i";expect -c "$script";echo "Job completed on $i";echo "-----------------------------------" ) >> "${log}_${i}" 2>&1 &
    fi
done < "$file"

答案3

我通常使用一種稍微不同的方法，它不依賴您追蹤生成的子進程的數量，並且只要隊列低於最大值，它就會填滿

max_jobs=100 # set queue size
while IFS= read -r i; do
  num_jobs=$(pgrep -c -P$$)
  if [[ $num_jobs -ge $max_jobs ]]; then
    wait -n $(pgrep -P$$) # Wait until a any subprocess terminates
  else
    <calling_awesome_script_here> &
  fi
done < "$file"

相關內容