chapman 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'http://rubygems.org'
+
+# Specify your gem's dependencies in chapman.gemspec
+gemspec

data/README.md

@@ -0,0 +1,4 @@
+chapman
+=======
+
+Like stalker, but can stalk concurrently

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/chapman.gemspec

@@ -0,0 +1,17 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/chapman/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Tyler Flint", 'Lyon Hill']
+  gem.email         = ["tylerflint@gmail.com", 'lyondhill@gmail.com']
+  gem.description   = %q{Like stalker, but can stalk concurrently}
+  gem.summary       = %q{Takes the concept of stalker and introduces a thread pool. It's great for scenarios where tasks are light and mostly IO bound.}
+  gem.homepage      = ""
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  gem.name          = "chapman"
+  gem.require_paths = ["lib"]
+  gem.version       = Chapman::VERSION
+end

data/lib/chapman/exceptions.rb

@@ -0,0 +1,8 @@
+module Chapman
+  module Exceptions
+    class NoJobsDefined < RuntimeError; end
+    class NoSuchJob < RuntimeError; end
+    class JobTimeout < RuntimeError; end
+    class BadURL < RuntimeError; end    
+  end
+end

data/lib/chapman/version.rb

@@ -0,0 +1,3 @@
+module Chapman
+  VERSION = "0.0.1"
+end

data/lib/chapman/worker.rb

@@ -0,0 +1,162 @@
+require 'beanstalk-client'
+require 'json'
+require 'uri'
+require 'timeout'
+
+module Chapman
+  class Worker
+
+    def initialize
+      @handlers        = Chapman.job_handlers
+      @before_handlers = Chapman.before_handlers
+      @error_handler   = Chapman.error_handler
+    end
+
+    def job_in_progress?
+      @in_progress ||= false
+    end
+
+    def prep(jobs=nil)
+      raise Chapman::Exceptions::NoJobsDefined unless defined?(@handlers)
+      @error_handler = nil unless defined?(@error_handler)
+
+      jobs ||= all_jobs
+
+      jobs.each do |job|
+        raise(Chapman::Exceptions::NoSuchJob, job) unless @handlers[job]
+      end
+
+      log "Working #{jobs.size} jobs: [ #{jobs.join(' ')} ]"
+
+      jobs.each { |job| beanstalk.watch(job) }
+
+      beanstalk.list_tubes_watched.each do |server, tubes|
+        tubes.each { |tube| beanstalk.ignore(tube) unless jobs.include?(tube) }
+      end
+    rescue Beanstalk::NotConnected => e
+      failed_connection(e)
+    end
+
+    def work(jobs=nil)
+      prep(jobs)
+      while not Chapman.soft_quit?
+        work_one_job
+      end
+    end
+
+    def work_one_job
+      return if Chapman.soft_quit? # just to be safe
+      job = beanstalk.reserve
+
+      @in_progress = true
+
+      name, args = JSON.parse job.body
+      log_job_begin(name, args)
+      
+      handler = @handlers[name]
+      raise(Chapman::Exceptions::NoSuchJob, name) unless handler
+
+      begin
+        if defined? @before_handlers and @before_handlers.respond_to? :each
+          @before_handlers.each do |block|
+            block.call(name)
+          end
+        end
+        handler.call(args)
+      end
+
+      job.delete
+
+      @in_progress = false
+      log_job_end(name)
+    rescue Beanstalk::NotConnected => e
+      failed_connection(e)
+    rescue SystemExit
+      raise
+    rescue => e
+      log_error exception_message(e)
+      job.bury rescue nil
+      log_job_end(name, 'failed') if @job_begun
+      if error_handler
+        if error_handler.arity == 1
+          error_handler.call(e)
+        else
+          error_handler.call(e, name, args)
+        end
+      end
+    end
+
+    def failed_connection(e)
+      log_error exception_message(e)
+      log_error "*** Failed connection to #{beanstalk_url}"
+      log_error "*** Check that beanstalkd is running (or set a different BEANSTALK_URL)"
+      exit 1
+    end
+
+    def log_job_begin(name, args)
+      args_flat = unless args.empty?
+        '(' + args.inject([]) do |accum, (key,value)|
+          accum << "#{key}=#{value}"
+        end.join(' ') + ')'
+      else
+        ''
+      end
+
+      log [ "Working", name, args_flat ].join(' ')
+      @job_begun = Time.now
+    end
+
+    def log_job_end(name, failed=false)
+      ellapsed = Time.now - @job_begun
+      ms = (ellapsed.to_f * 1000).to_i
+      log "Finished #{name} in #{ms}ms #{failed ? ' (failed)' : ''}"
+    end
+
+    def log(msg)
+      puts msg
+    end
+
+    def log_error(msg)
+      STDERR.puts msg
+    end
+
+    def beanstalk
+      @beanstalk ||= Beanstalk::Pool.new(beanstalk_addresses)
+    end
+
+    def beanstalk_url
+      Chapman.beanstalk_url
+    end
+
+    def beanstalk_addresses
+      uris = beanstalk_url.split(/[\s,]+/)
+      uris.map {|uri| beanstalk_host_and_port(uri)}
+    end
+
+    def beanstalk_host_and_port(uri_string)
+      uri = URI.parse(uri_string)
+      raise(Chapman::Exceptions::BadURL, uri_string) if uri.scheme != 'beanstalk'
+      "#{uri.host}:#{uri.port || 11300}"
+    end
+
+    def exception_message(e)
+      msg = [ "Exception #{e.class} -> #{e.message}" ]
+
+      base = File.expand_path(Dir.pwd) + '/'
+      e.backtrace.each do |t|
+        msg << "   #{File.expand_path(t).gsub(/#{base}/, '')}"
+      end
+
+      msg.join("\n")
+    end
+
+    def all_jobs
+      @handlers.keys
+    end
+
+    def error_handler
+      @error_handler
+    end
+
+  end
+end

data/lib/chapman.rb

@@ -0,0 +1,123 @@
+require 'chapman/version'
+require 'chapman/exceptions'
+require 'chapman/worker'
+
+STDOUT.sync = true
+
+module Chapman
+  extend self
+
+  def job(j, &block)
+    @@handlers ||= {}
+    @@handlers[j] = block
+  end
+
+  def before(&block)
+    @@before_handlers ||= []
+    @@before_handlers << block
+  end
+
+  def error(&blk)
+    @@error_handler = blk
+  end
+
+  def running
+    @@running ||= []
+  end
+
+  def soft_quit?
+    @@soft_quit ||= false
+  end
+
+  def soft_quit=(soft_quit)
+    @@soft_quit = soft_quit
+  end
+
+  def work(jobs=nil, thread_count=1)
+
+    # start a worker thread
+    thread_count.times do
+      w = Chapman::Worker.new()
+      t = Thread.new { w.work(jobs) }
+      running << {thread: t, worker: w}
+    end
+
+    # keep them alive
+    while not soft_quit?
+      maintain_workers
+      sleep 1
+    end
+
+    murder_workers!
+
+    reap_workers
+
+    log "SEPPUKU!!"
+  end
+
+  def maintain_workers
+    running.each_with_index do |runner, index|
+      if not runner[:thread].alive?
+        w = Creeper::Worker.new()
+        t =  Thread.new do
+          w.work(jobs)
+      end
+        running[index] = {thread: t, worker: w}
+      end
+    end
+  end
+
+  def murder_workers!
+    running.each do |runner|
+      if runner[:worker].job_in_progress?
+        log "Murder [scheduling]"
+      else
+        log "Murder [now]"
+        runner[:thread].kill
+      end
+    end
+  end
+
+  def reap_workers
+    running.each do |runner|
+      runner[:thread].join
+    end
+  end
+
+  def log(msg)
+    puts msg
+  end
+
+  def log_error(msg)
+    STDERR.puts msg
+  end
+
+  def beanstalk_url
+    return @@url if defined?(@@url) and @@url
+    ENV['BEANSTALK_URL'] || 'beanstalk://localhost/'
+  end
+
+  def all_jobs
+    @@handlers.keys
+  end
+
+  def job_handlers
+    @@handlers ||= {}
+  end
+
+  def before_handlers
+    @@before_handlers ||= []
+  end
+
+  def error_handler
+    @@error_handler ||= nil
+  end
+
+  def reset!
+    @@soft_quit       = false
+    @@running         = []
+    @@handlers        = nil
+    @@before_handlers = nil
+    @@error_handler   = nil
+  end
+end

metadata

@@ -0,0 +1,58 @@
+--- !ruby/object:Gem::Specification
+name: chapman
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Tyler Flint
+- Lyon Hill
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-26 00:00:00.000000000 Z
+dependencies: []
+description: Like stalker, but can stalk concurrently
+email:
+- tylerflint@gmail.com
+- lyondhill@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- chapman.gemspec
+- lib/chapman.rb
+- lib/chapman/exceptions.rb
+- lib/chapman/version.rb
+- lib/chapman/worker.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.20
+signing_key: 
+specification_version: 3
+summary: Takes the concept of stalker and introduces a thread pool. It's great for
+  scenarios where tasks are light and mostly IO bound.
+test_files: []
+has_rdoc: