fiber-mon 0.1.0

data/fiber-mon.rb

@@ -0,0 +1,252 @@
+#
+#   fiber-mon.rb - 
+#   	Copyright (C) 2010 Keiju ISHITSUKA
+#				(Penta Advanced Labrabries, Co.,Ltd)
+#
+
+require "forwardable"
+require "monitor"
+
+class FiberMon
+  include MonitorMixin
+
+  def initialize
+    super
+
+    #      @mon_mx = Mutex.new
+    #      @owner = nil
+    #      @mon_mxmx = Mutex.new
+
+    @started = false
+
+    @current = nil
+
+    @entries = []
+
+    @wait_resume = []
+    @wait_resume_mx = ::Mutex.new
+    @wait_resume_cv = ::ConditionVariable.new
+
+    mon_start
+  end
+
+  attr_reader :current
+
+  def mon_start
+    return if @started
+    @started = true
+    th = Thread.start{
+      loop do
+	@wait_resume_mx.synchronize do
+	  while @wait_resume.empty? && @entries.empty?
+	    @wait_resume_cv.wait(@wait_resume_mx)
+	  end
+	  
+	  if block =  @entries.shift
+	    @current = Fiber.new{block.call(self)}
+	  else
+	    @current = @wait_resume.shift
+	  end
+	end
+	begin
+	  mx = @current.resume if @current
+#	rescue
+#	  if $DEBUG
+#	    p $!
+#	    puts $@
+#	  end
+#	  raise
+	ensure
+	  if mx.kind_of?(Mutex)
+	    mx.unlock
+	  end
+	  @current = nil
+	end
+      end
+    }
+    th
+  end
+  alias start mon_start
+
+  def entry_fiber(p0 = nil, &block)
+    p0 = block if block
+    @wait_resume_mx.synchronize do
+      @entries.push p0
+      @wait_resume_cv.signal
+    end
+  end
+  alias entry entry_fiber
+#  alias start entry_fiber
+
+  def yield
+    @wait_resume_mx.synchronize do
+      @wait_resume.push @current
+      @wait_resume_cv.signal
+    end
+    fiber_yield
+  end
+
+  #     def synchronize(&block)
+  #       @mon_mxmx.synchronize do
+  # 	@mon_mx.lock
+  # 	@owner = Thread.current
+  #       end
+  #       begin
+  # 	block.call
+  #       ensure
+  # 	@mon_mxmx.synchronize do
+  # 	  @owner = nil
+  # 	  @mon_mx.unlock
+  # 	end
+  #       end
+  #     end
+
+  #     def fiber_yield
+  #       begin
+  # 	status = nil
+  # 	@mon_mxmx.synchronize do
+  # 	  if status = @owner == Thread.current && @mon_mx.locked?
+  # 	    @mon_mx.unlock
+  # 	    @owner = nil
+  # 	  end
+  # 	end
+  # 	Fiber.yield
+  #       rescue
+  # 	Log::debug_exception(self)
+  #       ensure
+  # 	@mon_mxmx.synchronize do
+  # 	  if status
+  # 	    @owner = Thread.current
+  # 	    @mon_mx.lock
+  # 	  end
+  # 	end
+  #       end
+  #     end
+
+  def fiber_yield
+    begin
+      mon_check_owner
+    rescue
+      return Fiber.yield
+    end
+    if (count = mon_exit_for_cond) > 0
+      begin
+	Fiber.yield @mon_mutex
+      ensure
+	@mon_mutex.lock
+      end
+    else
+      Fiber.yield
+    end
+    mon_enter_for_cond(count)
+  end
+
+  def new_mon
+    Monitor.new(self)
+  end
+
+  def new_cv
+    ConditionVariable.new(self)
+  end
+  alias new_cond new_cv
+
+  def entry_wait_resume(*fbs)
+    @wait_resume_mx.synchronize do
+      @wait_resume.concat(fbs)
+      @wait_resume_cv.signal
+    end
+  end
+
+  class Monitor<::Monitor
+    extend Forwardable
+
+    def initialize(fibmon)
+      super()
+      @fibmon = fibmon
+    end
+
+    def_delegator :@fibmon, :entry_wait_resume
+    def_delegator :@fibmon, :current
+
+    def new_cv
+      ConditionVariable.new(self)
+    end
+
+    def fiber_yield
+      begin
+	mon_check_owner
+      rescue
+	return Fiber.yield
+      end
+      if (count = mon_exit_for_cond) > 0
+	begin
+	  Fiber.yield @mon_mutex
+	ensure
+	  @mon_mutex.lock
+	end
+      else
+	Fiber.yield
+      end
+      mon_enter_for_cond(count)
+    end
+    
+    alias yield fiber_yield
+  end
+
+  class ConditionVariable
+    def initialize(monitor)
+      @mon = monitor
+
+      @waitings = []
+      @waitings_mx = Mutex.new
+    end
+
+    def synchronize(&block)
+      @waitings_mx.synchronize(&block)
+    end
+
+    def signal(&block)
+      @waitings_mx.synchronize do
+	if block_given?
+	  yield
+	end
+
+	if fb =  @waitings.shift
+	  @mon.entry_wait_resume(fb)
+	end
+      end
+    end
+
+    def broadcast(&block)
+      @waitings_mx.synchronize do
+	if block_given?
+	  yield
+	end
+
+	return if @waitings.empty?
+	fbs, @waitings = @waitings, []
+	@mon.entry_wait_resume(*fbs)
+      end
+    end
+
+    def wait
+      @waitings_mx.synchronize do
+	@waitings.push @mon.current
+      end
+      @mon.fiber_yield
+    end
+
+    def wait_until(&cond)
+      until cond.call
+	wait
+      end 
+    end
+
+    def wait_while(&cond)
+      while cond.call
+	wait
+      end 
+    end
+  end
+
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification 
+name: fiber-mon
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Keiju.Ishitsuka
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-10-28 00:00:00 +09:00
+default_executable: 
+dependencies: []
+
+description: |
+  Simple fiber scheduler for Ruby.
+
+email: keiju@ishitsuka.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- fiber-mon.rb
+has_rdoc: true
+homepage: http://github.com/keiju/fiber-mon
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- .
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: fiber-mon
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Simple fiber scheduler for Ruby
+test_files: []
+