执行带超时的shell函数

时间:2012-03-31 09:40:22

标签: bash function shell timeout

为什么会这样做

timeout 10s echo "foo bar" # foo bar

但这不会

function echoFooBar {
  echo "foo bar"
}

echoFooBar # foo bar

timeout 10s echoFooBar # timeout: failed to run command `echoFooBar': No such file or directory

我怎样才能让它发挥作用?

10 个答案:

答案 0 :(得分:40)

timeout是一个命令 - 因此它在你的bash shell的子进程中执行。因此,它无法访问当前shell中定义的函数。

给出的命令timeout作为超时的子进程执行 - 你的shell的一个子进程。

您可能会感到困惑,因为echo既是shell内置命令又是单独命令。

你可以做的是将你的函数放在它自己的脚本文件中,chmod它是可执行的,然后用timeout执行它。

或者fork,在子shell中执行你的函数 - 在原始进程中,监视进度,如果花费太长时间就终止子进程。

答案 1 :(得分:32)

正如Douglas Leeder所说,你需要一个单独的超时过程来发信号。通过将函数导出到子shell并手动运行子shell来解决此问题。

export -f echoFooBar
timeout 10s bash -c echoFooBar

答案 2 :(得分:21)

还有一个内联替代方案还可以启动bash shell的子进程:


timeout 10s bash <<EOT
function echoFooBar {
  echo foo
}

echoFooBar
sleep 20
EOT

答案 3 :(得分:9)

您可以创建一个允许您执行与超时相同但也适用于其他功能的功能:

function run_cmd { 
    cmd="$1"; timeout="$2";
    grep -qP '^\d+$' <<< $timeout || timeout=10

    ( 
        eval "$cmd" &
        child=$!
        trap -- "" SIGTERM 
        (       
                sleep $timeout
                kill $child 2> /dev/null 
        ) &     
        wait $child
    )
}

可以运行如下:

run_cmd "echoFooBar" 10

注意:解决方案来自我的一个问题: Elegant solution to implement timeout for bash commands and functions

答案 4 :(得分:4)

如果您只想添加超时作为整个现有脚本的附加选项,您可以让它测试timeout-option,然后让它在没有该选项的情况下以递归方式调用它。

example.sh:

#!/bin/bash
if [ "$1" == "-t" ]; then
  timeout 1m $0 $2
else
  #the original script
  echo $1
  sleep 2m
  echo YAWN...
fi

在没有超时的情况下运行此脚本:

$./example.sh -other_option # -other_option
                            # YAWN...

运行一分钟超时:

$./example.sh -t -other_option # -other_option

答案 5 :(得分:3)

function foo(){
    for i in {1..100};
    do 
        echo $i;  
        sleep 1;
    done;
}

cat <( foo ) # Will work 
timeout 3 cat <( foo ) # Will Work 
timeout 3 cat <( foo ) | sort # Wont work, As sort will fail 
cat <( timeout 3 cat <( foo ) ) | sort -r # Will Work 

答案 6 :(得分:1)

