使对象的行为类似于 Ruby 中并行赋值的数组

Question

假设您在 Ruby 中执行此操作：

ar = [1, 2]
x, y = ar

那么，x == 1 且 y == 2。我可以在自己的类中定义一个方法来产生相同的效果吗？ 例如

rb = AllYourCode.new
x, y = rb

，到目前为止，我对这样的赋值所能做的就是使 x == rb 且 y = nil。 Python有一个这样的特性：

>>> class Foo:
...     def __iter__(self):
...             return iter([1,2])
...
>>> x, y = Foo()
>>> x
1
>>> y
2

Answer 1

是的。 定义#to_ary。 这将使您的对象被视为分配的数组。

irb> o = Object.new
=> #<Object:0x3556ec>
irb> def o.to_ary
       [1, 2]
     end
=> nil
irb> x, y = o
=> [1,2]
irb> x
#=> 1
irb> y
#=> 2

#to_a 和 #to_ary 的区别在于 #to_a 用于尝试转换
将给定对象转换为数组，如果我们可以将给定对象视为数组，则 #to_ary 可用。 这是一个微妙的区别。

Answer 2

几乎：

class AllYourCode
   def to_a
     [1,2]
   end
end

rb = AllYourCode.new
x, y = *rb
p x
p y

Splat 将尝试调用 to_ary，然后尝试调用 to_a。 我不确定你为什么要这样做，这实际上是一个语法功能，恰好在其实现中使用了 Array，而不是 Array 的功能。

换句话说，多重赋值的用例如下：

# swap
x, y = y, x

# multiple return values
quot, rem = a.divmod(b)

# etc.
name, age = "Person", 100

换句话说，大多数时候从（Array）分配的对象甚至不明显。

Answer 3

您无法重新定义赋值，因为它是一个运算符而不是一个方法。 但如果您的 AllYourCode 类继承自 Array，您的示例就可以工作。

当 Ruby 遇到赋值时，它会查看右侧，如果有多个右值，则会将它们收集到一个数组中。 然后它看向左侧。 如果那里有一个左值，则将其分配给该数组。

def foo 
  return "a", "b", "c" # three rvalues
end

x = foo # => x == ["a", "b", "c"]

如果有多个左值（更具体地说，如果它看到逗号），它会连续分配右值并丢弃多余的值。

x, y, z = foo # => x == "a", y == "b", z == "c"
x, y = foo    # => x == "a", y == "b"
x, = foo      # => x == "a"

正如您所发现的，如果返回数组，您也可以进行并行分配。

def bar
  ["a", "b", "c"]
end

x = bar       # => x == ["a", "b", "c"]
x, y, z = bar # => x == "a", y == "b", z == "c"
x, y = bar    # => x == "a", y == "b"
x, = bar      # => x == "a"

因此，在您的示例中，如果 rb 是数组或继承自数组，则 x 和 y 将被分配其前 2 个值。