threadify 0.0.2

data/README

@@ -0,0 +1,74 @@
+NAME
+  threadify.rb
+
+
+SYNOPSIS
+  enumerable = %w( a b c d )
+  enumerable.threadify(2){ 'process this block using two worker threads' }
+
+DESCRIPTION
+  threadify.rb makes it stupid easy to process a bunch of data using 'n'
+  worker threads
+
+INSTALL
+  gem install threadify
+
+URI
+  http://rubyforge.org/projects/codeforpeople
+
+SAMPLES
+
+  <========< sample/a.rb >========>
+
+  ~ > cat sample/a.rb
+
+    require 'open-uri'
+    require 'yaml'
+    
+    require 'rubygems'
+    require 'threadify'
+    
+    
+    uris =
+      %w(
+        http://google.com
+        http://yahoo.com
+        http://rubyforge.org
+        http://ruby-lang.org
+        http://kcrw.org
+        http://drawohara.com
+        http://codeforpeople.com
+      )
+    
+    
+    time 'without threadify' do
+      uris.each do |uri|
+        body = open(uri){|pipe| pipe.read}
+      end
+    end
+    
+    
+    time 'with threadify' do
+      uris.threadify do |uri|
+        body = open(uri){|pipe| pipe.read}
+      end
+    end
+    
+    
+    BEGIN {
+      def time label
+        a = Time.now.to_f
+        yield
+      ensure
+        b = Time.now.to_f
+        y label => (b - a)
+      end
+    }
+
+  ~ > ruby sample/a.rb
+
+    --- 
+    without threadify: 7.41900205612183
+    --- 
+    with threadify: 3.69886112213135
+

data/gemspec.rb

@@ -0,0 +1,35 @@
+lib, version = File::basename(File::dirname(File::expand_path(__FILE__))).split %r/-/, 2
+
+require 'rubygems'
+
+Gem::Specification::new do |spec|
+  $VERBOSE = nil
+
+  shiteless = lambda do |list|
+    list.delete_if do |file|
+      file =~ %r/\.svn/ or
+      file =~ %r/\.tmp/
+    end
+  end
+
+  spec.name = lib 
+  spec.version = version 
+  spec.platform = Gem::Platform::RUBY
+  spec.summary = lib 
+
+  spec.files = shiteless[Dir::glob("**/**")]
+  spec.executables = shiteless[Dir::glob("bin/*")].map{|exe| File::basename exe}
+  
+  spec.require_path = "lib" 
+
+  spec.has_rdoc = File::exist? "doc" 
+  spec.test_suite_file = "test/#{ lib }.rb" if File::directory? "test"
+  #spec.add_dependency 'lib', '>= version'
+
+  spec.extensions << "extconf.rb" if File::exists? "extconf.rb"
+
+  spec.rubyforge_project = 'codeforpeople'
+  spec.author = "Ara T. Howard"
+  spec.email = "ara.t.howard@gmail.com"
+  spec.homepage = "http://codeforpeople.com/lib/ruby/#{ lib }/"
+end

data/gen_readme.rb

@@ -0,0 +1,32 @@
+require 'pathname'
+
+$VERBOSE=nil
+
+def indent s, n = 2
+  ws = ' ' * n
+  s.gsub %r/^/, ws 
+end
+
+template = IO::read 'README.tmpl'
+
+samples = ''
+prompt = '~ > '
+
+Dir['sample*/*'].sort.each do |sample|
+  samples << "\n" << "  <========< #{ sample } >========>" << "\n\n"
+
+  cmd = "cat #{ sample }" 
+  samples << indent(prompt + cmd, 2) << "\n\n" 
+  samples << indent(`#{ cmd }`, 4) << "\n" 
+
+  cmd = "ruby #{ sample }" 
+  samples << indent(prompt + cmd, 2) << "\n\n" 
+
+  cmd = "ruby -Ilib #{ sample }" 
+  samples << indent(`#{ cmd } 2>&1`, 4) << "\n" 
+end
+
+#samples.gsub! %r/^/, '  '
+
+readme  = template.gsub %r/^\s*@samples\s*$/, samples
+print readme 

data/install.rb

