115 lines
3.5 KiB
Bash
Executable File
115 lines
3.5 KiB
Bash
Executable File
#!/bin/sh -e
|
|
#
|
|
# run-one - run just one instance at a time of some command and
|
|
# unique set of arguments (useful for cronjobs, eg)
|
|
#
|
|
# run-this-one - kill any identical command/args processes
|
|
# before running this one
|
|
#
|
|
# run-one-constantly - run-one, but respawn every time that one exits
|
|
#
|
|
# run-one-until-success - run-one, but respawn until a successful exit
|
|
#
|
|
# run-one-until-failure - run-one, but respawn until an unsuccessful exit
|
|
#
|
|
# Copyright (C) 2010-2013 Dustin Kirkland <kirkland@ubuntu.com>
|
|
#
|
|
# Authors:
|
|
# Dustin Kirkland <kirkland@ubuntu.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
PROG="run-one"
|
|
|
|
if [ $# -eq 0 ]; then
|
|
echo "ERROR: no arguments specified" 1>&2
|
|
exit 1
|
|
fi
|
|
|
|
# Cache hashes here, to keep one user from DoS'ing another
|
|
# Test if home is writeable and owned by the current user;
|
|
# which is not always the case -- e.g. daemon
|
|
USER=$(id -un)
|
|
for i in "${HOME}" "/dev/shm/${PROG}_${USER}"*; do
|
|
if [ -w "$i" ] && [ -O "$i" ]; then
|
|
DIR="$i"
|
|
break
|
|
fi
|
|
done
|
|
if [ -w "${DIR}" ] && [ -O "${DIR}" ]; then
|
|
DIR="${DIR}/.cache/${PROG}"
|
|
else
|
|
DIR=$(mktemp -d "/dev/shm/${PROG}_${USER}_XXXXXXXX")
|
|
DIR="${DIR}/.cache/${PROG}"
|
|
fi
|
|
mkdir -p "$DIR"
|
|
|
|
# Calculate the hash of the command and arguments
|
|
CMD="$@"
|
|
CMDHASH=$(echo "$CMD" | md5sum | awk '{print $1}')
|
|
FLAG="$DIR/$CMDHASH"
|
|
|
|
base="$(basename $0)"
|
|
case "$base" in
|
|
run-one)
|
|
# Run the specified command, assuming we can flock this command string's hash
|
|
flock -xn "$FLAG" "$@"
|
|
;;
|
|
run-this-one)
|
|
ps="$CMD"
|
|
# Loop through matching pids
|
|
for p in $(pgrep -u "$USER" -f "^$ps$" || true); do
|
|
# Try to kill pid
|
|
kill $p
|
|
# And then block until killed
|
|
while ps $p >/dev/null 2>&1; do
|
|
kill $p
|
|
sleep 1
|
|
done
|
|
done
|
|
# Also try to use lsof, but it seems that flock()'s
|
|
# are not always persistent enough, so this is purely auxilliary.
|
|
pid=$(lsof "$FLAG" 2>/dev/null | awk '{print $2}' | grep "^[0-9]") || true
|
|
[ -z "$pid" ] || kill $pid
|
|
sleep 0.5
|
|
# Run the specified command, assuming we can flock this command string's hash
|
|
flock -xn "$FLAG" "$@"
|
|
;;
|
|
keep-one-running|run-one-constantly|run-one-until-success|run-one-until-failure)
|
|
backoff=0
|
|
retries=0
|
|
while true; do
|
|
# Run the specified command, assuming we can flock this command string's hash
|
|
set +e
|
|
flock -xn "$FLAG" "$@"
|
|
if [ "$?" = 0 ]; then
|
|
# If we were waiting for success, we're done
|
|
[ "$base" = "run-one-until-success" ] && exit $?
|
|
# Last run finished successfully, reset to minimum back-off of 1 second
|
|
backoff=0
|
|
retries=0
|
|
else
|
|
# If we were waiting for failure, we're done
|
|
[ "$base" = "run-one-until-failure" ] && exit $?
|
|
# Last run failed, so slow down the retries
|
|
retries=$((retries + 1))
|
|
backoff=$((retries / 10))
|
|
logger -t "${base}[$$]" "last run failed; sleeping [$backoff] seconds before next run"
|
|
fi
|
|
# Don't sleep more than 60 seconds
|
|
[ $backoff -gt 60 ] && backoff=60
|
|
sleep $backoff
|
|
done
|
|
;;
|
|
esac
|