summaryrefslogtreecommitdiff
path: root/lib/ffi_yajl/benchmark/encode.rb
blob: 01e808b107fea2478334155988292a00f2b1e846 (plain)
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
$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/..')
$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')

require 'rubygems'
require 'benchmark'
require 'yajl'
require 'stringio'
require 'ffi_yajl'
begin
  require 'json'
rescue LoadError
end
begin
  require 'psych'
rescue LoadError
end
begin
  require 'active_support'
rescue LoadError
end

filename = ARGV[0] || 'benchmark/subjects/ohai.json'
hash = File.open(filename, 'rb') { |f| Yajl::Parser.new.parse(f.read) }

times = ARGV[1] ? ARGV[1].to_i : 1000
puts "Starting benchmark encoding #{filename} #{times} times\n\n"
Benchmark.bmbm { |x|
  io_encoder = Yajl::Encoder.new
  string_encoder = Yajl::Encoder.new

  x.report("FFI_Yajl::Encoder.encode (to a String)") {
    times.times {
      output = FFI_Yajl::Encoder.encode(hash)
    }
  }

  x.report("Yajl::Encoder.encode (to a String)") {
    times.times {
      output = Yajl::Encoder.encode(hash)
    }
  }

  x.report("Yajl::Encoder#encode (to an IO)") {
    times.times {
      io_encoder.encode(hash, StringIO.new)
    }
  }
  x.report("Yajl::Encoder#encode (to a String)") {
    times.times {
      output = string_encoder.encode(hash)
    }
  }
  if defined?(JSON)
    x.report("JSON.generate") {
      times.times {
        JSON.generate(hash)
      }
    }
  end
  if defined?(Psych)
    x.report("Psych.to_json") {
      times.times {
        Psych.to_json(hash)
      }
    }
    if defined?(Psych::JSON::Stream)
      x.report("Psych::JSON::Stream") {
        times.times {
          io = StringIO.new
          stream = Psych::JSON::Stream.new io
          stream.start
          stream.push hash
          stream.finish
        }
      }
    end
  end
  if defined?(ActiveSupport::JSON)
    x.report("ActiveSupport::JSON.encode") {
      times.times {
        ActiveSupport::JSON.encode(hash)
      }
    }
  end
}