Ruby中超时终止系统调用并设置Mock的解决方案需求

我明白你的痛点——Ruby内置的Timeout模块一超时就抛出异常，直接打断代码流程，根本没法优雅地设置Mock值继续往下走。而且更坑的是，只用Timeout包裹反引号调用的话，超时后Ruby线程被终止，但后台的系统进程还会继续跑，浪费资源。

下面给你两种可靠的实现方案，既能在超时后终止系统进程，又能设置Mock值让代码正常执行到末尾。

方案1：结合Timeout模块+手动管理子进程 #

这个方案利用Timeout的超时检测，同时通过Process.spawn获取子进程PID，确保超时后能彻底终止系统命令：

require 'timeout'

def run_with_timeout(cmd, timeout_sec, mock_value)
  output = nil
  pid = nil
  # 创建管道捕获子进程的标准输出
  stdout_r, stdout_w = IO.pipe

  begin
    Timeout.timeout(timeout_sec) do
      # 启动子进程，重定向输出到管道
      pid = Process.spawn(cmd, out: stdout_w)
      stdout_w.close # 关闭写端，让读端在子进程结束时收到EOF
      # 读取子进程输出
      output = stdout_r.read.strip
      # 等待子进程退出，避免僵尸进程
      Process.waitpid(pid)
    end
  rescue Timeout::Error
    # 超时后终止子进程
    if pid
      begin
        # 先尝试优雅终止（TERM信号），不行再强制杀死（KILL）
        Process.kill(:TERM, pid)
        # 非阻塞等待进程退出
        Process.waitpid(pid, Process::WNOHANG)
      rescue Errno::ESRCH
        # 进程已经提前退出，忽略该错误
      end
    end
    # 设置Mock值
    output = mock_value
  ensure
    # 确保管道被关闭，避免资源泄漏
    stdout_r.close unless stdout_r.closed?
    stdout_w.close unless stdout_w.closed?
  end

  output
end

# 使用示例
a = run_with_timeout('sleep 30; echo 1', 5, 'mock_result')
puts a # 超时输出"mock_result"，正常完成输出"1"

方案1说明： #

• 用管道捕获子进程的输出，替代反引号的同步调用，这样能同时获取PID和输出内容。
• 超时捕获Timeout::Error后，立即终止子进程，避免它在后台继续运行。
• ensure块负责清理管道资源，防止内存泄漏。

方案2：手动线程管理（不依赖Timeout模块） #

如果你不想用Timeout模块，也可以直接用Ruby线程实现超时逻辑，可控性更强：

def run_with_timeout(cmd, timeout_sec, mock_value)
  output = nil
  pid = nil
  stdout_r, stdout_w = IO.pipe

  # 创建线程执行系统调用
  worker_thread = Thread.new do
    pid = Process.spawn(cmd, out: stdout_w)
    stdout_w.close
    output = stdout_r.read.strip
    Process.waitpid(pid)
  end

  # 主线程等待worker线程，最多等待timeout_sec秒
  if worker_thread.join(timeout_sec).nil?
    # 超时：杀死线程和子进程
    worker_thread.kill
    if pid
      begin
        Process.kill(:TERM, pid)
        Process.waitpid(pid, Process::WNOHANG)
      rescue Errno::ESRCH
        # 忽略进程已退出的错误
      end
    end
    output = mock_value
  end

  # 清理管道资源
  stdout_r.close unless stdout_r.closed?
  stdout_w.close unless stdout_w.closed?

  output
end

# 使用示例
a = run_with_timeout('sleep 30; echo 1', 5, 'mock_result')
puts a

方案2说明： #

• 主线程通过Thread.join(timeout_sec)等待工作线程完成，如果超时则返回nil，此时我们就可以终止线程和子进程。
• 逻辑和方案1类似，但完全脱离Timeout模块，避免一些Timeout可能带来的边缘问题（比如某些情况下线程无法被正确终止）。

关键注意事项 #

• 一定要终止子进程：如果只终止Ruby线程，后台的系统进程会继续运行，导致资源浪费甚至逻辑混乱。
• 处理异常：杀死进程时要捕获Errno::ESRCH，因为子进程可能在我们操作前就已经退出了。
• 资源清理：管道、线程等资源要及时关闭/清理，避免内存泄漏。
• 优雅终止优先：先发送TERM信号让进程优雅退出，必要时再用KILL强制杀死。

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