23 Ruby异常处理之确保代码安全

在学习 Ruby 的异常处理时，我们已经探讨了如何捕捉异常。通过捕捉异常，我们能够理解什么地方可能会发生错误，并采取适当措施。然而，确保代码的安全性不仅仅是在发生错误时捕捉它，还要在代码中使用一些策略，以降低发生错误的概率，并确保代码在出现异常时表现良好。

本篇将专注于如何确保代码的安全性，特别是在处理可能抛出异常的代码段时。接下来，我们将通过一些案例来讨论如何有效地使用异常处理，以保证程序的健壮性。

使用 begin-rescue 块

在 Ruby 中，当我们知道某个操作可能会引发异常时，使用 begin-rescue 块显得尤为重要。这种结构不仅可以捕捉异常，还能确保在发生异常时执行一些清理工作。

案例 1：安全读取文件

考虑一个从文件中读取内容的简单示例。如果文件不存在，直接读取将抛出异常。我们可以使用 begin-rescue 块来确保代码安全，并处理可能的异常。

def read_file(file_path)
  begin
    file = File.open(file_path, "r")
    content = file.read
    puts content
  rescue Errno::ENOENT => e
    puts "错误：文件不存在 - #{e.message}"
  ensure
    file&.close  # 确保文件句柄被关闭
  end
end

read_file("不存在的文件.txt")

在上面的代码中：

• begin 块包含可能引发异常的代码。
• rescue 会捕捉文件未找到的异常，并输出友好的错误信息。
• ensure 块确保不论发生什么情况，文件都能被正确关闭。

案例 2：网络请求

当进行网络请求时，网络不可用或请求超时都可能引发异常。我们可以确保通过 begin-rescue 块来处理这些情况，保证程序的稳定性。

require 'net/http'

def fetch_data(url)
  begin
    response = Net::HTTP.get_response(URI(url))
    if response.is_a?(Net::HTTPSuccess)
      puts "成功获取数据: #{response.body}"
    else
      puts "请求失败: #{response.code} - #{response.message}"
    end
  rescue SocketError => e
    puts "网络错误: #{e.message}"
  rescue StandardError => e
    puts "其他错误: #{e.message}"
  end
end

fetch_data("http://example.com")

在这个示例中：

• 我们尝试获取一个 URL 的响应。
• 捕捉 SocketError 来处理网络错误，同时捕获其他常规错误，确保程序不会因为未处理的异常而崩溃。

使用 raise 重新抛出异常

在某些情况下，你可能想在处理异常后将其重新抛出。这可以让上层调用者知道发生了什么，从而进行额外的处理。

def safe_divide(a, b)
  begin
    result = a / b
    puts "结果：#{result}"
  rescue ZeroDivisionError => e
    puts "错误：不能除以零"
    raise  # 重新抛出异常
  end
end

begin
  safe_divide(10, 0)
rescue ZeroDivisionError => e
  puts "在主程序中捕捉到异常: #{e.message}"
end

在这个例子中：

• 我们在 safe_divide 方法中捕捉了 ZeroDivisionError 并打印错误信息。
• 然后使用 raise 重新抛出异常，让主程序捕捉到。

小结

通过使用 begin-rescue 块和 ensure 语句，我们能够在 Ruby 中有效地处理异常，确保代码的安全性。这使得我们的程序更加健壮，并且能够更友好地与用户交互。

在下一篇中，我们将讨论如何自定义异常，以便在特定情况下抛出自定义的错误类型。自定义异常处理是提升代码质量和可读性的重要手段，敬请期待！