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
|
# Poor Man's Fiber (API compatible Thread based Fiber implementation for Ruby 1.8)
# (c) 2008 Aman Gupta (tmm1)
unless defined? Fiber
require 'thread'
class FiberError < StandardError; end
class Fiber
def initialize
raise ArgumentError, 'new Fiber requires a block' unless block_given?
@yield = Queue.new
@resume = Queue.new
@thread = Thread.new{ @yield.push [ *yield(*@resume.pop) ] }
@thread.abort_on_exception = true
@thread[:fiber] = self
end
attr_reader :thread
def alive?
@thread.alive?
end
def resume *args
raise FiberError, 'dead fiber called' unless @thread.alive?
raise FiberError, 'double resume' if @thread == Thread.current
@resume.push(args)
result = @yield.pop
result.size > 1 ? result : result.first
end
def resume!
@resume.push []
end
def yield *args
@yield.push(args)
result = @resume.pop
result.size > 1 ? result : result.first
end
def self.yield *args
raise FiberError, "can't yield from root fiber" unless fiber = Thread.current[:fiber]
fiber.yield(*args)
end
def self.current
Thread.current[:fiber] or raise FiberError, 'not inside a fiber'
end
def inspect
"#<#{self.class}:0x#{self.object_id.to_s(16)}>"
end
end
else
require 'fiber' unless Fiber.respond_to?(:current)
end
if __FILE__ == $0
f = Fiber.new{ puts 'hi'; p Fiber.yield(1); puts 'bye'; :done }
p f.resume
p f.resume(2)
end
__END__
$ ruby fbr.rb
hi
1
2
bye
:done
$ ruby1.9 fbr.rb
hi
1
2
bye
:done
|