version 0.2.0 imported
This commit is contained in:
commit
7ba8740369
57
Rakefile
Normal file
57
Rakefile
Normal file
|
@ -0,0 +1,57 @@
|
|||
require 'rubygems'
|
||||
require 'rake'
|
||||
|
||||
begin
|
||||
require 'jeweler'
|
||||
Jeweler::Tasks.new do |gem|
|
||||
gem.name = "select"
|
||||
gem.summary = %Q{IO-event-handler based on select}
|
||||
gem.description = %Q{Select based event-handler for servers and sockets}
|
||||
gem.email = "Denis.Knauf@gmail.com"
|
||||
gem.homepage = "http://github.com/DenisKnauf/select"
|
||||
gem.authors = ["Denis Knauf"]
|
||||
gem.files = ["README", "VERSION", "lib/**/*.rb", "test/**/*.rb"]
|
||||
gem.require_paths = ["lib"]
|
||||
end
|
||||
Jeweler::GemcutterTasks.new
|
||||
rescue LoadError
|
||||
puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
|
||||
end
|
||||
|
||||
require 'rake/testtask'
|
||||
Rake::TestTask.new(:test) do |test|
|
||||
test.libs << 'lib' << 'test' << 'ext'
|
||||
test.pattern = 'test/**/*_test.rb'
|
||||
test.verbose = true
|
||||
end
|
||||
|
||||
begin
|
||||
require 'rcov/rcovtask'
|
||||
Rcov::RcovTask.new do |test|
|
||||
test.libs << 'test'
|
||||
test.pattern = 'test/**/*_test.rb'
|
||||
test.verbose = true
|
||||
end
|
||||
rescue LoadError
|
||||
task :rcov do
|
||||
abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
|
||||
end
|
||||
end
|
||||
|
||||
task :test => :check_dependencies
|
||||
|
||||
task :default => :test
|
||||
|
||||
require 'rake/rdoctask'
|
||||
Rake::RDocTask.new do |rdoc|
|
||||
if File.exist?('VERSION')
|
||||
version = File.read('VERSION')
|
||||
else
|
||||
version = ""
|
||||
end
|
||||
|
||||
rdoc.rdoc_dir = 'rdoc'
|
||||
rdoc.title = "sbdb #{version}"
|
||||
rdoc.rdoc_files.include('README*')
|
||||
rdoc.rdoc_files.include('lib/**/*.rb')
|
||||
end
|
256
lib/select.rb
Normal file
256
lib/select.rb
Normal file
|
@ -0,0 +1,256 @@
|
|||
class Select
|
||||
READ, WRITE, ERROR = 1, 2, 3
|
||||
|
||||
attr_reader :read, :write, :error
|
||||
attr_accessor :exit, :exit_on_empty
|
||||
|
||||
def empty?
|
||||
@read.empty? && @write.empty? && @error.empty?
|
||||
end
|
||||
|
||||
def self.new *p, &e
|
||||
r = super *p
|
||||
if e
|
||||
e.call r
|
||||
r.close
|
||||
else r
|
||||
end
|
||||
end
|
||||
|
||||
def initialize timeout = 30
|
||||
@read, @write, @error = {}, {}, {}
|
||||
@read.default = @write.default = @error.default = lambda{}
|
||||
@timeout, @tevent, @exit = timeout, lambda{}, false
|
||||
end
|
||||
|
||||
def timeout timeout = nil, &event
|
||||
return @timeout if timeout.nil?
|
||||
raise ArgumentError, "Numeric value expected, not: '#{timeout}'" unless timeout.kind_of? Numeric
|
||||
@timeout = timeout
|
||||
@tevent = event if event
|
||||
timeout
|
||||
end
|
||||
|
||||
def set hook, type = :read, &event
|
||||
raise ArgumentError, "This hook isn't supported: '#{hook.inspect}'" unless hook.kind_of? IO
|
||||
raise ArgumentError, "Unexpected Event: '#{event.inspect}'" unless event.kind_of? Proc
|
||||
case type
|
||||
when READ, :read then @read[ hook] = event
|
||||
when WRITE, :write then @write[ hook] = event
|
||||
when ERROR, :error then @error[ hook] = event
|
||||
when nil
|
||||
@read[ hook] = event
|
||||
@write[ hook] = event
|
||||
@error[ hook] = event
|
||||
else raise ArgumentError, "Unknown event-type: '#{type}'"
|
||||
end
|
||||
end
|
||||
|
||||
def del hook, type = nil
|
||||
case type
|
||||
when READ, :read then @read.delete hook
|
||||
when WRITE, :write then @write.delete hook
|
||||
when ERROR, :error then @error.delete hook
|
||||
when nil
|
||||
@read.delete hook
|
||||
@write.delete hook
|
||||
@error.delete hook
|
||||
else raise ArgumentError, "Unknown event-type: '#{type}'"
|
||||
end
|
||||
end
|
||||
|
||||
def read_set( hook, &event) self.set hook, :read, &event end
|
||||
def write_set( hook, &event) self.set hook, :write, &event end
|
||||
def error_set( hook, &event) self.set hook, :error, &event end
|
||||
def read_del( hook) @read.delete hook end
|
||||
def write_del( hook) @write.delete hook end
|
||||
def error_del( hook) @error.delete hook end
|
||||
|
||||
def run_once timeout = @timeout
|
||||
r, w, e = Kernel.select( @read.keys, @write.keys, @error.keys, timeout)
|
||||
return @tevent.call unless r or w or e
|
||||
r.each {|h| @read[ h].call h, :read }
|
||||
w.each {|h| @write[ h].call h, :write }
|
||||
e.each {|h| @error[ h].call h, :error }
|
||||
end
|
||||
|
||||
def run &e
|
||||
if e
|
||||
until @exit || (@exit_on_empty && self.empty?)
|
||||
self.run_once
|
||||
e.call
|
||||
end
|
||||
else
|
||||
self.run_once until @exit || (@exit_on_empty && self.empty?)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
class Select::Buffer <String
|
||||
def inspect
|
||||
"#<Buffer:#{super}>"
|
||||
end
|
||||
|
||||
def remove x
|
||||
self.slice! 0...x
|
||||
end
|
||||
alias remove! remove
|
||||
|
||||
def each! x
|
||||
return Enumerator.new( self, :each!, x) unless block_given?
|
||||
s = nil
|
||||
yield s while s = self.slice!( x)
|
||||
self
|
||||
end
|
||||
|
||||
def + s
|
||||
self.class.new super(s)
|
||||
end
|
||||
end
|
||||
|
||||
class Select::Socket
|
||||
attr_reader :select, :sock, :bufsize, :parent
|
||||
|
||||
def initialize opts
|
||||
self.init opts
|
||||
@select.read_set @sock, &method( :event_read)
|
||||
@select.error_set @sock, &method( :event_error)
|
||||
end
|
||||
|
||||
def init opts
|
||||
@select = opts[ :select] || Select.new
|
||||
@sock = opts[ :sock] || raise( ArgumentError, "need sock")
|
||||
@bufsize = opts[ :bufsize] || 4096
|
||||
@parent = opts[ :parent] || nil
|
||||
self.delimiter = opts[ :delimiter] || $/
|
||||
@linebuf, @writebuf = Select::Buffer.new(""), Select::Buffer.new("")
|
||||
end
|
||||
|
||||
def delimiter= delimiter
|
||||
@delimiter = case delimiter
|
||||
when Regexp then Regexp.new "^.*?"+delimiter.source
|
||||
when Fixnum, Bignum then /^.{#{delimiter}}/
|
||||
else Regexp.new "^.*?"+Regexp.quote( delimiter.to_s)
|
||||
end
|
||||
end
|
||||
|
||||
def write str
|
||||
e = @writebuf.empty?
|
||||
@writebuf += str
|
||||
@select.write_set @sock, &method( :event_write) if e
|
||||
end
|
||||
alias :print :write
|
||||
alias :<< :write
|
||||
|
||||
def puts str
|
||||
self.write "#{str}\n"
|
||||
end
|
||||
|
||||
def close
|
||||
@select.del @sock
|
||||
@sock.close
|
||||
@parent.event_client_closed self if @parent.respond_to? :event_client_closed
|
||||
rescue IOError
|
||||
end
|
||||
|
||||
def event_line line
|
||||
end
|
||||
|
||||
def event_read sock = @sock, event = :read
|
||||
@linebuf += sock.readpartial @bufsize
|
||||
@linebuf.each! @delimiter, &self.method( :event_line)
|
||||
rescue EOFError
|
||||
self.event_eof sock
|
||||
rescue Errno::EPIPE => e
|
||||
self.event_errno e, sock, event
|
||||
rescue IOError
|
||||
self.event_ioerror sock, event
|
||||
rescue Errno::ECONNRESET => e
|
||||
self.event_errno e, sock, event
|
||||
end
|
||||
|
||||
def event_write sock = @sock, event = :write
|
||||
@writebuf.remove! sock.syswrite( @writebuf)
|
||||
@select.write_del sock if @writebuf.empty?
|
||||
@writebuf
|
||||
rescue IOError
|
||||
@select.del @sock
|
||||
@writebuf
|
||||
end
|
||||
|
||||
def event_eof sock = @sock
|
||||
self.close
|
||||
end
|
||||
|
||||
def event_errno errno, sock = @sock, event = :error
|
||||
self.close
|
||||
end
|
||||
|
||||
def event_error sock = @sock, event = :error
|
||||
self.close
|
||||
end
|
||||
|
||||
def event_ioerror sock = @sock, event = :error
|
||||
end
|
||||
|
||||
def closed?
|
||||
@sock.closed?
|
||||
end
|
||||
end
|
||||
|
||||
class Select::Server
|
||||
attr_reader :select, :clientclass
|
||||
|
||||
def initialize opts
|
||||
raise "You can't use this class directly. Create a subclass" if self.class.superclass == Object
|
||||
init opts
|
||||
select.read_set @sock, &self.method( :event_conn)
|
||||
end
|
||||
|
||||
def init opts
|
||||
@sock = opts[ :sock] || raise( ArgumentError, "need sock")
|
||||
@select = opts[ :select] || Select.new
|
||||
@clientclass = opts[ :clientclass] || Select::Socket
|
||||
@clients = []
|
||||
end
|
||||
|
||||
def run
|
||||
@select.run
|
||||
end
|
||||
|
||||
def delete sock
|
||||
@select.del sock
|
||||
end
|
||||
|
||||
def event_conn sock = @sock, event = :read
|
||||
a = sock.accept
|
||||
c = event_new_client a
|
||||
if c.kind_of? Hash
|
||||
cc = c[ :clientclass] || @clientclass
|
||||
h = { :sock => a, :select => @select, :parent => self }.update c
|
||||
c = cc.new h
|
||||
end
|
||||
@clients.push c
|
||||
end
|
||||
|
||||
def event_error sock = @sock, event = :error
|
||||
end
|
||||
|
||||
def event_new_client sock
|
||||
Hash.new
|
||||
end
|
||||
|
||||
def event_client_closed client
|
||||
@clients.delete client
|
||||
end
|
||||
|
||||
def close
|
||||
@select.del @sock
|
||||
@sock.close
|
||||
rescue IOError
|
||||
end
|
||||
|
||||
def clients_close
|
||||
@clients.each {|c| c.close }
|
||||
end
|
||||
end
|
35
select.gemspec
Normal file
35
select.gemspec
Normal file
|
@ -0,0 +1,35 @@
|
|||
# Generated by jeweler
|
||||
# DO NOT EDIT THIS FILE DIRECTLY
|
||||
# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
Gem::Specification.new do |s|
|
||||
s.name = %q{select}
|
||||
s.version = "0.2.0"
|
||||
|
||||
s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
|
||||
s.authors = ["Denis Knauf"]
|
||||
s.date = %q{2010-01-30}
|
||||
s.description = %q{Select based event-handler for servers and sockets}
|
||||
s.email = %q{Denis.Knauf@gmail.com}
|
||||
s.files = [
|
||||
"VERSION",
|
||||
"lib/select.rb"
|
||||
]
|
||||
s.homepage = %q{http://github.com/DenisKnauf/select}
|
||||
s.rdoc_options = ["--charset=UTF-8"]
|
||||
s.require_paths = ["lib"]
|
||||
s.rubygems_version = %q{1.3.5}
|
||||
s.summary = %q{IO-event-handler based on select}
|
||||
|
||||
if s.respond_to? :specification_version then
|
||||
current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
|
||||
s.specification_version = 3
|
||||
|
||||
if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
|
||||
else
|
||||
end
|
||||
else
|
||||
end
|
||||
end
|
||||
|
Loading…
Reference in a new issue