Ruby Array＃to_yaml不会递归调用to_yaml

Question

以下代码

require 'yaml'

class MyObject
  def initialize(value)
    @value = value
  end

  def to_yaml()
    @value + @value
  end
end

puts [MyObject.new("a"), MyObject.new("b")]

在Ruby 2.1.3p242上生成以下输出：

---
- !ruby/object:MyObject
  value: a
- !ruby/object:MyObject
  value: b

我期望在哪里

---
- aa
- bb

就像我在Array内的每个对象上调用to_yaml ：

puts [MyObject.new("a").to_yaml, MyObject.new("b").to_yaml]

我究竟做错了什么？

Answer 1

我也将保留先前的答案，因为它可能对某人有用，但这是更好的解决方案。

我实际上已经过分简化了原始问题。 我试图将我的自定义对象呈现为YAML序列[1, 2, 3, ...] 。 先前的答案可能适用于呈现为字符串的对象。

这是工作版本：

require 'yaml'

class MyObject
  def initialize(value)
    @value = value
  end

  def encode_with coder
    coder.tag = nil
    coder.seq = [@value, @value]
  end
end

puts [MyObject.new("a"), MyObject.new("b")].to_yaml

一些参考：

http://blog.mustmodify.com/pages/psych-ruby-1-9-yaml

http://ruby-doc.org/stdlib-1.9.3/libdoc/psych/rdoc/Psych/Coder.html

Answer 2

我要替换Psych::Visitors::YAMLTree的visit_Array方法

class MyVisitor < Psych::Visitors::YAMLTree
  def visit_Array o
    super o.map { |i| i.respond_to?(:to_yaml) ? i.to_yaml : i }
  end
end

然后，我以这种方式转储YAML：

a = [MyObject.new("a"), MyObject.new("b")]
visitor = MyVisitor.create
visitor << a
puts visitor.tree.yaml