如何使用 Ruby 的 Net::SSH 库获取退出状态？

Question

我有一段代码，只是尝试在远程服务器上执行脚本，如果失败，我想进行后续调用，想象一下：

require 'rubygems'
require 'net/ssh'
require 'etc'

server = 'localhost'

Net::SSH.start(server, Etc.getlogin) do |ssh|
  puts (ssh.exec("true")  ? 'Exit Success' : "Exit Failure")
  puts (ssh.exec("false") ? 'Exit Success' : "Exit Failure")  
end

我希望（忽略 stdout 和 stderr在我设计的示例中打印） - 但第一行应该以 0 退出，我希望 Ruby 会解释为 false 并显示“退出失败”（当然，所以逻辑是错误，三元需要翻转） - 但第二行应该以相反的状态退出，但事实并非如此。

我什至在文档中找不到有关如何执行此操作的任何内容，我有点担心我可能做错了？！

Answer 1

我发现以下使用 Net::SSH 运行进程的方法更有用。它为您提供了不同的stdout和stderr、退出代码和退出信号。

require 'rubygems'
require 'net/ssh'
require 'etc'

server = 'localhost'

def ssh_exec!(ssh, command)
  stdout_data = ""
  stderr_data = ""
  exit_code = nil
  exit_signal = nil
  ssh.open_channel do |channel|
    channel.exec(command) do |ch, success|
      unless success
        abort "FAILED: couldn't execute command (ssh.channel.exec)"
      end
      channel.on_data do |ch,data|
        stdout_data+=data
      end

      channel.on_extended_data do |ch,type,data|
        stderr_data+=data
      end

      channel.on_request("exit-status") do |ch,data|
        exit_code = data.read_long
      end

      channel.on_request("exit-signal") do |ch, data|
        exit_signal = data.read_long
      end
    end
  end
  ssh.loop
  [stdout_data, stderr_data, exit_code, exit_signal]
end

Net::SSH.start(server, Etc.getlogin) do |ssh|
  puts ssh_exec!(ssh, "true").inspect
  # => ["", "", 0, nil]

  puts ssh_exec!(ssh, "false").inspect  
  # => ["", "", 1, nil]

end

希望这有帮助。

Answer 2

基于 flitzwald 的答案 - 我已经将我的版本修补到 Net::SSH (Ruby 1.9+)

class Net::SSH::Connection::Session
  class CommandFailed < StandardError
  end

  class CommandExecutionFailed < StandardError
  end

  def exec_sc!(command)
    stdout_data,stderr_data = "",""
    exit_code,exit_signal = nil,nil
    self.open_channel do |channel|
      channel.exec(command) do |_, success|
        raise CommandExecutionFailed, "Command \"#{command}\" was unable to execute" unless success

        channel.on_data do |_,data|
          stdout_data += data
        end

        channel.on_extended_data do |_,_,data|
          stderr_data += data
        end

        channel.on_request("exit-status") do |_,data|
          exit_code = data.read_long
        end

        channel.on_request("exit-signal") do |_, data|
          exit_signal = data.read_long
        end
      end
    end
    self.loop

    raise CommandFailed, "Command \"#{command}\" returned exit code #{exit_code}" unless exit_code == 0

    {
      stdout:stdout_data,
      stderr:stderr_data,
      exit_code:exit_code,
      exit_signal:exit_signal
    }
  end
end

Answer 3

对于较新版本的 Net::SSH，您只需将状态哈希传递给 Net::SSH::Connection::Session#exec：

status = {}

Net::SSH.start(hostname, user, options) do |ssh|
  channel = ssh.exec(command, status: status)
  channel.wait # wait for the command to actually be executed
end

puts status.inspect
# {:exit_code=>0}

默认情况下，exec 会传输其输出到 $stdout 和 $stderr。您可以将一个块传递给 exec 来执行不同的操作，例如：

ssh.exec(command, status: status) do |ch, stream, data|
  if stream == :stdout
    do_something_with_stdout(data)
  else
    do_something_with_stderr(data)
  end
end

这适用于 6.1.0 - 不确定旧版本的可用性。请参阅 http://net -ssh.github.io/net-ssh/Net/SSH/Connection/Session.html#method-i-exec 了解更多详细信息。