如何实现Ruby的Array#my_map!方法以修改原数组实例

核心实现逻辑 #

你之前尝试用@前缀访问实例变量失败的原因很简单：Ruby的原生核心类（如Array、Hash等）底层由C实现，没有在Ruby层面暴露可直接访问的实例变量，你直接操作self（当前调用方法的数组实例本身）即可完成修改。

前置参考：你已实现的基础方法示例 #

class Array
  # 自定义my_each实现
  def my_each
    i = 0
    while i < self.length
      yield(self[i])
      i += 1
    end
    self
  end

  # 原有非可变my_map实现
  def my_map
    new_arr = []
    my_each { |item| new_arr << yield(item) }
    new_arr
  end
end

my_map!的实现方案 #

完全基于你已有的my_each实现，无需额外依赖Ruby原生迭代方法：

class Array
  def my_map!
    index = 0
    # 基于自定义my_each遍历
    my_each do |item|
      # 直接修改当前实例对应下标的值
      self[index] = yield(item)
      index += 1
    end
    # 遵循Ruby可变方法惯例，返回修改后的实例本身
    self
  end
end

功能测试示例 #

test_arr = [1, 2, 3, 4]
test_arr.my_map! { |num| num ** 2 }
puts test_arr # 输出 [1, 4, 9, 16]，原数组已被修改

针对有Java基础开发者的补充说明 #

• Ruby的数组是可变结构，和Java的ArrayList逻辑类似，你可以把self[index] = 新值理解为等价于Java里ArrayList.set(index, 新值)的操作
• 带!后缀的方法是Ruby的通用命名惯例，代表该方法会修改接收者本身，无!的同名方法默认返回新对象，和Java里String不可变、StringBuilder可变的设计逻辑是相通的

这类场景的最佳实践 #

• 练习场景下的核心类扩展（俗称猴子补丁）没问题，生产环境使用建议改用Ruby的Refinement特性限定扩展的生效范围，避免全局污染其他代码
• 自定义核心类方法时尽量对齐Ruby原生方法的行为惯例，比如my_map!处理空块的行为、参数兼容规则都可以和官方map!对齐
• 不要尝试访问核心类的内部私有实现，用官方暴露的公共API（如[]=、length、delete_at等）操作实例即可，核心类的底层实现不会向上层Ruby代码暴露细节

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