Script Bash aguarde processos e obtenha código de retorno

13

Estou tentando criar um script que iniciará muitos comandos em segundo plano. Para cada comando em segundo plano, preciso obter o código de retorno.

Eu tenho tentado o seguinte script:

 #!/bin/bash
set -x
pid=()
return=()


for i in 1 2
do
 echo start $i
 ssh mysql "/root/test$i.sh" &
 pid[$i]=$!
done

for i in ${#pid[@]}
do
echo ${pid[$i]}
wait ${pid[$i]}
return[$i]=$?

if [ ${return[$i]} -ne 0 ]
then
  echo mail error
fi

done

echo ${return[1]}
echo ${return[2]}

Meu problema é durante o loop de espera; se o segundo pid terminar antes do primeiro, não poderei obter o código de retorno.

Eu sei que posso executar wait pid1 pid2, mas com este comando não consigo obter o código de retorno de todos os comandos.

Qualquer ideia ?

Hugo
fonte

Respostas:

6

Você pode fazer isso usando um diretório temporário.

# Create a temporary directory to store the statuses
dir=$(mktemp -d)

# Execute the backgrouded code. Create a file that contains the exit status.
# The filename is the PID of this group's subshell.
for i in 1 2; do
    { ssh mysql "/root/test$i.sh" ; echo "$?" > "$dir/$BASHPID" ; } &
done

# Wait for all jobs to complete
wait

# Get return information for each pid
for file in "$dir"/*; do
    printf 'PID %d returned %d\n' "${file##*/}" "$(<"$file")"
done

# Remove the temporary directory
rm -r "$dir"
Chris Down
fonte
9

O problema está mais no seu

for i in ${#pid[@]}

Qual é for i in 2.

Deveria ser:

for i in 1 2

ou

for ((i = 1; i <= ${#pid[@]}; i++))

wait "$pid" irá retornar o código de saída do trabalho com bash(e conchas POSIX, mas não zsh), mesmo se o trabalho já havia terminado quando waitfoi iniciado.

Stéphane Chazelas
fonte
5

Uma implementação genérica sem arquivos temporários.

#!/usr/bin/env bash

## associative array for job status
declare -A JOBS

## run command in the background
background() {
  eval $1 & JOBS[$!]="$1"
}

## check exit status of each job
## preserve exit status in ${JOBS}
## returns 1 if any job failed
reap() {
  local cmd
  local status=0
  for pid in ${!JOBS[@]}; do
    cmd=${JOBS[${pid}]}
    wait ${pid} ; JOBS[${pid}]=$?
    if [[ ${JOBS[${pid}]} -ne 0 ]]; then
      status=${JOBS[${pid}]}
      echo -e "[${pid}] Exited with status: ${status}\n${cmd}"
    fi
  done
  return ${status}
}

background 'sleep 1 ; false'
background 'sleep 3 ; true'
background 'sleep 2 ; exit 5'
background 'sleep 5 ; true'

reap || echo "Ooops! Some jobs failed"
Jon Nalley
fonte
Obrigado :-) Isto é exatamente o que eu estava procurando!
Qorbani
0

A resposta de Stéphane é boa, mas eu preferiria

for i in ${!pid[@]}
do
    wait ${pid[i]}
    return[i]=$?
    unset "pid[$i]"
done

que iterará sobre as chaves da pidmatriz, independentemente de quais entradas ainda existam, para que você possa adaptá-lo, sair do loop e reiniciar o loop inteiro e funcionará. E você não precisa de valores consecutivos ipara começar.

Obviamente, se você estiver lidando com milhares de processos, talvez a abordagem de Stépane seja fracamente mais eficiente quando você tiver uma lista não esparsa.

Martin Kealey
fonte