24 自定义异常处理

在上一篇中，我们探讨了异常处理的基本概念以及如何确保代码的安全性。这一篇，我们将深度了解如何在 Ruby 中创建和使用自定义异常。自定义异常可以帮助我们捕获和处理特定的错误情况，以便代码在异常情况下也能优雅地运行。

什么是自定义异常？

在 Ruby 中，异常处理通常使用内置的异常类。然而，有时候内置的异常类不能满足我们的需求，这时我们就可以创建自定义异常。自定义异常可以使我们的代码更具可读性和可维护性。

创建自定义异常

在 Ruby 中，创建自定义异常实际上是定义一个新的类，并使它继承自 StandardError 类。这个自定义异常可以拥有额外的属性和方法，以便存储更多的上下文信息。

基本示例

以下是一个创建自定义异常的简单示例：

class MyCustomError < StandardError
  attr_reader :error_code

  def initialize(message, error_code)
    super(message)
    @error_code = error_code
  end
end

在上述代码中，我们定义了一个名为 MyCustomError 的异常类，它接受两个参数：message 和 error_code。这样，我们可以在抛出异常时提供更多的信息。

使用自定义异常

一旦你定义了自定义异常，就可以在代码的适当位置抛出它。这里是一个使用自定义异常的例子：

def divide(a, b)
  raise MyCustomError.new("不能除以零", 1001) if b == 0
  a / b
rescue MyCustomError => e
  puts "发生异常: #{e.message}，错误代码: #{e.error_code}"
end

divide(10, 0)  # 输出: 发生异常: 不能除以零，错误代码: 1001

在这个例子中，divide 方法尝试执行除法操作。如果第二个参数 b 为零，它将抛出一个 MyCustomError 异常。rescue 块捕获这个异常，并输出自定义的错误信息和错误代码。

异常层次结构

创建自定义异常时，我们也可以定义一个异常层次结构。这使得处理不同类型的异常变得更加灵活。可以通过创建一个基类和多个子类来实现这一点。

创建异常层次结构的示例

class MyBaseError < StandardError; end

class NotFoundError < MyBaseError; end

class PermissionDeniedError < MyBaseError; end

在这个例子中，MyBaseError 是异常层次的基类，NotFoundError 和 PermissionDeniedError 是两个子类。这样，调用者可以选择捕获更具体的异常。

使用异常层次结构

以下是如何使用这个异常层次结构进行异常处理的示例：

def find_item(item)
  raise NotFoundError.new("项目未找到") if item.nil?
  raise PermissionDeniedError.new("权限不足") unless has_permission?(item)
  
  item
rescue MyBaseError => e
  puts "发生异常: #{e.message}"
end

def has_permission?(item)
  # 假设这里包含权限检查的逻辑
  false
end

find_item(nil)            # 输出: 发生异常: 项目未找到
find_item("restricted")   # 输出: 发生异常: 权限不足

在上述代码中，find_item 方法根据不同的条件抛出不同的异常。rescue 块捕获所有 MyBaseError 的子类。

总结

自定义异常使得 Ruby 的异常处理更加灵活和强大。通过创建自己的异常类和层次结构，我们可以更准确地捕获和处理程序中的特殊情况，提高代码的可读性和可维护性。在下一篇中，我们将继续探讨 Ruby 中的常用标准库以及如何利用这些库获取更多的功能和效率。

继续保持学习的热情，Ruby 的世界总有新知识等待发现！