Updated atomic benchmarks to test all available implementations.

Author: jdantonio

examples/bench_atomic.rb

@@ -1,6 +1,7 @@
 require 'benchmark'
-require 'concurrent'
+require 'rbconfig'
 require 'thread'
+require 'concurrent'
 Thread.abort_on_exception = true
 
 $go = false # for synchronizing parallel threads
@@ -11,18 +12,28 @@
 # number of threads for parallel test
 M = ARGV[0] ? ARGV[0].to_i : 100
 
+# list of platform-specific implementations
+ATOMICS = [
+  'MutexAtomic',
+  'CAtomic',
+  'JavaAtomic',
+  'RbxAtomic',
+]
+
+puts "Testing with #{RbConfig::CONFIG['ruby_install_name']} #{RUBY_VERSION}"
 
-puts "*** Sequential updates ***"
+puts
+puts '*** Sequential updates ***'
 Benchmark.bm(10) do |x|
   value = 0
-  x.report "no lock" do
+  x.report 'no lock' do
     N.times do
       value += 1
     end
   end
 
   @lock = Mutex.new
-  x.report "mutex" do
+  x.report 'mutex' do
     value = 0
     N.times do
       @lock.synchronize do
@@ -31,19 +42,23 @@
     end
   end
 
-  @atom = Concurrent::Atomic.new(0)
-  x.report "atomic" do
-    N.times do
-      @atom.update{|x| x += 1}
+  ATOMICS.each do |clazz|
+    if Concurrent.const_defined? clazz
+      @atom = Concurrent.const_get(clazz).new(0)
+      x.report clazz do
+        N.times do
+          @atom.update{|x| x += 1}
+        end
+      end
     end
   end
 end
 
 def para_setup(num_threads, count, &block)
   if num_threads % 2 > 0
-    raise ArgumentError, "num_threads must be a multiple of two"
+    raise ArgumentError, 'num_threads must be a multiple of two'
   end
-  raise ArgumentError, "need block" unless block_given?
+  raise ArgumentError, 'need block' unless block_given?
 
   # Keep those threads together
   tg = ThreadGroup.new
@@ -62,7 +77,7 @@ def para_setup(num_threads, count, &block)
   end
 
   # Make sure all threads are started
-  while tg.list.find{|t| t.status != "run"}
+  while tg.list.find{|t| t.status != 'run'}
     Thread.pass
   end
 
@@ -78,15 +93,15 @@ def para_run(tg)
   $go = false
 end
 
-puts "*** Parallel updates ***"
+puts
+puts '*** Parallel updates ***'
 Benchmark.bm(10) do |bm|
   # This is not secure
   value = 0
   tg = para_setup(M, N/M) do |diff|
     value += diff
   end
-  bm.report("no lock"){ para_run(tg) }
-
+  bm.report('no lock'){ para_run(tg) }
 
   value = 0
   @lock = Mutex.new
@@ -95,15 +110,17 @@ def para_run(tg)
       value += diff
     end
   end
-  bm.report("mutex"){ para_run(tg) }
+  bm.report('mutex'){ para_run(tg) }
   raise unless value == 0
 
-
-  @atom = Concurrent::Atomic.new(0)
-  tg = para_setup(M, N/M) do |diff|
-    @atom.update{|x| x + diff}
+  ATOMICS.each do |clazz|
+    if Concurrent.const_defined? clazz
+      @atom = Concurrent.const_get(clazz).new(0)
+      tg = para_setup(M, N/M) do |diff|
+        @atom.update{|x| x + diff}
+      end
+      bm.report(clazz){ para_run(tg) }
+      raise unless @atom.value == 0
+    end
   end
-  bm.report("atomic"){ para_run(tg) }
-  raise unless @atom.value == 0
-
 end