File: perf_fast.rb

package info (click to toggle)
ruby-oj 3.16.15-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 14,208 kB
  • sloc: ansic: 20,028; ruby: 11,642; sh: 70; makefile: 17
file content (162 lines) | stat: -rwxr-xr-x 6,227 bytes parent folder | download | duplicates (2)
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