@@ -0,0 +1,214 @@
+#!/usr/bin/env ruby
+require 'rbconfig'
+require 'find'
+require 'ftools'
+require 'tempfile'
+include Config
+
+LIBDIR      = "lib"
+LIBDIR_MODE = 0644
+
+BINDIR      = "bin"
+BINDIR_MODE = 0755
+
+
+$srcdir            = CONFIG["srcdir"]
+$version           = CONFIG["MAJOR"]+"."+CONFIG["MINOR"]
+$libdir            = File.join(CONFIG["libdir"], "ruby", $version)
+$archdir           = File.join($libdir, CONFIG["arch"])
+$site_libdir       = $:.find {|x| x =~ /site_ruby$/}
+$bindir            = CONFIG["bindir"] || CONFIG['BINDIR']
+$ruby_install_name = CONFIG['ruby_install_name'] || CONFIG['RUBY_INSTALL_NAME'] || 'ruby'
+$ruby_ext          = CONFIG['EXEEXT'] || ''
+$ruby              = File.join($bindir, ($ruby_install_name + $ruby_ext))
+
+if !$site_libdir
+  $site_libdir = File.join($libdir, "site_ruby")
+elsif $site_libdir !~ %r/#{Regexp.quote($version)}/
+  $site_libdir = File.join($site_libdir, $version)
+end
+
+def install_rb(srcdir=nil, destdir=nil, mode=nil, bin=nil)
+#{{{
+  path   = []
+  dir    = []
+  Find.find(srcdir) do |f|
+    next unless FileTest.file?(f)
+    next if (f = f[srcdir.length+1..-1]) == nil
+    next if (/CVS$/ =~ File.dirname(f))
+    next if (/\.svn/ =~ File.dirname(f))
+    next if f =~ %r/\.lnk/
+    next if f =~ %r/\.svn/
+    next if f =~ %r/\.swp/
+    next if f =~ %r/\.svn/
+    path.push f
+    dir |= [File.dirname(f)]
+  end
+  for f in dir
+    next if f == "."
+    next if f == "CVS"
+    File::makedirs(File.join(destdir, f))
+  end
+  for f in path
+    next if (/\~$/ =~ f)
+    next if (/^\./ =~ File.basename(f))
+    unless bin
+      File::install(File.join(srcdir, f), File.join(destdir, f), mode, true)
+    else
+      from = File.join(srcdir, f)
+      to = File.join(destdir, f)
+      shebangify(from) do |sf|
+        $deferr.print from, " -> ", File::catname(from, to), "\n"
+        $deferr.printf "chmod %04o %s\n", mode, to 
+        File::install(sf, to, mode, false)
+      end
+    end
+  end
+#}}}
+end
+def shebangify f
+#{{{
+  open(f) do |fd|
+    buf = fd.read 42 
+    if buf =~ %r/^\s*#\s*!.*ruby/o
+      ftmp = Tempfile::new("#{ $$ }_#{ File::basename(f) }")
+      begin
+        fd.rewind
+        ftmp.puts "#!#{ $ruby  }"
+        while((buf = fd.read(8192)))
+          ftmp.write buf
+        end
+        ftmp.close
+        yield ftmp.path
+      ensure
+        ftmp.close!
+      end
+    else
+      yield f
+    end
+  end
+#}}}
+end
+def ARGV.switch
+#{{{
+  return nil if self.empty?
+  arg = self.shift
+  return nil if arg == '--'
+  if arg =~ /^-(.)(.*)/
+    return arg if $1 == '-'
+    raise 'unknown switch "-"' if $2.index('-')
+    self.unshift "-#{$2}" if $2.size > 0
+    "-#{$1}"
+  else
+    self.unshift arg
+    nil
+  end
+#}}}
+end
+def ARGV.req_arg
+#{{{
+  self.shift || raise('missing argument')
+#}}}
+end
+def linkify d, linked = []
+#--{{{
+  if test ?d, d
+    versioned = Dir[ File::join(d, "*-[0-9].[0-9].[0-9].rb") ]
+    versioned.each do |v| 
+      src, dst = v, v.gsub(%r/\-[\d\.]+\.rb$/, '.rb')
+      lnk = nil
+      begin
+        if test ?l, dst
+          lnk = "#{ dst }.lnk"
+          puts "#{ dst } -> #{ lnk }"
+          File::rename dst, lnk
+        end
+        unless test ?e, dst
+          puts "#{ src } -> #{ dst }"
+          File::copy src, dst 
+          linked << dst
+        end
+      ensure
+        if lnk
+          at_exit do
+            puts "#{ lnk } -> #{ dst }"
+            File::rename lnk, dst
+          end
+        end
+      end
+    end
+  end
+  linked
+#--}}}
+end
+
+
+#
+# main program
+#
+
+libdir = $site_libdir
+bindir = $bindir
+no_linkify = false
+linked = nil
+help = false
+
+usage = <<-usage
+  #{ File::basename $0 }
+    -d, --destdir    <destdir>
+    -l, --libdir     <libdir>
+    -b, --bindir     <bindir>
+    -r, --ruby       <ruby>
+    -n, --no_linkify
+    -s, --sudo
+    -h, --help
+usage
+
+begin
+  while switch = ARGV.switch
+    case switch
+    when '-d', '--destdir'
+      libdir = ARGV.req_arg
+    when '-l', '--libdir'
+      libdir = ARGV.req_arg
+    when '-b', '--bindir'
+      bindir = ARGV.req_arg
+    when '-r', '--ruby'
+      $ruby = ARGV.req_arg
+    when '-n', '--no_linkify'
+      no_linkify = true
+    when '-s', '--sudo'
+      sudo = 'sudo' 
+    when '-h', '--help'
+      help = true
+    else
+      raise "unknown switch #{switch.dump}"
+    end
+  end
+rescue
+  STDERR.puts $!.to_s
+  STDERR.puts usage
+  exit 1
+end    
+
+if help
+  STDOUT.puts usage
+  exit
+end
+
+system "#{ sudo } #{ $ruby } pre-install.rb" if test(?s, 'pre-install.rb')
+
+unless no_linkify
+  linked = linkify('lib') + linkify('bin')
+end
+
+system "#{ $ruby } extconf.rb && make && #{ sudo } make install" if test(?s, 'extconf.rb')
+
+install_rb(LIBDIR, libdir, LIBDIR_MODE)
+install_rb(BINDIR, bindir, BINDIR_MODE, bin=true)
+
+if linked
+  linked.each{|path| File::rm_f path}
+end
+
+system "#{ sudo } #{ $ruby } post-install.rb" if test(?s, 'post-install.rb')

