Використання пастки - це не завжди варіант. Наприклад, якщо ви пишете якусь функцію повторного використання, яка потребує обробки помилок, і яку можна викликати з будь-якого сценарію (після отримання файлу допоміжними функціями), ця функція не може припускати нічого про час виходу зовнішнього скрипту, що робить використання пасток дуже важким. Ще одним недоліком використання пасток є погана компостування, оскільки ви ризикуєте перезаписати попередню пастку, яка може бути встановлена раніше у ланцюжку абонентів.
Існує невелика хитрість, яку можна використовувати для правильного поводження з помилками без пасток. Як ви вже можете дізнатися з інших відповідей, set -e
це не працює всередині команд, якщо ви використовуєте ||
оператор після них, навіть якщо ви запускаєте їх у підшалі; наприклад, це не працює:
#!/bin/sh
# prints:
#
# --> outer
# --> inner
# ./so_1.sh: line 16: some_failed_command: command not found
# <-- inner
# <-- outer
set -e
outer() {
echo '--> outer'
(inner) || {
exit_code=$?
echo '--> cleanup'
return $exit_code
}
echo '<-- outer'
}
inner() {
set -e
echo '--> inner'
some_failed_command
echo '<-- inner'
}
outer
Але ||
оператор потрібен для запобігання поверненню із зовнішньої функції перед очищенням. Хитрість полягає в тому, щоб запустити внутрішню команду у фоновому режимі, а потім негайно дочекатися її. wait
Вбудований повертають код завершення внутрішньої команди, і тепер ви використовуєте ||
після того wait
, а не внутрішньою функції, тому set -e
працює належним чином всередині останній:
#!/bin/sh
# prints:
#
# --> outer
# --> inner
# ./so_2.sh: line 27: some_failed_command: command not found
# --> cleanup
set -e
outer() {
echo '--> outer'
inner &
wait $! || {
exit_code=$?
echo '--> cleanup'
return $exit_code
}
echo '<-- outer'
}
inner() {
set -e
echo '--> inner'
some_failed_command
echo '<-- inner'
}
outer
Ось загальна функція, яка ґрунтується на цій ідеї. Він повинен працювати у всіх сумісних з POSIX оболонках, якщо ви вилучите local
ключові слова, тобто замініть всі local x=y
просто x=y
:
# [CLEANUP=cleanup_cmd] run cmd [args...]
#
# `cmd` and `args...` A command to run and its arguments.
#
# `cleanup_cmd` A command that is called after cmd has exited,
# and gets passed the same arguments as cmd. Additionally, the
# following environment variables are available to that command:
#
# - `RUN_CMD` contains the `cmd` that was passed to `run`;
# - `RUN_EXIT_CODE` contains the exit code of the command.
#
# If `cleanup_cmd` is set, `run` will return the exit code of that
# command. Otherwise, it will return the exit code of `cmd`.
#
run() {
local cmd="$1"; shift
local exit_code=0
local e_was_set=1; if ! is_shell_attribute_set e; then
set -e
e_was_set=0
fi
"$cmd" "$@" &
wait $! || {
exit_code=$?
}
if [ "$e_was_set" = 0 ] && is_shell_attribute_set e; then
set +e
fi
if [ -n "$CLEANUP" ]; then
RUN_CMD="$cmd" RUN_EXIT_CODE="$exit_code" "$CLEANUP" "$@"
return $?
fi
return $exit_code
}
is_shell_attribute_set() { # attribute, like "x"
case "$-" in
*"$1"*) return 0 ;;
*) return 1 ;;
esac
}
Приклад використання:
#!/bin/sh
set -e
# Source the file with the definition of `run` (previous code snippet).
# Alternatively, you may paste that code directly here and comment the next line.
. ./utils.sh
main() {
echo "--> main: $@"
CLEANUP=cleanup run inner "$@"
echo "<-- main"
}
inner() {
echo "--> inner: $@"
sleep 0.5; if [ "$1" = 'fail' ]; then
oh_my_god_look_at_this
fi
echo "<-- inner"
}
cleanup() {
echo "--> cleanup: $@"
echo " RUN_CMD = '$RUN_CMD'"
echo " RUN_EXIT_CODE = $RUN_EXIT_CODE"
sleep 0.3
echo '<-- cleanup'
return $RUN_EXIT_CODE
}
main "$@"
Запуск прикладу:
$ ./so_3 fail; echo "exit code: $?"
--> main: fail
--> inner: fail
./so_3: line 15: oh_my_god_look_at_this: command not found
--> cleanup: fail
RUN_CMD = 'inner'
RUN_EXIT_CODE = 127
<-- cleanup
exit code: 127
$ ./so_3 pass; echo "exit code: $?"
--> main: pass
--> inner: pass
<-- inner
--> cleanup: pass
RUN_CMD = 'inner'
RUN_EXIT_CODE = 0
<-- cleanup
<-- main
exit code: 0
Єдине, про що потрібно знати, використовуючи цей метод, - це те, що всі модифікації змінних Shell, виконані з команди, яку ви передаєте, run
не поширюватимуться на функцію виклику, оскільки команда працює в підзарядці.