Ruby中基于菜单定义方法：SFTP代码重构咨询

一、现有实现思路的合理性 #

你的代码整体方向是对的：用类封装功能，通过run作为入口串联流程，把参数解析、连接操作、辅助功能拆分到不同方法，符合Ruby面向对象的基础设计逻辑。不过有几个语法和细节问题需要修正：

• parse方法里直接操作全局ARGV而非传入的arguments参数，会导致自定义参数失效；
• 代码缩进错误，parse方法内的begin...rescue块未正确闭合，导致connect等方法被嵌套在parse内部，属于语法错误；
• @options的命名不够直观，比如set_host可改为host，wharf改为port更符合行业惯例；
• .red和.bold颜色方法依赖colorize gem，未引入的话会报错，要么添加依赖要么移除。

二、拆分为三个独立方法的优化方案 #

按照职责单一原则，我们可以把核心逻辑拆分为「参数解析」「SFTP连接与操作」「入口执行」三个独立方法，同时修正细节问题：

优化后的完整代码 #

require 'net/sftp'
require 'ostruct'
require 'optparse'
# 如需颜色输出，先安装colorize：gem install colorize
# require 'colorize'

class Sftp
  def initialize
    @options = OpenStruct.new
  end

  # 方法1：参数解析（仅处理命令行参数的解析与校验）
  def parse_arguments(arguments)
    arguments << "-h" if arguments.empty?

    parser = OptionParser.new do |opts|
      opts.banner = "Usage: #{__FILE__} [options]"

      opts.on("-s", "--host=HOST", String, "目标主机地址") do |host|
        @options.host = host
      end

      opts.on("-u", "--username=USERNAME", String, "登录用户名") do |username|
        @options.username = username
      end

      opts.on("-p", "--password=PASSWORD", String, "登录密码") do |password|
        @options.password = password
      end

      opts.on("-P", "--port=PORT", Integer, "SFTP服务端口（默认22）") do |port|
        @options.port = port
      end

      opts.on("-m", "--mkdir=DIR", String, "要创建的远程目录") do |dir|
        @options.mkdir = dir
      end

      opts.on("-h", "--help", "显示帮助信息并退出") do
        puts opts
        exit
      end
    end

    begin
      parser.parse!(arguments)
    rescue OptionParser::MissingArgument => e
      puts "[!] #{e.message}"
      exit
    rescue OptionParser::InvalidOption => e
      puts "[!] #{e.message}"
      exit
    end
  end

  # 方法2：SFTP连接与操作（仅处理连接建立和业务操作）
  def execute_sftp_operations
    # 设置端口默认值
    port = @options.port || 22

    Net::SFTP.start(@options.host, @options.username, password: @options.password, port: port) do |sftp|
      create_remote_directory(sftp) if @options.mkdir
      output("SFTP操作执行完成")
    end

    output("退出时间 => (#{Time.now})")
  end

  # 方法3：入口方法（仅串联流程，不做具体业务）
  def run(arguments)
    parse_arguments(arguments)
    execute_sftp_operations
  end

  private

  def create_remote_directory(sftp)
    sftp.mkdir!(@options.mkdir)
    output("已创建远程目录 => #{@options.mkdir}")
  end

  def output(message)
    separator = "-" * 60
    puts separator
    puts " #{message}"
    puts separator
  end
end

sftp = Sftp.new
sftp.run(ARGV)

拆分说明 #

1. parse_arguments：只负责命令行参数的定义、解析和错误捕获，不涉及任何SFTP相关逻辑，职责单一；
2. execute_sftp_operations：专注于SFTP连接建立、执行指定操作（如创建目录），所有参数从@options读取，无需重复传递；
3. run：作为程序入口，仅调用前两个方法串联流程，逻辑清晰，便于后续扩展。

三、额外优化建议 #

• 补充必填参数校验：比如检查host、username、password是否为空，避免连接时报错；
• 捕获SFTP操作异常：比如创建目录时的权限不足、目录已存在等问题，可添加rescue Net::SFTP::StatusException块处理；
• 支持配置文件：如果需要频繁使用固定参数，可添加从YAML/JSON配置文件读取参数的功能；
• 扩展操作类型：比如新增文件上传、下载、删除等功能，每个操作封装为独立私有方法，保持代码整洁。

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