此功能仅使用内置

  • 也许可以考虑评估&#34; $ *&#34;而不是根据您的需要直接运行$ @

  • 它使用在第一个作为超时值的arg之后指定的命令字符串启动作业并监视作业pid

  • 它每1秒检查一次,bash支持的时间减少到0.01,因此可以调整

  • 此外,如果您的脚本需要stdin,read应该依赖专用的fd(exec {tofd}<> <(:)

  • 此外,您可能需要调整kill信号(循环内部的信号),默认为-15,您可能需要-9

## forking is evil
timeout() {
    to=$1; shift
    $@ & local wp=$! start=0
     while kill -0 $wp; do
        read -t 1
        start=$((start+1))
        if [ $start -ge $to ]; then
            kill $wp && break
        fi
    done
}

答案 7 :(得分:1)

将我对Tiago Lopo的回答的评论放入更具可读性的形式:

我认为在最近的子shell上加上超时更为容易理解,这样我们就不需要评估字符串,整个脚本可以由您喜欢的编辑器突出显示为shell。我只是将带有eval的子shell生成的命令放入shell函数(已通过zsh测试,但应与bash一起使用):

timeout_child () {
    trap -- "" SIGTERM
    child=$!
    timeout=$1
    (
            sleep $timeout
            kill $child
    ) &
    wait $child
}

用法示例:

( while true; do echo -n .; sleep 0.1; done) & timeout_child 2

这样,它也可以与shell函数一起使用(如果它在后台运行):

 print_dots () {
     while true
     do
         sleep 0.1
         echo -n .
     done
 }


 > print_dots & timeout_child 2
 [1] 21725
 [3] 21727
 ...................[1]    21725 terminated  print_dots
 [3]  + 21727 done       ( sleep $timeout; kill $child; )

答案 8 :(得分:0)

我对@Tiago Lopo的答案做了些微修改,可以处理带有多个参数的命令。我还测试了TauPan的解决方案,但是如果您在脚本中多次使用它,则该方法不起作用,而Tiago却可以。

function timeout_cmd { 
  local arr
  local cmd
  local timeout

  arr=( "$@" )

  # timeout: first arg
  # cmd: the other args
  timeout="${arr[0]}"
  cmd=( "${arr[@]:1}" )

  ( 
    eval "${cmd[@]}" &
    child=$!

    echo "child: $child"
    trap -- "" SIGTERM 
    (       
      sleep "$timeout"
      kill "$child" 2> /dev/null 
    ) &     
    wait "$child"
  )
}

这是一个功能齐全的脚本,可以用来测试上述功能:

$ ./test_timeout.sh -h
Usage:
  test_timeout.sh [-n] [-r REPEAT] [-s SLEEP_TIME] [-t TIMEOUT]
  test_timeout.sh -h

Test timeout_cmd function.

Options:
  -n              Dry run, do not actually sleep. 
  -r REPEAT       Reapeat everything multiple times [default: 1].
  -s SLEEP_TIME   Sleep for SLEEP_TIME seconds [default: 5].
  -t TIMEOUT      Timeout after TIMEOUT seconds [default: no timeout].

例如,您像这样进行启动:

$ ./test_timeout.sh -r 2 -s 5 -t 3
Try no: 1
  - Set timeout to: 3
child: 2540
    -> retval: 143
    -> The command timed out
Try no: 2
  - Set timeout to: 3
child: 2593
    -> retval: 143
    -> The command timed out
Done!
#!/usr/bin/env bash

#shellcheck disable=SC2128
SOURCED=false && [ "$0" = "$BASH_SOURCE" ] || SOURCED=true

if ! $SOURCED; then
  set -euo pipefail
  IFS=$'\n\t'
fi

#################### helpers
function check_posint() {
  local re='^[0-9]+$'
  local mynum="$1"
  local option="$2"

  if ! [[ "$mynum" =~ $re ]] ; then
     (echo -n "Error in option '$option': " >&2)
     (echo "must be a positive integer, got $mynum." >&2)
     exit 1
  fi

  if ! [ "$mynum" -gt 0 ] ; then
     (echo "Error in option '$option': must be positive, got $mynum." >&2)
     exit 1
  fi
}
#################### end: helpers

#################### usage
function short_usage() {
  (>&2 echo \
"Usage:
  test_timeout.sh [-n] [-r REPEAT] [-s SLEEP_TIME] [-t TIMEOUT]
  test_timeout.sh -h"
  )
}

function usage() {
  (>&2 short_usage )
  (>&2 echo \
"
Test timeout_cmd function.

Options:
  -n              Dry run, do not actually sleep. 
  -r REPEAT       Reapeat everything multiple times [default: 1].
  -s SLEEP_TIME   Sleep for SLEEP_TIME seconds [default: 5].
  -t TIMEOUT      Timeout after TIMEOUT seconds [default: no timeout].
")
}
#################### end: usage

help_flag=false
dryrun_flag=false
SLEEP_TIME=5
TIMEOUT=-1
REPEAT=1

while getopts ":hnr:s:t:" opt; do
  case $opt in
    h)
      help_flag=true
      ;;    
    n)
      dryrun_flag=true
      ;;
    r)
      check_posint "$OPTARG" '-r'

      REPEAT="$OPTARG"
      ;;
    s)
      check_posint "$OPTARG" '-s'

      SLEEP_TIME="$OPTARG"
      ;;
    t)
      check_posint "$OPTARG" '-t'

      TIMEOUT="$OPTARG"
      ;;
    \?)
      (>&2 echo "Error. Invalid option: -$OPTARG.")
      (>&2 echo "Try -h to get help")
      short_usage
      exit 1
      ;;
    :)
      (>&2 echo "Error.Option -$OPTARG requires an argument.")
      (>&2 echo "Try -h to get help")
      short_usage
      exit 1
      ;;
  esac
done

if $help_flag; then
  usage
  exit 0
fi

#################### utils
if $dryrun_flag; then
  function wrap_run() {
    ( echo -en "[dry run]\\t" )
    ( echo "$@" )
  }
else
  function wrap_run() { "$@"; }
fi

# Execute a shell function with timeout
# https://stackoverflow.com/a/24416732/2377454
function timeout_cmd { 
  local arr
  local cmd
  local timeout

  arr=( "$@" )

  # timeout: first arg
  # cmd: the other args
  timeout="${arr[0]}"
  cmd=( "${arr[@]:1}" )

  ( 
    eval "${cmd[@]}" &
    child=$!

    echo "child: $child"
    trap -- "" SIGTERM 
    (       
      sleep "$timeout"
      kill "$child" 2> /dev/null 
    ) &     
    wait "$child"
  )
}
####################

function sleep_func() {
  local secs
  local waitsec

  waitsec=1
  secs=$(($1))
  while [ "$secs" -gt 0 ]; do
   echo -ne "$secs\033[0K\r"
   sleep "$waitsec"
   secs=$((secs-waitsec))
  done

}

command=("wrap_run" \
         "sleep_func" "${SLEEP_TIME}"
         )

for i in $(seq 1 "$REPEAT"); do
  echo "Try no: $i"

  if [ "$TIMEOUT" -gt 0 ]; then
    echo "  - Set timeout to: $TIMEOUT"
    set +e
    timeout_cmd "$TIMEOUT" "${command[@]}"
    retval="$?"
    set -e

    echo "    -> retval: $retval"
    # check if (retval % 128) == SIGTERM (== 15)
    if [[ "$((retval % 128))" -eq 15 ]]; then
      echo "    -> The command timed out"
    fi
  else
    echo "  - No timeout"
    "${command[@]}"
    retval="$?"
  fi
done

echo "Done!"

exit 0

答案 9 :(得分:0)

这支班轮将在10秒后退出您的Bash会话

$ TMOUT=10 && echo "foo bar"