1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
|
#!/usr/bin/env ruby -wW1
# frozen_string_literal: true
$LOAD_PATH << '.'
$LOAD_PATH << File.join(__dir__, '../lib')
$LOAD_PATH << File.join(__dir__, '../ext')
require 'optparse'
# require 'yajl'
require 'perf'
require 'json'
require 'json/ext'
require 'oj'
@verbose = false
@indent = 0
@iter = 100_000
@gets = 0
@fetch = false
@write = false
@read = false
opts = OptionParser.new
opts.on('-v', 'verbose') { @verbose = true }
opts.on('-c', '--count [Int]', Integer, 'iterations') { |i| @iter = i }
opts.on('-i', '--indent [Int]', Integer, 'indentation') { |i| @indent = i }
opts.on('-g', '--gets [Int]', Integer, 'number of gets') { |i| @gets = i }
opts.on('-f', 'fetch') { @fetch = true }
opts.on('-w', 'write') { @write = true }
opts.on('-r', 'read') { @read = true }
opts.on('-h', '--help', 'Show this display') { puts opts; Process.exit!(0) }
opts.parse(ARGV)
# This just navigates to each leaf of the JSON structure.
def dig(obj, &blk)
case obj
when Array
obj.each { |e| dig(e, &blk) }
when Hash
obj.each_value { |e| dig(e, &blk) }
else
blk.yield(obj)
end
end
@obj = {
'a' => 'Alpha', # string
'b' => true, # boolean
'c' => 12_345, # number
'd' => [ true, [false, {'12345' => 12_345, 'nil' => nil}, 3.967, { 'x' => 'something', 'y' => false, 'z' => true}, nil]], # mix it up array
'e' => { 'one' => 1, 'two' => 2 }, # hash
'f' => nil, # nil
'g' => 12_345_678_901_234_567_890_123_456_789, # big number
'h' => { 'a' => { 'b' => { 'c' => { 'd' => {'e' => { 'f' => { 'g' => nil }}}}}}}, # deep hash, not that deep
'i' => [[[[[[[nil]]]]]]] # deep array, again, not that deep
}
Oj.default_options = { :indent => @indent, :mode => :compat }
@json = Oj.dump(@obj)
@failed = {} # key is same as String used in tests later
def capture_error(tag, orig, load_key, dump_key, &blk)
obj = blk.call(orig)
raise "#{tag} #{dump_key} and #{load_key} did not return the same object as the original." unless orig == obj
rescue Exception => e
@failed[tag] = "#{e.class}: #{e.message}"
end
# Verify that all packages dump and load correctly and return the same Object as the original.
capture_error('Oj::Doc', @obj, 'load', 'dump') { |o| Oj::Doc.open(Oj.dump(o, :mode => :strict)) { |f| f.fetch() } }
# capture_error('Yajl', @obj, 'encode', 'parse') { |o| Yajl::Parser.parse(Yajl::Encoder.encode(o)) }
capture_error('JSON::Ext', @obj, 'generate', 'parse') { |o| JSON.generator = JSON::Ext::Generator; JSON::Ext::Parser.new(JSON.generate(o)).parse }
if @verbose
puts "json:\n#{@json}\n"
end
puts '-' * 80
puts 'Parse Performance'
perf = Perf.new()
perf.add('Oj::Doc', 'parse') { Oj::Doc.open(@json) { |f| } } unless @failed.key?('Oj::Doc')
# perf.add('Yajl', 'parse') { Yajl::Parser.parse(@json) } unless @failed.has_key?('Yajl')
perf.add('JSON::Ext', 'parse') { JSON::Ext::Parser.new(@json).parse } unless @failed.key?('JSON::Ext')
perf.run(@iter)
puts '-' * 80
puts 'JSON generation Performance'
Oj::Doc.open(@json) do |doc|
perf = Perf.new()
perf.add('Oj::Doc', 'dump') { doc.dump() }
# perf.add('Yajl', 'encode') { Yajl::Encoder.encode(@obj) }
perf.add('JSON::Ext', 'fast_generate') { JSON.fast_generate(@obj) }
perf.before('JSON::Ext') { JSON.generator = JSON::Ext::Generator }
perf.run(@iter)
end
if 0 < @gets
puts '-' * 80
puts 'Parse and get all values Performance'
perf = Perf.new()
perf.add('Oj::Doc', 'parse') { Oj::Doc.open(@json) { |f| @gets.times { f.each_value() {} } } } unless @failed.key?('Oj::Doc')
# perf.add('Yajl', 'parse') { @gets.times { dig(Yajl::Parser.parse(@json)) {} } } unless @failed.has_key?('Yajl')
perf.add('JSON::Ext', 'parse') { @gets.times { dig(JSON::Ext::Parser.new(@json).parse) {} } } unless @failed.key?('JSON::Ext')
perf.run(@iter)
end
if @fetch
puts '-' * 80
puts 'fetch nested Performance'
json_hash = Oj.load(@json, :mode => :strict)
Oj::Doc.open(@json) do |fast|
# puts "*** C fetch: #{fast.fetch('/d/2/4/y')}"
# puts "*** Ruby fetch: #{json_hash.fetch('d', []).fetch(1, []).fetch(3, []).fetch('x', nil)}"
perf = Perf.new()
perf.add('Oj::Doc', 'fetch') { fast.fetch('/d/2/4/x'); fast.fetch('/h/a/b/c/d/e/f/g'); fast.fetch('/i/1/1/1/1/1/1/1') }
# version that fails gracefully
perf.add('Ruby', 'fetch') do
json_hash.fetch('d', []).fetch(1, []).fetch(3, []).fetch('x', nil)
json_hash.fetch('h', {}).fetch('a', {}).fetch('b', {}).fetch('c', {}).fetch('d', {}).fetch('e', {}).fetch('f', {}).fetch('g', {})
json_hash.fetch('i', []).fetch(0, []).fetch(0, []).fetch(0, []).fetch(0, []).fetch(0, []).fetch(0, []).fetch(0, nil)
end
# version that raises if the path is incorrect
# perf.add('Ruby', 'fetch') { @fetch.times { json_hash['d'][1][3][1] } }
perf.run(@iter)
end
end
if @write
puts '-' * 80
puts 'JSON write to file Performance'
Oj::Doc.open(@json) do |doc|
perf = Perf.new()
perf.add('Oj::Doc', 'dump') { doc.dump(nil, 'oj.json') }
# perf.add('Yajl', 'encode') { File.open('yajl.json', 'w') { |f| Yajl::Encoder.encode(@obj, f) } }
perf.add('JSON::Ext', 'fast_generate') { File.write('json_ext.json', JSON.fast_generate(@obj)) }
perf.before('JSON::Ext') { JSON.generator = JSON::Ext::Generator }
perf.run(@iter)
end
end
if @read
puts '-' * 80
puts 'JSON read from file Performance'
Oj::Doc.open(@json) { |doc| doc.dump(nil, 'oj.json') }
# File.open('yajl.json', 'w') { |f| Yajl::Encoder.encode(@obj, f) }
JSON.generator = JSON::Ext::Generator
File.write('json_ext.json', JSON.fast_generate(@obj))
Oj::Doc.open(@json) do |_doc|
perf = Perf.new()
perf.add('Oj::Doc', 'open_file') { Oj::Doc.open_file('oj.json') }
# perf.add('Yajl', 'decode') { Yajl::decoder.decode(File.read('yajl.json')) }
perf.add('JSON::Ext', '') { JSON.parse(File.read('json_ext.json')) }
perf.before('JSON::Ext') { JSON.parser = JSON::Ext::Parser }
perf.run(@iter)
end
end
unless @failed.empty?
puts 'The following packages were not included for the reason listed'
@failed.each { |tag, msg| puts "***** #{tag}: #{msg}" }
end
|