data/lib/threadify.rb

@@ -0,0 +1,94 @@
+module Threadify
+  VERSION = '0.0.2'
+  def Threadify.version() Threadify::VERSION end
+
+  require 'thread'
+
+  @threads = 8
+  @abort_on_exception = true
+
+  class << self
+    attr_accessor :threads
+    attr_accessor :abort_on_exception
+  end
+end
+
+module Enumerable
+  def threadify opts = {}, &block
+  # setup
+  #
+    opts = {:threads => opts} if Numeric === opts
+    threads = Integer(opts[:threads] || opts['threads'] || Threadify.threads)
+    done = Object.new.freeze
+    jobs = Queue.new
+
+  # produce jobs
+  #
+    each_with_index{|elem, i| jobs.push [elem, i]}
+    threads.times{ jobs.push done} # mark the end
+
+  # fire off consumers
+  #
+    consumers = Array.new threads 
+
+    threads.times do |i|
+      consumers[i] = Thread.new do
+        this = Thread.current
+        this.abort_on_exception = Threadify.abort_on_exception
+        loop{
+          job = jobs.pop
+          break if job == done
+          args = job.first
+          jobs << (job << block.call(*args))
+        }
+      end
+    end
+
+  # wait for consumers to finish
+  #
+    consumers.map{|t| t.join}
+
+  # collect the results and return them
+  #
+    jobs.push done
+    ret = []
+    while((job = jobs.pop) != done)
+      elem, i, value = job
+      ret[i] = value
+    end
+    ret
+  end
+end
+
+class Thread
+  def Thread.ify enumerable, *args, &block
+    enumerable.send :threadify, *args, &block
+  end
+end
+
+
+if __FILE__ == $0
+  require 'open-uri'
+  require 'yaml'
+
+  uris = %w( http://google.com http://yahoo.com http://rubyforge.org/ http://ruby-lang.org)
+
+  Thread.ify uris, :threads => 3 do |uri|
+    body = open(uri){|pipe| pipe.read}
+    y uri => body.size
+  end
+end
+
+
+__END__
+
+sample output
+
+--- 
+http://yahoo.com: 9562
+--- 
+http://google.com: 6290
+--- 
+http://rubyforge.org/: 22352
+--- 
+http://ruby-lang.org: 9984

data/sample/a.rb

@@ -0,0 +1,42 @@
+require 'open-uri'
+require 'yaml'
+
+require 'rubygems'
+require 'threadify'
+
+
+uris =
+  %w(
+    http://google.com
+    http://yahoo.com
+    http://rubyforge.org
+    http://ruby-lang.org
+    http://kcrw.org
+    http://drawohara.com
+    http://codeforpeople.com
+  )
+
+
+time 'without threadify' do
+  uris.each do |uri|
+    body = open(uri){|pipe| pipe.read}
+  end
+end
+
+
+time 'with threadify' do
+  uris.threadify do |uri|
+    body = open(uri){|pipe| pipe.read}
+  end
+end
+
+
+BEGIN {
+  def time label
+    a = Time.now.to_f
+    yield
+  ensure
+    b = Time.now.to_f
+    y label => (b - a)
+  end
+}

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification 
+name: threadify
+version: !ruby/object:Gem::Version 
+  version: 0.0.2
+platform: ruby
+authors: 
+- Ara T. Howard
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-07-01 00:00:00 -06:00
+default_executable: 
+dependencies: []
+
+description: 
+email: ara.t.howard@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- gemspec.rb
+- gen_readme.rb
+- install.rb
+- lib
+- lib/threadify.rb
+- README
+- sample
+- sample/a.rb
+- threadify-0.0.2.gem
+has_rdoc: false
+homepage: http://codeforpeople.com/lib/ruby/threadify/
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: codeforpeople
+rubygems_version: 1.0.1
+signing_key: 
+specification_version: 2
+summary: threadify
+test_files: []
+