mirror of
https://github.com/pyenv/pyenv.git
synced 2024-11-21 20:47:00 -05:00
Experimental implementation to wait rehash until acquiring lock
For now the code is using traditional pseudo locking mechanism based on `noclobber`.
This commit is contained in:
parent
63c4b7f45c
commit
a6c15fb242
1 changed files with 38 additions and 18 deletions
|
@ -10,29 +10,49 @@ PROTOTYPE_SHIM_PATH="${SHIM_PATH}/.pyenv-shim"
|
||||||
# Create the shims directory if it doesn't already exist.
|
# Create the shims directory if it doesn't already exist.
|
||||||
mkdir -p "$SHIM_PATH"
|
mkdir -p "$SHIM_PATH"
|
||||||
|
|
||||||
# Ensure only one instance of pyenv-rehash is running at a time by
|
acquire_lock() {
|
||||||
# setting the shell's `noclobber` option and attempting to write to
|
# Ensure only one instance of pyenv-rehash is running at a time by
|
||||||
# the prototype shim file. If the file already exists, print a warning
|
# setting the shell's `noclobber` option and attempting to write to
|
||||||
# to stderr and exit with a non-zero status.
|
# the prototype shim file. If the file already exists, print a warning
|
||||||
set -o noclobber
|
# to stderr and exit with a non-zero status.
|
||||||
{ echo > "$PROTOTYPE_SHIM_PATH"
|
local ret
|
||||||
} 2>| /dev/null ||
|
set -o noclobber
|
||||||
{ if [ -w "$SHIM_PATH" ]; then
|
echo > "$PROTOTYPE_SHIM_PATH" 2>| /dev/null || ret=1
|
||||||
|
set +o noclobber
|
||||||
|
[ -z "${ret}" ]
|
||||||
|
}
|
||||||
|
|
||||||
|
# If we were able to obtain a lock, register a trap to clean up the
|
||||||
|
# prototype shim when the process exits.
|
||||||
|
trap release_lock EXIT
|
||||||
|
|
||||||
|
remove_prototype_shim() {
|
||||||
|
rm -f "$PROTOTYPE_SHIM_PATH"
|
||||||
|
}
|
||||||
|
|
||||||
|
release_lock() {
|
||||||
|
remove_prototype_shim
|
||||||
|
}
|
||||||
|
|
||||||
|
unset acquired
|
||||||
|
for _ in $(seq "${PYENV_REHASH_LOCK_TIMEOUT:-60}"); do
|
||||||
|
if acquire_lock 2>/dev/null; then
|
||||||
|
acquired=1
|
||||||
|
break
|
||||||
|
else
|
||||||
|
# POSIX sleep(1) doesn't provides time precision of subsecond
|
||||||
|
sleep 1
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if [ -z "${acquired}" ]; then
|
||||||
|
if [ -w "$SHIM_PATH" ]; then
|
||||||
echo "pyenv: cannot rehash: $PROTOTYPE_SHIM_PATH exists"
|
echo "pyenv: cannot rehash: $PROTOTYPE_SHIM_PATH exists"
|
||||||
else
|
else
|
||||||
echo "pyenv: cannot rehash: $SHIM_PATH isn't writable"
|
echo "pyenv: cannot rehash: $SHIM_PATH isn't writable"
|
||||||
fi
|
fi
|
||||||
exit 1
|
exit 1
|
||||||
} >&2
|
fi
|
||||||
set +o noclobber
|
|
||||||
|
|
||||||
# If we were able to obtain a lock, register a trap to clean up the
|
|
||||||
# prototype shim when the process exits.
|
|
||||||
trap remove_prototype_shim EXIT
|
|
||||||
|
|
||||||
remove_prototype_shim() {
|
|
||||||
rm -f "$PROTOTYPE_SHIM_PATH"
|
|
||||||
}
|
|
||||||
|
|
||||||
# The prototype shim file is a script that re-execs itself, passing
|
# The prototype shim file is a script that re-execs itself, passing
|
||||||
# its filename and any arguments to `pyenv exec`. This file is
|
# its filename and any arguments to `pyenv exec`. This file is
|
||||||
|
|
Loading…
Reference in a new issue