creeper 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 'https://rubygems.org'
+
+# Specify your gem's dependencies in creeper.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 lyon
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# Creeper
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'creeper'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install creeper
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,6 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new('spec')

data/creeper.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/creeper/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["lyon"]
+  gem.email         = ["lyondhill@gmail.com"]
+  gem.description   = %q{Stalker with threads}
+  gem.summary       = %q{A better solution for io bound jobs, same as stalker in functionality but more threadie.}
+  gem.homepage      = "https://github.com/lyondhill/creeper"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "creeper"
+  gem.require_paths = ["lib"]
+  gem.version       = Creeper::VERSION
+
+  gem.add_dependency "beanstalk-client"
+  gem.add_development_dependency 'rspec'
+  gem.add_dependency 'pry'
+
+end
+

data/lib/creeper.rb

@@ -0,0 +1,161 @@
+require 'beanstalk-client'
+require "creeper/version"
+require "creeper/worker"
+require 'json'
+require 'uri'
+require 'timeout'
+
+STDOUT.sync = true
+
+module Creeper
+  extend self
+
+  def connect(url)
+    @@url = url
+    beanstalk
+  end
+
+  def enqueue(job, args={}, opts={})
+    pri   = opts[:pri]   || 65536
+    delay = [0, opts[:delay].to_i].max  
+    ttr   = opts[:ttr]   || 120
+    beanstalk.use job
+    beanstalk.put [ job, args ].to_json, pri, delay, ttr
+  rescue Beanstalk::NotConnected => e
+    failed_connection(e)
+  end
+
+  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)
+    thread_count.times do
+      w = Creeper::Worker.new()
+      t =  Thread.new do
+        w.work(jobs)
+      end
+      running << {thread: t, worker: w}
+    end
+
+    while not soft_quit?
+      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
+      sleep 1
+    end
+    running.each do |runner|
+      if runner[:worker].job_in_progress?
+        log "Murder [scheduling]"
+        runner[:worker].soft_quit = true  
+      else
+        log "Murder [now]"
+        runner[:thread].kill
+      end
+    end
+    running.each do |runner|
+      runner[:thread].join
+    end
+    log "SEPPUKU!!"
+  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(msg)
+    puts msg
+  end
+
+  def log_error(msg)
+    STDERR.puts msg
+  end
+
+  def beanstalk
+    @@beanstalk ||= Beanstalk::Pool.new(beanstalk_addresses)
+  end
+
+  def beanstalk_url
+    return @@url if defined?(@@url) and @@url
+    ENV['BEANSTALK_URL'] || 'beanstalk://localhost/'
+  end
+
+  class BadURL < RuntimeError; 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(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 job_handlers
+    @@handlers ||= {}
+  end
+
+  def before_handlers
+    @@before_handlers ||= []
+  end
+
+  def error_handler
+    @@error_handler ||= nil
+  end
+
+  def clear!
+    @@soft_quit = false
+    @@running = []
+    @@handlers = nil
+    @@before_handlers = nil
+    @@error_handler = nil
+  end
+end

data/lib/creeper/version.rb

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

data/lib/creeper/worker.rb

@@ -0,0 +1,168 @@
+require 'beanstalk-client'
+require 'json'
+require 'uri'
+require 'timeout'
+
+class Creeper::Worker
+  def initialize
+    @handlers = Creeper.job_handlers
+    @before_handlers = Creeper.before_handlers
+    @error_handler = Creeper.error_handler
+  end
+
+  class NoJobsDefined < RuntimeError; end
+  class NoSuchJob < RuntimeError; end
+  class JobTimeout < RuntimeError; end
+  class BadURL < RuntimeError; end
+
+  def soft_quit?
+    @soft_quit ||= false
+  end
+
+  def soft_quit=(soft_quit)
+    # exit if !job_in_progress? && soft_quit
+    @soft_quit = soft_quit
+  end
+
+  def job_in_progress?
+    @in_progress ||= false
+  end
+
+  def prep(jobs=nil)
+    raise NoJobsDefined unless defined?(@handlers)
+    @error_handler = nil unless defined?(@error_handler)
+
+    jobs ||= all_jobs
+
+    jobs.each do |job|
+      raise(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)
+    loop { work_one_job }
+  end
+
+  def work_one_job
+    Thread.current.kill if soft_quit?
+    job = beanstalk.reserve
+    name, args = JSON.parse job.body
+    log_job_begin(name, args)
+    handler = @handlers[name]
+    raise(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
+    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)
+    @in_progress = true
+    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)
+    @in_progress = 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
+    Creeper.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(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
+

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: creeper
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- lyon
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: beanstalk-client
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Stalker with threads
+email:
+- lyondhill@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- creeper.gemspec
+- lib/creeper.rb
+- lib/creeper/version.rb
+- lib/creeper/worker.rb
+homepage: https://github.com/lyondhill/creeper
+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: A better solution for io bound jobs, same as stalker in functionality but
+  more threadie.
+test_files: []
+has_rdoc: