如何在Ruby的ensure块中仅在脚本正常退出（exit 0）时输出消息

要实现仅在脚本正常退出（exit 0）时输出指定消息，你可以通过以下几种方式实现：

方法一：标记变量追踪正常退出 #

在脚本中定义一个标记变量，所有正常退出的节点主动将标记设为true，在ensure块中检查该标记：

normal_exit = false

begin
  # 业务逻辑示例
  puts "执行业务逻辑"
  # 正常退出时设置标记
  normal_exit = true
  exit 0

  # 其他可能的正常退出点
  # normal_exit = true
  # exit 0
rescue => e
  # 异常时可选择处理或直接抛出，不设置标记
  puts "发生异常: #{e.message}"
  raise e
ensure
  puts 'my message' if normal_exit
end

原理：只有主动触发正常退出时才会设置normal_exit为true，异常崩溃时标记保持false，ensure块根据标记判断是否输出消息。

方法二：利用at_exit钩子结合全局状态 #

Ruby的at_exit钩子会在进程退出时执行，结合全局变量$!（记录最后一个异常）和$?.exitstatus（退出状态码）判断是否为正常退出：

at_exit do
  # $!.nil? 表示无异常导致退出，$?.exitstatus == 0 表示退出状态码为0
  if $!.nil? && $?.exitstatus == 0
    puts 'my message'
  end
end

# 测试场景：
# exit 0 # 会输出消息
# exit 1 # 不会输出
# raise "脚本崩溃" # 不会输出

原理：当脚本因异常崩溃时，$!会包含对应的异常对象；主动调用exit时$!为nil，再通过$?.exitstatus确认退出状态码是否为0。

方法三：重写exit方法记录退出状态 #

通过重写exit方法记录退出状态，在ensure块中结合异常状态判断：

# 保存原始exit方法
original_exit = Kernel.method(:exit)

def exit(status = 0)
  # 全局变量记录退出状态
  $exit_status = status
  original_exit.call(status)
end

begin
  # exit 0 # 正常退出，会输出消息
  # exit 1 # 非0退出，不输出
  # raise "错误" # 异常崩溃，不输出
ensure
  # $!.nil? 确保不是异常退出，$exit_status == 0 确保状态码为0
  puts 'my message' if $!.nil? && $exit_status == 0
end

原理：重写exit方法后，所有主动退出都会记录状态码，ensure块同时检查是否有异常、退出状态码是否为0，满足条件才输出消息。

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