如何在 YAML 中读取 ruby​​ 对象

Question

我有一个如下所示的 YAML 文件：

--- !ruby/object:Hi
num: 1
--- !ruby/object:Hi
num: 2

还有我的红宝石代码：

require 'yaml'
class A
 attr_accessor :num
 def initialize num
  @num=num
 end
end
a=A.new 1
b=A.new 2
File.open 'test.yml', 'r+' do |f|
 f.write YAML.dump a
 f.write YAML.dump b
 f.seek(0)
 #c=obj a
 #d=ojb b
end

我想让 p c 和 p d 输出如下：

#<A:0x00000 @num=1>
#<A:0x00001 @num=2>

Answer 1

我不知道为什么某些人如此坚持不能将多个 YAML 文档放在一个文件中，因为 YAML 格式是 explicitly designed to accommodate that particular use case。在 Ruby 中，您可以使用 YAML.load_stream 解析多个文档，这会将反序列化的对象作为数组返回：

require "yaml"

class A
  attr_accessor :num
  def initialize num
    @num = num
  end
end

a = A.new 1
b = A.new 2

File.open "test.yml", "r+" do |f|
  f.write YAML.dump(a)
  f.write YAML.dump(b)
  f.seek(0)

  c, d = YAML.load_stream(f)

  p c
  # => #<A:0x000055d423387cd0 @num=1>
  p d
  # => #<A:0x000055d423387528 @num=2>
end

您也可以将一个块传递给YAML.load_stream，在这种情况下，每个文档都会调用它：

YAML.load_stream(f) do |obj|
  p obj.num
end
# => 1
# => 2

您可以在 repl.it 上看到这两种情况：https://repl.it/@jrunning/FavorableElderlyAddition

Answer 2

object = JSON.parse(YAML::load_file("./test.yaml").to_json, object_class: OpenStruct)

不是 100% 确定这些会以什么格式出现，但它会给你一个很好的起点

Answer 3

我更喜欢将对象写入单独的文件，但您的序列化方式也可以。 这是您的情况的解决方案；

RUBY_OBJECT = '--- !ruby/object:'

def load_objects(file_content)
  object_contents = file_content.split(RUBY_OBJECT)

  object_contents.map do |object_content|
    YAML.load(RUBY_OBJECT + object_content)
  end.compact
end

##########

File.open 'test.yml', 'r+' do |f|
  f.write YAML.dump(a)
  f.write YAML.dump(b)

  f.seek(0)
  file_content = f.read
  c, d = *load_objects(file_content)

  puts c
  puts d
end

您还可以对文件内容运行正则表达式以获取单个对象定义。

如果使用一个文件进行序列化是一项严格要求，您可以像这样将一组对象序列化到文件中；

require 'yaml'

class A
  attr_accessor :num

  def initialize(num)
    @num = num
  end
end

a = A.new(1)
b = A.new(2)

File.open('test.yml', 'w+') do |f|
  f.write YAML.dump([a, b])
end

objects = YAML.load_file('test.yml')