benchmark/bm_vm2_fiber_chain.rb


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

#!/usr/bin/env ruby

require 'benchmark'

def make_link(previous)
  Fiber.new do
    while message = previous.resume
      Fiber.yield(message)
    end
  end
end

def make_chain(length, &block)
  chain = Fiber.new(&block)

  (length - 1).times do
    chain = make_link(chain)
  end

  return chain
end

def run_benchmark(length, repeats, message = :hello)
  chain = nil

  time = Benchmark.realtime do
    chain = make_chain(length) do
      while true
        Fiber.yield(message)
      end
    end
  end

  puts "Creating #{fibers} fibers took #{time}..."

  time = Benchmark.realtime do
    repeats.times do
      abort "invalid result" unless chain.resume == message
    end
  end

  puts "Passing #{repeats} messages took #{time}..."
end

n = (ARGV[0] || 1000).to_i
m = (ARGV[1] || 1000).to_i

5.times do
  run_benchmark(n, m)
end