extern/starship
1
0
Fork 0
mirror of https://github.com/starship/starship.git synced 2024-11-25 09:43:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

perf: use built-in EPOCHREALTIME env-var to compute cmd_duration for ZSH5+ (#1751)

Browse Source 
We currently invoke `starship time` in a sub-shell to compute time, which is non-performant. By using $EPOCHREALTIME,
which is an inbuilt env-var in ZSH5, we can get better performance. This commit only targets ZSH5+ and maintains the
old behaviour.
This commit is contained in:
Gautham Warrier 2021-01-23 15:55:41 +05:30 committed by GitHub
parent bcaf835321
commit 5eacd3abee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/init/starship.zsh
View File

@ -16,6 +16,20 @@ starship_render() {
PROMPT="$(::STARSHIP:: prompt --keymap="${KEYMAP-}" --status=$STARSHIP_CMD_STATUS --cmd-duration=${STARSHIP_DURATION-} --jobs="$NUM_JOBS")" PROMPT="$(::STARSHIP:: prompt --keymap="${KEYMAP-}" --status=$STARSHIP_CMD_STATUS --cmd-duration=${STARSHIP_DURATION-} --jobs="$NUM_JOBS")"
} }
# Defines a function `__starship_get_time` that sets the time since epoch in millis in STARSHIP_CAPTURED_TIME.
if [[ $ZSH_VERSION == ([1-4]*) ]]; then
# ZSH <= 5; Does not have a built-in variable so we will rely on Starship's inbuilt time function.
__starship_get_time() {
STARSHIP_CAPTURED_TIME=$(::STARSHIP:: time)
}
else
zmodload zsh/datetime
zmodload zsh/mathfunc
__starship_get_time() {
((STARSHIP_CAPTURED_TIME = int(rint($EPOCHREALTIME * 1000))))
}
fi
# Will be run before every prompt draw # Will be run before every prompt draw
starship_precmd() { starship_precmd() {
# Save the status, because commands in this pipeline will change $? # Save the status, because commands in this pipeline will change $?
@ -24,8 +38,8 @@ starship_precmd() {
# Compute cmd_duration, if we have a time to consume, otherwise clear the # Compute cmd_duration, if we have a time to consume, otherwise clear the
# previous duration # previous duration
if [[ -n "${STARSHIP_START_TIME+1}" ]]; then if [[ -n "${STARSHIP_START_TIME+1}" ]]; then
STARSHIP_END_TIME=$(::STARSHIP:: time) __starship_get_time && STARSHIP_END_TIME=$STARSHIP_CAPTURED_TIME
STARSHIP_DURATION=$((STARSHIP_END_TIME - STARSHIP_START_TIME)) ((STARSHIP_DURATION = STARSHIP_END_TIME - STARSHIP_START_TIME))
unset STARSHIP_START_TIME unset STARSHIP_START_TIME
else else
unset STARSHIP_DURATION unset STARSHIP_DURATION
@ -35,7 +49,7 @@ starship_precmd() {
starship_render starship_render
} }
starship_preexec() { starship_preexec() {
STARSHIP_START_TIME=$(::STARSHIP:: time) __starship_get_time && STARSHIP_START_TIME=$STARSHIP_CAPTURED_TIME
} }
# If precmd/preexec arrays are not already set, set them. If we don't do this, # If precmd/preexec arrays are not already set, set them. If we don't do this,
@ -74,7 +88,8 @@ else
zle -N zle-keymap-select starship_zle-keymap-select-wrapped; zle -N zle-keymap-select starship_zle-keymap-select-wrapped;
fi fi
STARSHIP_START_TIME=$(::STARSHIP:: time) __starship_get_time && STARSHIP_START_TIME=$STARSHIP_CAPTURED_TIME
export STARSHIP_SHELL="zsh" export STARSHIP_SHELL="zsh"
# Set up the session key that will be used to store logs # Set up the session key that will be used to store logs