如何在Zsh中实现被source脚本返回时的代码执行？

Zsh不支持Bash专属的trap <code> RETURN语法，也没有对应的TRAPRETURN函数，但可以通过以下两种方案实现source脚本返回时执行指定代码（比如解锁flock锁）：

方案一：函数封装+函数级EXIT陷阱 #

Zsh中，函数内部设置的trap ... EXIT会在函数返回时触发（而非等待整个shell退出），刚好匹配source脚本执行完毕返回的场景。

示例代码：

# 将脚本逻辑封装为函数
_my_locked_script() {
  # 初始化flock锁
  local lock_fd
  exec {lock_fd}>"/var/run/my_script.lock"
  flock -x "$lock_fd"

  # 设置函数退出时执行的解锁操作
  trap 'flock -u "$lock_fd"; echo "flock锁已解锁"' EXIT

  # 脚本核心业务逻辑
  echo "执行脚本核心操作..."
}

# 调用函数，触发逻辑执行
_my_locked_script

当source该脚本时，函数执行完成返回（即脚本执行结束），陷阱中的解锁代码会自动执行。

方案二：临时precmd钩子（无需封装函数） #

如果不想封装整个脚本为函数，可以借助Zsh的precmd钩子——该钩子会在每次显示命令提示符前执行。我们可以临时添加钩子，执行解锁后立即移除，避免影响后续shell会话。

示例代码：

# 初始化flock锁
local lock_fd
exec {lock_fd}>"/var/run/my_script.lock"
flock -x "$lock_fd"

# 定义返回时要执行的解锁函数
_unlock_flock() {
  flock -u "$lock_fd"
  echo "flock锁已解锁"
  # 移除钩子，避免后续每次提示符都执行
  add-zsh-hook -D precmd _unlock_flock
}

# 添加precmd钩子，确保脚本返回前执行解锁
add-zsh-hook precmd _unlock_flock

# 脚本核心业务逻辑
echo "执行脚本核心操作..."

注意事项 #

• 方案一的陷阱是函数级别的，仅在当前函数返回时触发，不会干扰全局shell的EXIT陷阱
• 方案二必须记得移除钩子，否则每次显示命令提示符都会执行解锁逻辑，引发不必要的操作

内容的提问来源于stack exchange，提问作者Urpagin