Make benchmarks JRuby compatible

Co-Authored-By: Charles Oliver Nutter <[email protected]>
ruby · Jan 29, 2025 · 01f0bc9 · 01f0bc9
1 parent 39ef430
commit 01f0bc9
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 10 deletions.
diff --git a/benchmark/encoder.rb b/benchmark/encoder.rb
@@ -1,9 +1,11 @@
 require "benchmark/ips"
 require "json"
 require "date"
-require "oj"
-
-Oj.default_options = Oj.default_options.merge(mode: :compat)
+begin
+  require "oj"
+  Oj.default_options = Oj.default_options.merge(mode: :compat)
+rescue LoadError
+end
 
 if ENV["ONLY"]
   RUN = ENV["ONLY"].split(/[,: ]/).map{|x| [x.to_sym, true] }.to_h
@@ -18,11 +20,16 @@
 def implementations(ruby_obj)
   state = JSON::State.new(JSON.dump_default_options)
   coder = JSON::Coder.new
-  {
+  implementations = {
     json: ["json", proc { JSON.generate(ruby_obj) }],
     json_coder: ["json_coder", proc { coder.dump(ruby_obj) }],
-    oj: ["oj", proc { Oj.dump(ruby_obj) }],
   }
+
+  if defined?(Oj)
+    implementations[:oj] = ["oj", proc { Oj.dump(ruby_obj) }]
+  end
+
+  implementations
 end
 
 def benchmark_encoding(benchmark_name, ruby_obj, check_expected: true, except: [])

diff --git a/benchmark/parser.rb b/benchmark/parser.rb
@@ -1,7 +1,14 @@
 require "benchmark/ips"
 require "json"
-require "oj"
-require "rapidjson"
+begin
+  require "oj"
+rescue LoadError
+end
+
+begin
+  require "rapidjson"
+rescue LoadError
+end
 
 if ENV["ONLY"]
   RUN = ENV["ONLY"].split(/[,: ]/).map{|x| [x.to_sym, true] }.to_h
@@ -20,9 +27,16 @@ def benchmark_parsing(name, json_output)
   Benchmark.ips do |x|
     x.report("json")      { JSON.parse(json_output) } if RUN[:json]
     x.report("json_coder") { coder.load(json_output) } if RUN[:json_coder]
-    x.report("oj")        { Oj.load(json_output) } if RUN[:oj]
-    x.report("Oj::Parser") { Oj::Parser.new(:usual).parse(json_output) } if RUN[:oj]
-    x.report("rapidjson") { RapidJSON.parse(json_output) } if RUN[:rapidjson]
+
+    if defined?(Oj)
+      x.report("oj")        { Oj.load(json_output) } if RUN[:oj]
+      x.report("Oj::Parser") { Oj::Parser.new(:usual).parse(json_output) } if RUN[:oj]
+    end
+
+    if defined?(RapidJSON)
+      x.report("rapidjson") { RapidJSON.parse(json_output) } if RUN[:rapidjson]
+    end
+
     x.compare!(order: :baseline)
   end
   puts