Dmitry Burmistrov | 7939e7f | 2016-06-06 12:20:17 +0300 | [diff] [blame] | 1 | #!/bin/bash |
| 2 | |
| 3 | cleanup_and_exit() |
| 4 | { |
| 5 | trap EXIT |
Dmitry Burmistrov | 81860c1 | 2016-06-06 16:03:12 +0300 | [diff] [blame] | 6 | exit "${1:-0}" |
Dmitry Burmistrov | 7939e7f | 2016-06-06 12:20:17 +0300 | [diff] [blame] | 7 | } |
| 8 | |
| 9 | fail_exit() |
| 10 | { |
| 11 | echo "$@" |
| 12 | cleanup_and_exit 1 |
| 13 | } |
| 14 | |
| 15 | job_lock() { |
| 16 | [ -z "$1" ] && fail_exit "Lock file is not specified" |
| 17 | local LOCKFILE=$1 |
| 18 | shift |
| 19 | local fd=1000 |
| 20 | eval "exec $fd>>$LOCKFILE" |
| 21 | case $1 in |
| 22 | "set") |
Dmitry Burmistrov | 81860c1 | 2016-06-06 16:03:12 +0300 | [diff] [blame] | 23 | flock -x -n "$fd" \ |
Dmitry Burmistrov | 7939e7f | 2016-06-06 12:20:17 +0300 | [diff] [blame] | 24 | || fail_exit "Process already running. Lockfile: $LOCKFILE" |
| 25 | ;; |
| 26 | "unset") |
Dmitry Burmistrov | 81860c1 | 2016-06-06 16:03:12 +0300 | [diff] [blame] | 27 | flock -u "$fd" |
| 28 | [ -f "$LOCKFILE" ] && rm -f "$LOCKFILE" |
Dmitry Burmistrov | 7939e7f | 2016-06-06 12:20:17 +0300 | [diff] [blame] | 29 | ;; |
| 30 | "wait") |
| 31 | local TIMEOUT=${2:-3600} |
Dmitry Burmistrov | 81860c1 | 2016-06-06 16:03:12 +0300 | [diff] [blame] | 32 | [ "$IS_VERBOSE" == "true" ] \ |
| 33 | && echo "[INFO] Waiting of concurrent process (lockfile: $LOCKFILE, timeout = $TIMEOUT seconds) ..." |
| 34 | flock -x -w "$TIMEOUT" "$fd" \ |
Dmitry Burmistrov | 7939e7f | 2016-06-06 12:20:17 +0300 | [diff] [blame] | 35 | || fail_exit "Timeout error (lockfile: $LOCKFILE)" |
Dmitry Burmistrov | 81860c1 | 2016-06-06 16:03:12 +0300 | [diff] [blame] | 36 | [ "${IS_VERBOSE}" == "true" ] && echo "[INFO] Done" |
Dmitry Burmistrov | 7939e7f | 2016-06-06 12:20:17 +0300 | [diff] [blame] | 37 | ;; |
| 38 | esac |
| 39 | } |
| 40 | |
| 41 | trap fail_exit EXIT |